Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 1 | :mod:`pty` --- Pseudo-terminal utilities |
| 2 | ======================================== |
| 3 | |
| 4 | .. module:: pty |
Benjamin Peterson | a0dfa82 | 2009-11-13 02:25:08 +0000 | [diff] [blame] | 5 | :platform: Linux |
| 6 | :synopsis: Pseudo-Terminal Handling for Linux. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 7 | .. moduleauthor:: Steen Lumholt |
| 8 | .. sectionauthor:: Moshe Zadka <moshez@zadka.site.co.il> |
| 9 | |
| 10 | |
| 11 | The :mod:`pty` module defines operations for handling the pseudo-terminal |
| 12 | concept: starting another process and being able to write to and read from its |
| 13 | controlling terminal programmatically. |
| 14 | |
Mark Dickinson | 934896d | 2009-02-21 20:59:32 +0000 | [diff] [blame] | 15 | Because pseudo-terminal handling is highly platform dependent, there is code to |
Benjamin Peterson | a0dfa82 | 2009-11-13 02:25:08 +0000 | [diff] [blame] | 16 | do it only for Linux. (The Linux code is supposed to work on other platforms, |
| 17 | but hasn't been tested yet.) |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 18 | |
| 19 | The :mod:`pty` module defines the following functions: |
| 20 | |
| 21 | |
| 22 | .. function:: fork() |
| 23 | |
| 24 | Fork. Connect the child's controlling terminal to a pseudo-terminal. Return |
| 25 | value is ``(pid, fd)``. Note that the child gets *pid* 0, and the *fd* is |
| 26 | *invalid*. The parent's return value is the *pid* of the child, and *fd* is a |
| 27 | file descriptor connected to the child's controlling terminal (and also to the |
| 28 | child's standard input and output). |
| 29 | |
| 30 | |
| 31 | .. function:: openpty() |
| 32 | |
| 33 | Open a new pseudo-terminal pair, using :func:`os.openpty` if possible, or |
Benjamin Peterson | a0dfa82 | 2009-11-13 02:25:08 +0000 | [diff] [blame] | 34 | emulation code for generic Unix systems. Return a pair of file descriptors |
| 35 | ``(master, slave)``, for the master and the slave end, respectively. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 36 | |
| 37 | |
| 38 | .. function:: spawn(argv[, master_read[, stdin_read]]) |
| 39 | |
| 40 | Spawn a process, and connect its controlling terminal with the current |
| 41 | process's standard io. This is often used to baffle programs which insist on |
| 42 | reading from the controlling terminal. |
| 43 | |
| 44 | The functions *master_read* and *stdin_read* should be functions which read from |
Georg Brandl | 9afde1c | 2007-11-01 20:32:30 +0000 | [diff] [blame] | 45 | a file descriptor. The defaults try to read 1024 bytes each time they are |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 46 | called. |
| 47 | |
Georg Brandl | 4cf83f4 | 2010-12-30 17:22:33 +0000 | [diff] [blame] | 48 | |
| 49 | Example |
| 50 | ------- |
| 51 | |
| 52 | .. sectionauthor:: Steen Lumholt |
| 53 | |
| 54 | The following program acts like the Unix command :manpage:`script(1)`, using a |
| 55 | pseudo-terminal to record all input and output of a terminal session in a |
| 56 | "typescript". :: |
| 57 | |
| 58 | import sys, os, time, getopt |
| 59 | import pty |
| 60 | |
| 61 | mode = 'wb' |
| 62 | shell = 'sh' |
| 63 | filename = 'typescript' |
| 64 | if 'SHELL' in os.environ: |
| 65 | shell = os.environ['SHELL'] |
| 66 | |
| 67 | try: |
| 68 | opts, args = getopt.getopt(sys.argv[1:], 'ap') |
| 69 | except getopt.error as msg: |
| 70 | print('%s: %s' % (sys.argv[0], msg)) |
| 71 | sys.exit(2) |
| 72 | |
| 73 | for opt, arg in opts: |
| 74 | # option -a: append to typescript file |
| 75 | if opt == '-a': |
| 76 | mode = 'ab' |
| 77 | # option -p: use a Python shell as the terminal command |
| 78 | elif opt == '-p': |
| 79 | shell = sys.executable |
| 80 | if args: |
| 81 | filename = args[0] |
| 82 | |
| 83 | script = open(filename, mode) |
| 84 | |
| 85 | def read(fd): |
| 86 | data = os.read(fd, 1024) |
| 87 | script.write(data) |
| 88 | return data |
| 89 | |
| 90 | sys.stdout.write('Script started, file is %s\n' % filename) |
| 91 | script.write(('Script started on %s\n' % time.asctime()).encode()) |
| 92 | pty.spawn(shell, read) |
| 93 | script.write(('Script done on %s\n' % time.asctime()).encode()) |
| 94 | sys.stdout.write('Script done, file is %s\n' % filename) |