Demo/scripts/script.py

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