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) 43
