Demo/scripts/script.py

   1 #! /usr/bin/env python
   2 # script.py -- Make typescript of terminal session.
   3 # Usage:
   4 #       -a      Append to typescript.
   5 #       -p      Use Python as shell.
   6 # Author: Steen Lumholt.
   7
   8
   9 import os, time, sys
  10 import pty
  11
  12 def read(fd):
  13     data = os.read(fd, 1024)
  14     file.write(data)
  15     return data
  16
  17 shell = 'sh'
  18 filename = 'typescript'
  19 mode = 'w'
  20 if os.environ.has_key('SHELL'):
  21     shell = os.environ['SHELL']
  22 if '-a' in sys.argv:
  23     mode = 'a'
  24 if '-p' in sys.argv:
  25     shell = 'python'
  26
  27 file = open(filename, mode)
  28
  29 sys.stdout.write('Script started, file is %s\n' % filename)
  30 file.write('Script started on %s\n' % time.ctime(time.time()))
  31 pty.spawn(shell, read)
  32 file.write('Script done on %s\n' % time.ctime(time.time()))
  33 sys.stdout.write('Script done, file is %s\n' % filename)