Commit cc152712 authored by Berker Peksag's avatar Berker Peksag

Issue #23983: Update the pty module example.

Changes:

* Fixed a ResourceWarning warning
* Used argparse instead of getopt
parent 0f4db8b0
...@@ -58,40 +58,32 @@ The following program acts like the Unix command :manpage:`script(1)`, using a ...@@ -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 pseudo-terminal to record all input and output of a terminal session in a
"typescript". :: "typescript". ::
import sys, os, time, getopt import argparse
import os
import pty import pty
import sys
import time
mode = 'wb' parser = argparse.ArgumentParser()
shell = 'sh' parser.add_argument('-a', dest='append', action='store_true')
filename = 'typescript' parser.add_argument('-p', dest='use_python', action='store_true')
if 'SHELL' in os.environ: parser.add_argument('filename', nargs='?', default='typescript')
shell = os.environ['SHELL'] 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)
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]
script = open(filename, mode)
shell = sys.executable if options.use_python else os.environ.get('SHELL', 'sh')
filename = options.filename
mode = 'ab' if options.append else 'wb'
with open(filename, mode) as script:
def read(fd): def read(fd):
data = os.read(fd, 1024) data = os.read(fd, 1024)
script.write(data) script.write(data)
return data return data
sys.stdout.write('Script started, file is %s\n' % filename) print('Script started, file is', filename)
script.write(('Script started on %s\n' % time.asctime()).encode()) script.write(('Script started on %s\n' % time.asctime()).encode())
pty.spawn(shell, read) pty.spawn(shell, read)
script.write(('Script done on %s\n' % time.asctime()).encode()) script.write(('Script done on %s\n' % time.asctime()).encode())
sys.stdout.write('Script done, file is %s\n' % filename) print('Script done, file is', filename)
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment