Issue #23983: Update the pty module example.

Changes:

* Fixed a ResourceWarning warning
* Used argparse instead of getopt
This commit is contained in:
Berker Peksag 2015-05-12 17:25:06 +03:00
parent ea6d5592f2
commit c2dd680115
1 changed files with 23 additions and 31 deletions

View File

@ -58,40 +58,32 @@ The following program acts like the Unix command :manpage:`script(1)`, using a
pseudo-terminal to record all input and output of a terminal session in a
"typescript". ::
import sys, os, time, getopt
import pty
import argparse
import os
import pty
import sys
import time
mode = 'wb'
shell = 'sh'
filename = 'typescript'
if 'SHELL' in os.environ:
shell = os.environ['SHELL']
parser = argparse.ArgumentParser()
parser.add_argument('-a', dest='append', action='store_true')
parser.add_argument('-p', dest='use_python', action='store_true')
parser.add_argument('filename', nargs='?', default='typescript')
options = parser.parse_args()
try:
opts, args = getopt.getopt(sys.argv[1:], 'ap')
except getopt.error as msg:
print('%s: %s' % (sys.argv[0], msg))
sys.exit(2)
shell = sys.executable if options.use_python else os.environ.get('SHELL', 'sh')
filename = options.filename
mode = 'ab' if options.append else 'wb'
for opt, arg in opts:
# option -a: append to typescript file
if opt == '-a':
mode = 'ab'
# option -p: use a Python shell as the terminal command
elif opt == '-p':
shell = sys.executable
if args:
filename = args[0]
with open(filename, mode) as script:
def read(fd):
data = os.read(fd, 1024)
script.write(data)
return data
script = open(filename, mode)
print('Script started, file is', filename)
script.write(('Script started on %s\n' % time.asctime()).encode())
def read(fd):
data = os.read(fd, 1024)
script.write(data)
return data
pty.spawn(shell, read)
sys.stdout.write('Script started, file is %s\n' % filename)
script.write(('Script started on %s\n' % time.asctime()).encode())
pty.spawn(shell, read)
script.write(('Script done on %s\n' % time.asctime()).encode())
sys.stdout.write('Script done, file is %s\n' % filename)
script.write(('Script done on %s\n' % time.asctime()).encode())
print('Script done, file is', filename)