2000-02-04 15:10:34 +00:00
|
|
|
"""Spawn a command with pipes to its stdin, stdout, and optionally stderr.
|
|
|
|
|
|
|
|
The normal os.popen(cmd, mode) call spawns a shell command and provides a
|
|
|
|
file interface to just the input or output of the process depending on
|
|
|
|
whether mode is 'r' or 'w'. This module provides the functions popen2(cmd)
|
|
|
|
and popen3(cmd) which return two or three pipes to the spawned command.
|
|
|
|
"""
|
|
|
|
|
1995-01-12 12:29:47 +00:00
|
|
|
import os
|
|
|
|
import sys
|
|
|
|
import string
|
|
|
|
|
1998-03-26 21:13:24 +00:00
|
|
|
MAXFD = 256 # Max number of file descriptors (os.getdtablesize()???)
|
1995-01-12 12:29:47 +00:00
|
|
|
|
1997-08-11 03:27:24 +00:00
|
|
|
_active = []
|
|
|
|
|
|
|
|
def _cleanup():
|
|
|
|
for inst in _active[:]:
|
1998-03-26 21:13:24 +00:00
|
|
|
inst.poll()
|
1997-08-11 03:27:24 +00:00
|
|
|
|
|
|
|
class Popen3:
|
2000-02-04 15:10:34 +00:00
|
|
|
"""Class representing a child process. Normally instances are created
|
|
|
|
by the factory functions popen2() and popen3()."""
|
|
|
|
|
1997-09-29 04:04:39 +00:00
|
|
|
def __init__(self, cmd, capturestderr=0, bufsize=-1):
|
2000-02-04 15:10:34 +00:00
|
|
|
"""The parameter 'cmd' is the shell command to execute in a
|
|
|
|
sub-process. The 'capturestderr' flag, if true, specifies that
|
|
|
|
the object should capture standard error output of the child process.
|
|
|
|
The default is false. If the 'bufsize' parameter is specified, it
|
|
|
|
specifies the size of the I/O buffers to/from the child process."""
|
1998-03-26 21:13:24 +00:00
|
|
|
if type(cmd) == type(''):
|
|
|
|
cmd = ['/bin/sh', '-c', cmd]
|
|
|
|
p2cread, p2cwrite = os.pipe()
|
|
|
|
c2pread, c2pwrite = os.pipe()
|
|
|
|
if capturestderr:
|
|
|
|
errout, errin = os.pipe()
|
|
|
|
self.pid = os.fork()
|
|
|
|
if self.pid == 0:
|
|
|
|
# Child
|
|
|
|
os.close(0)
|
|
|
|
os.close(1)
|
|
|
|
if os.dup(p2cread) <> 0:
|
|
|
|
sys.stderr.write('popen2: bad read dup\n')
|
|
|
|
if os.dup(c2pwrite) <> 1:
|
|
|
|
sys.stderr.write('popen2: bad write dup\n')
|
|
|
|
if capturestderr:
|
|
|
|
os.close(2)
|
|
|
|
if os.dup(errin) <> 2: pass
|
|
|
|
for i in range(3, MAXFD):
|
|
|
|
try:
|
|
|
|
os.close(i)
|
|
|
|
except: pass
|
|
|
|
try:
|
|
|
|
os.execvp(cmd[0], cmd)
|
|
|
|
finally:
|
|
|
|
os._exit(1)
|
|
|
|
# Shouldn't come here, I guess
|
|
|
|
os._exit(1)
|
|
|
|
os.close(p2cread)
|
|
|
|
self.tochild = os.fdopen(p2cwrite, 'w', bufsize)
|
|
|
|
os.close(c2pwrite)
|
|
|
|
self.fromchild = os.fdopen(c2pread, 'r', bufsize)
|
|
|
|
if capturestderr:
|
|
|
|
os.close(errin)
|
|
|
|
self.childerr = os.fdopen(errout, 'r', bufsize)
|
|
|
|
else:
|
|
|
|
self.childerr = None
|
|
|
|
self.sts = -1 # Child not completed yet
|
|
|
|
_active.append(self)
|
2000-02-04 15:10:34 +00:00
|
|
|
|
1997-08-11 03:27:24 +00:00
|
|
|
def poll(self):
|
2000-02-04 15:10:34 +00:00
|
|
|
"""Return the exit status of the child process if it has finished,
|
|
|
|
or -1 if it hasn't finished yet."""
|
1998-03-26 21:13:24 +00:00
|
|
|
if self.sts < 0:
|
|
|
|
try:
|
|
|
|
pid, sts = os.waitpid(self.pid, os.WNOHANG)
|
|
|
|
if pid == self.pid:
|
|
|
|
self.sts = sts
|
|
|
|
_active.remove(self)
|
|
|
|
except os.error:
|
|
|
|
pass
|
|
|
|
return self.sts
|
2000-02-04 15:10:34 +00:00
|
|
|
|
1997-08-11 03:27:24 +00:00
|
|
|
def wait(self):
|
2000-02-04 15:10:34 +00:00
|
|
|
"""Wait for and return the exit status of the child process."""
|
1998-03-26 21:13:24 +00:00
|
|
|
pid, sts = os.waitpid(self.pid, 0)
|
|
|
|
if pid == self.pid:
|
|
|
|
self.sts = sts
|
|
|
|
_active.remove(self)
|
|
|
|
return self.sts
|
1997-04-21 14:15:55 +00:00
|
|
|
|
2000-07-09 23:35:24 +00:00
|
|
|
if hasattr(os, "popen2"):
|
|
|
|
def popen2(cmd, mode='t', bufsize=-1):
|
|
|
|
"""Execute the shell command 'cmd' in a sub-process. If 'bufsize' is
|
|
|
|
specified, it sets the buffer size for the I/O pipes. The file objects
|
|
|
|
(child_stdout, child_stdin) are returned."""
|
|
|
|
w, r = os.popen2(cmd, mode, bufsize)
|
|
|
|
return r, w
|
|
|
|
else:
|
2000-07-09 17:59:32 +00:00
|
|
|
def popen2(cmd, mode='t', bufsize=-1):
|
|
|
|
"""Execute the shell command 'cmd' in a sub-process. If 'bufsize' is
|
|
|
|
specified, it sets the buffer size for the I/O pipes. The file objects
|
|
|
|
(child_stdout, child_stdin) are returned."""
|
|
|
|
if type(mode) is type(0) and bufsize == -1:
|
|
|
|
bufsize = mode
|
|
|
|
mode = 't'
|
|
|
|
assert mode in ('t', 'b')
|
|
|
|
_cleanup()
|
|
|
|
inst = Popen3(cmd, 0, bufsize)
|
|
|
|
return inst.fromchild, inst.tochild
|
1997-04-21 14:15:55 +00:00
|
|
|
|
2000-07-09 23:35:24 +00:00
|
|
|
if hasattr(os, "popen3"):
|
|
|
|
def popen3(cmd, mode='t', bufsize=-1):
|
|
|
|
"""Execute the shell command 'cmd' in a sub-process. If 'bufsize' is
|
|
|
|
specified, it sets the buffer size for the I/O pipes. The file objects
|
|
|
|
(child_stdout, child_stdin, child_stderr) are returned."""
|
|
|
|
w, r, e = os.popen3(cmd, mode, bufsize)
|
|
|
|
return r, w, e
|
|
|
|
else:
|
2000-07-09 17:59:32 +00:00
|
|
|
def popen3(cmd, mode='t', bufsize=-1):
|
|
|
|
"""Execute the shell command 'cmd' in a sub-process. If 'bufsize' is
|
|
|
|
specified, it sets the buffer size for the I/O pipes. The file objects
|
|
|
|
(child_stdout, child_stdin, child_stderr) are returned."""
|
|
|
|
if type(mode) is type(0) and bufsize == -1:
|
|
|
|
bufsize = mode
|
|
|
|
mode = 't'
|
|
|
|
assert mode in ('t', 'b')
|
|
|
|
_cleanup()
|
|
|
|
inst = Popen3(cmd, 1, bufsize)
|
|
|
|
return inst.fromchild, inst.tochild, inst.childerr
|
|
|
|
|
2000-07-09 23:35:24 +00:00
|
|
|
if hasattr(os, "popen4"):
|
|
|
|
def popen4(cmd, mode='t', bufsize=-1):
|
|
|
|
"""Execute the shell command 'cmd' in a sub-process. If 'bufsize' is
|
|
|
|
specified, it sets the buffer size for the I/O pipes. The file objects
|
|
|
|
(child_stdout_stderr, child_stdin) are returned."""
|
|
|
|
w, r = os.popen4(cmd, mode, bufsize)
|
|
|
|
return r, w
|
|
|
|
else:
|
|
|
|
pass # not yet on unix
|
1997-08-11 03:27:24 +00:00
|
|
|
|
|
|
|
def _test():
|
2000-08-20 05:57:36 +00:00
|
|
|
cmd = "cat"
|
1997-08-11 03:27:24 +00:00
|
|
|
teststr = "abc\n"
|
2000-08-20 05:57:36 +00:00
|
|
|
resultstr = teststr
|
|
|
|
if os.name == "nt":
|
|
|
|
cmd = "more"
|
|
|
|
resultstr = "\n" + resultstr
|
1997-08-11 03:27:24 +00:00
|
|
|
print "testing popen2..."
|
2000-08-20 05:57:36 +00:00
|
|
|
r, w = popen2(cmd)
|
1997-08-11 03:27:24 +00:00
|
|
|
w.write(teststr)
|
|
|
|
w.close()
|
2000-08-20 05:57:36 +00:00
|
|
|
assert r.read() == resultstr
|
1997-08-11 03:27:24 +00:00
|
|
|
print "testing popen3..."
|
2000-07-09 17:59:32 +00:00
|
|
|
try:
|
2000-08-20 05:57:36 +00:00
|
|
|
r, w, e = popen3([cmd])
|
2000-07-09 17:59:32 +00:00
|
|
|
except:
|
2000-08-20 05:57:36 +00:00
|
|
|
r, w, e = popen3(cmd)
|
1997-08-11 03:27:24 +00:00
|
|
|
w.write(teststr)
|
|
|
|
w.close()
|
2000-08-20 05:57:36 +00:00
|
|
|
assert r.read() == resultstr
|
1997-08-11 03:27:24 +00:00
|
|
|
assert e.read() == ""
|
1999-04-20 12:27:31 +00:00
|
|
|
for inst in _active[:]:
|
|
|
|
inst.wait()
|
1997-08-11 03:27:24 +00:00
|
|
|
assert not _active
|
|
|
|
print "All OK"
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
_test()
|