blob: bd158f66601ceaccad7e28aec735d335466f34cc [file] [log] [blame]
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001# subprocess - Subprocesses with accessible I/O streams
2#
Tim Peterse718f612004-10-12 21:51:32 +00003# For more information about this module, see PEP 324.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00004#
Peter Astrand3a708df2005-09-23 17:37:29 +00005# Copyright (c) 2003-2005 by Peter Astrand <astrand@lysator.liu.se>
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00006#
Peter Astrand69bf13f2005-02-14 08:56:32 +00007# Licensed to PSF under a Contributor Agreement.
Peter Astrand3a708df2005-09-23 17:37:29 +00008# See http://www.python.org/2.4/license for licensing details.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00009
Raymond Hettinger837dd932004-10-17 16:36:53 +000010r"""subprocess - Subprocesses with accessible I/O streams
Fredrik Lundh5b3687d2004-10-12 15:26:28 +000011
Fredrik Lundh15aaacc2004-10-17 14:47:05 +000012This module allows you to spawn processes, connect to their
13input/output/error pipes, and obtain their return codes. This module
14intends to replace several other, older modules and functions, like:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +000015
16os.system
17os.spawn*
Fredrik Lundh5b3687d2004-10-12 15:26:28 +000018
19Information about how the subprocess module can be used to replace these
20modules and functions can be found below.
21
22
23
24Using the subprocess module
25===========================
26This module defines one class called Popen:
27
28class Popen(args, bufsize=0, executable=None,
29 stdin=None, stdout=None, stderr=None,
30 preexec_fn=None, close_fds=False, shell=False,
31 cwd=None, env=None, universal_newlines=False,
32 startupinfo=None, creationflags=0):
33
34
35Arguments are:
36
37args should be a string, or a sequence of program arguments. The
38program to execute is normally the first item in the args sequence or
39string, but can be explicitly set by using the executable argument.
40
41On UNIX, with shell=False (default): In this case, the Popen class
42uses os.execvp() to execute the child program. args should normally
43be a sequence. A string will be treated as a sequence with the string
44as the only item (the program to execute).
45
46On UNIX, with shell=True: If args is a string, it specifies the
47command string to execute through the shell. If args is a sequence,
48the first item specifies the command string, and any additional items
49will be treated as additional shell arguments.
50
51On Windows: the Popen class uses CreateProcess() to execute the child
52program, which operates on strings. If args is a sequence, it will be
53converted to a string using the list2cmdline method. Please note that
54not all MS Windows applications interpret the command line the same
55way: The list2cmdline is designed for applications using the same
56rules as the MS C runtime.
57
58bufsize, if given, has the same meaning as the corresponding argument
59to the built-in open() function: 0 means unbuffered, 1 means line
60buffered, any other positive value means use a buffer of
61(approximately) that size. A negative bufsize means to use the system
62default, which usually means fully buffered. The default value for
63bufsize is 0 (unbuffered).
64
65stdin, stdout and stderr specify the executed programs' standard
66input, standard output and standard error file handles, respectively.
67Valid values are PIPE, an existing file descriptor (a positive
68integer), an existing file object, and None. PIPE indicates that a
69new pipe to the child should be created. With None, no redirection
70will occur; the child's file handles will be inherited from the
71parent. Additionally, stderr can be STDOUT, which indicates that the
72stderr data from the applications should be captured into the same
73file handle as for stdout.
74
75If preexec_fn is set to a callable object, this object will be called
76in the child process just before the child is executed.
77
78If close_fds is true, all file descriptors except 0, 1 and 2 will be
79closed before the child process is executed.
80
81if shell is true, the specified command will be executed through the
82shell.
83
84If cwd is not None, the current directory will be changed to cwd
85before the child is executed.
86
87If env is not None, it defines the environment variables for the new
88process.
89
90If universal_newlines is true, the file objects stdout and stderr are
91opened as a text files, but lines may be terminated by any of '\n',
92the Unix end-of-line convention, '\r', the Macintosh convention or
93'\r\n', the Windows convention. All of these external representations
94are seen as '\n' by the Python program. Note: This feature is only
95available if Python is built with universal newline support (the
96default). Also, the newlines attribute of the file objects stdout,
97stdin and stderr are not updated by the communicate() method.
98
99The startupinfo and creationflags, if given, will be passed to the
100underlying CreateProcess() function. They can specify things such as
101appearance of the main window and priority for the new process.
102(Windows only)
103
104
Georg Brandlf9734072008-12-07 15:30:06 +0000105This module also defines some shortcut functions:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000106
Peter Astrand5f5e1412004-12-05 20:15:36 +0000107call(*popenargs, **kwargs):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000108 Run command with arguments. Wait for command to complete, then
109 return the returncode attribute.
110
111 The arguments are the same as for the Popen constructor. Example:
112
113 retcode = call(["ls", "-l"])
114
Peter Astrand454f7672005-01-01 09:36:35 +0000115check_call(*popenargs, **kwargs):
116 Run command with arguments. Wait for command to complete. If the
117 exit code was zero then return, otherwise raise
118 CalledProcessError. The CalledProcessError object will have the
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000119 return code in the returncode attribute.
Peter Astrand454f7672005-01-01 09:36:35 +0000120
121 The arguments are the same as for the Popen constructor. Example:
122
123 check_call(["ls", "-l"])
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000124
Brett Cannona23810f2008-05-26 19:04:21 +0000125getstatusoutput(cmd):
126 Return (status, output) of executing cmd in a shell.
127
128 Execute the string 'cmd' in a shell with os.popen() and return a 2-tuple
129 (status, output). cmd is actually run as '{ cmd ; } 2>&1', so that the
130 returned output will contain output or error messages. A trailing newline
131 is stripped from the output. The exit status for the command can be
132 interpreted according to the rules for the C function wait(). Example:
133
134 >>> import subprocess
135 >>> subprocess.getstatusoutput('ls /bin/ls')
136 (0, '/bin/ls')
137 >>> subprocess.getstatusoutput('cat /bin/junk')
138 (256, 'cat: /bin/junk: No such file or directory')
139 >>> subprocess.getstatusoutput('/bin/junk')
140 (256, 'sh: /bin/junk: not found')
141
142getoutput(cmd):
143 Return output (stdout or stderr) of executing cmd in a shell.
144
145 Like getstatusoutput(), except the exit status is ignored and the return
146 value is a string containing the command's output. Example:
147
148 >>> import subprocess
149 >>> subprocess.getoutput('ls /bin/ls')
150 '/bin/ls'
151
Georg Brandlf9734072008-12-07 15:30:06 +0000152check_output(*popenargs, **kwargs):
153 Run command with arguments and return its output as a byte string.
154
155 If the exit code was non-zero it raises a CalledProcessError. The
156 CalledProcessError object will have the return code in the returncode
157 attribute and output in the output attribute.
158
159 The arguments are the same as for the Popen constructor. Example:
160
161 output = subprocess.check_output(["ls", "-l", "/dev/null"])
162
Brett Cannona23810f2008-05-26 19:04:21 +0000163
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000164Exceptions
165----------
166Exceptions raised in the child process, before the new program has
167started to execute, will be re-raised in the parent. Additionally,
168the exception object will have one extra attribute called
169'child_traceback', which is a string containing traceback information
170from the childs point of view.
171
172The most common exception raised is OSError. This occurs, for
173example, when trying to execute a non-existent file. Applications
174should prepare for OSErrors.
175
176A ValueError will be raised if Popen is called with invalid arguments.
177
Georg Brandlf9734072008-12-07 15:30:06 +0000178check_call() and check_output() will raise CalledProcessError, if the
179called process returns a non-zero return code.
Peter Astrand454f7672005-01-01 09:36:35 +0000180
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000181
182Security
183--------
184Unlike some other popen functions, this implementation will never call
185/bin/sh implicitly. This means that all characters, including shell
186metacharacters, can safely be passed to child processes.
187
188
189Popen objects
190=============
191Instances of the Popen class have the following methods:
192
193poll()
194 Check if child process has terminated. Returns returncode
195 attribute.
196
197wait()
198 Wait for child process to terminate. Returns returncode attribute.
199
200communicate(input=None)
201 Interact with process: Send data to stdin. Read data from stdout
202 and stderr, until end-of-file is reached. Wait for process to
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000203 terminate. The optional input argument should be a string to be
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000204 sent to the child process, or None, if no data should be sent to
205 the child.
Tim Peterse718f612004-10-12 21:51:32 +0000206
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000207 communicate() returns a tuple (stdout, stderr).
208
209 Note: The data read is buffered in memory, so do not use this
210 method if the data size is large or unlimited.
211
212The following attributes are also available:
213
214stdin
215 If the stdin argument is PIPE, this attribute is a file object
216 that provides input to the child process. Otherwise, it is None.
217
218stdout
219 If the stdout argument is PIPE, this attribute is a file object
220 that provides output from the child process. Otherwise, it is
221 None.
222
223stderr
224 If the stderr argument is PIPE, this attribute is file object that
225 provides error output from the child process. Otherwise, it is
226 None.
227
228pid
229 The process ID of the child process.
230
231returncode
232 The child return code. A None value indicates that the process
233 hasn't terminated yet. A negative value -N indicates that the
234 child was terminated by signal N (UNIX only).
235
236
237Replacing older functions with the subprocess module
238====================================================
239In this section, "a ==> b" means that b can be used as a replacement
240for a.
241
242Note: All functions in this section fail (more or less) silently if
243the executed program cannot be found; this module raises an OSError
244exception.
245
246In the following examples, we assume that the subprocess module is
247imported with "from subprocess import *".
248
249
250Replacing /bin/sh shell backquote
251---------------------------------
252output=`mycmd myarg`
253==>
254output = Popen(["mycmd", "myarg"], stdout=PIPE).communicate()[0]
255
256
257Replacing shell pipe line
258-------------------------
259output=`dmesg | grep hda`
260==>
261p1 = Popen(["dmesg"], stdout=PIPE)
Peter Astrand6fdf3cb2004-11-30 18:06:42 +0000262p2 = Popen(["grep", "hda"], stdin=p1.stdout, stdout=PIPE)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000263output = p2.communicate()[0]
264
265
266Replacing os.system()
267---------------------
268sts = os.system("mycmd" + " myarg")
269==>
270p = Popen("mycmd" + " myarg", shell=True)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000271pid, sts = os.waitpid(p.pid, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000272
273Note:
274
275* Calling the program through the shell is usually not required.
276
277* It's easier to look at the returncode attribute than the
278 exitstatus.
279
280A more real-world example would look like this:
281
282try:
283 retcode = call("mycmd" + " myarg", shell=True)
284 if retcode < 0:
Guido van Rossumc2f93dc2007-05-24 00:50:02 +0000285 print("Child was terminated by signal", -retcode, file=sys.stderr)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000286 else:
Guido van Rossumc2f93dc2007-05-24 00:50:02 +0000287 print("Child returned", retcode, file=sys.stderr)
288except OSError as e:
289 print("Execution failed:", e, file=sys.stderr)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000290
291
292Replacing os.spawn*
293-------------------
Tim Peterse718f612004-10-12 21:51:32 +0000294P_NOWAIT example:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000295
296pid = os.spawnlp(os.P_NOWAIT, "/bin/mycmd", "mycmd", "myarg")
297==>
298pid = Popen(["/bin/mycmd", "myarg"]).pid
299
300
301P_WAIT example:
302
303retcode = os.spawnlp(os.P_WAIT, "/bin/mycmd", "mycmd", "myarg")
304==>
305retcode = call(["/bin/mycmd", "myarg"])
306
307
Tim Peterse718f612004-10-12 21:51:32 +0000308Vector example:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000309
310os.spawnvp(os.P_NOWAIT, path, args)
311==>
312Popen([path] + args[1:])
313
314
Tim Peterse718f612004-10-12 21:51:32 +0000315Environment example:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000316
317os.spawnlpe(os.P_NOWAIT, "/bin/mycmd", "mycmd", "myarg", env)
318==>
319Popen(["/bin/mycmd", "myarg"], env={"PATH": "/usr/bin"})
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000320"""
321
322import sys
323mswindows = (sys.platform == "win32")
324
Guido van Rossumfa0054a2007-05-24 04:05:35 +0000325import io
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000326import os
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000327import traceback
Christian Heimesfdab48e2008-01-20 09:06:41 +0000328import gc
Christian Heimesa342c012008-04-20 21:01:16 +0000329import signal
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000330
Peter Astrand454f7672005-01-01 09:36:35 +0000331# Exception classes used by this module.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000332class CalledProcessError(Exception):
Georg Brandlf9734072008-12-07 15:30:06 +0000333 """This exception is raised when a process run by check_call() or
334 check_output() returns a non-zero exit status.
335 The exit status will be stored in the returncode attribute;
336 check_output() will also store the output in the output attribute.
337 """
338 def __init__(self, returncode, cmd, output=None):
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000339 self.returncode = returncode
340 self.cmd = cmd
Georg Brandlf9734072008-12-07 15:30:06 +0000341 self.output = output
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000342 def __str__(self):
343 return "Command '%s' returned non-zero exit status %d" % (self.cmd, self.returncode)
344
Peter Astrand454f7672005-01-01 09:36:35 +0000345
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000346if mswindows:
347 import threading
348 import msvcrt
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000349 if 0: # <-- change this to use pywin32 instead of the _subprocess driver
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000350 import pywintypes
Tim Peterse8374a52004-10-13 03:15:00 +0000351 from win32api import GetStdHandle, STD_INPUT_HANDLE, \
352 STD_OUTPUT_HANDLE, STD_ERROR_HANDLE
353 from win32api import GetCurrentProcess, DuplicateHandle, \
354 GetModuleFileName, GetVersion
Peter Astrandc1d65362004-11-07 14:30:34 +0000355 from win32con import DUPLICATE_SAME_ACCESS, SW_HIDE
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000356 from win32pipe import CreatePipe
Tim Peterse8374a52004-10-13 03:15:00 +0000357 from win32process import CreateProcess, STARTUPINFO, \
358 GetExitCodeProcess, STARTF_USESTDHANDLES, \
Peter Astrandc1d65362004-11-07 14:30:34 +0000359 STARTF_USESHOWWINDOW, CREATE_NEW_CONSOLE
Christian Heimesa342c012008-04-20 21:01:16 +0000360 from win32process import TerminateProcess
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000361 from win32event import WaitForSingleObject, INFINITE, WAIT_OBJECT_0
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000362 else:
363 from _subprocess import *
364 class STARTUPINFO:
365 dwFlags = 0
366 hStdInput = None
367 hStdOutput = None
368 hStdError = None
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000369 wShowWindow = 0
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000370 class pywintypes:
371 error = IOError
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000372else:
373 import select
374 import errno
375 import fcntl
376 import pickle
377
Brett Cannona23810f2008-05-26 19:04:21 +0000378__all__ = ["Popen", "PIPE", "STDOUT", "call", "check_call", "getstatusoutput",
Georg Brandlf9734072008-12-07 15:30:06 +0000379 "getoutput", "check_output", "CalledProcessError"]
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000380
381try:
382 MAXFD = os.sysconf("SC_OPEN_MAX")
383except:
384 MAXFD = 256
385
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000386_active = []
387
388def _cleanup():
389 for inst in _active[:]:
Georg Brandl6aa2d1f2008-08-12 08:35:52 +0000390 res = inst._internal_poll(_deadstate=sys.maxsize)
Guido van Rossumb5d47ef2006-08-24 02:27:45 +0000391 if res is not None and res >= 0:
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000392 try:
393 _active.remove(inst)
394 except ValueError:
395 # This can happen if two threads create a new Popen instance.
396 # It's harmless that it was already removed, so ignore.
397 pass
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000398
399PIPE = -1
400STDOUT = -2
401
402
Peter Astrand5f5e1412004-12-05 20:15:36 +0000403def call(*popenargs, **kwargs):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000404 """Run command with arguments. Wait for command to complete, then
405 return the returncode attribute.
406
407 The arguments are the same as for the Popen constructor. Example:
408
409 retcode = call(["ls", "-l"])
410 """
Peter Astrand5f5e1412004-12-05 20:15:36 +0000411 return Popen(*popenargs, **kwargs).wait()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000412
413
Peter Astrand454f7672005-01-01 09:36:35 +0000414def check_call(*popenargs, **kwargs):
415 """Run command with arguments. Wait for command to complete. If
416 the exit code was zero then return, otherwise raise
417 CalledProcessError. The CalledProcessError object will have the
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000418 return code in the returncode attribute.
Peter Astrand454f7672005-01-01 09:36:35 +0000419
420 The arguments are the same as for the Popen constructor. Example:
421
422 check_call(["ls", "-l"])
423 """
424 retcode = call(*popenargs, **kwargs)
Peter Astrand454f7672005-01-01 09:36:35 +0000425 if retcode:
Georg Brandlf9734072008-12-07 15:30:06 +0000426 cmd = kwargs.get("args")
427 if cmd is None:
428 cmd = popenargs[0]
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000429 raise CalledProcessError(retcode, cmd)
Georg Brandlf9734072008-12-07 15:30:06 +0000430 return 0
431
432
433def check_output(*popenargs, **kwargs):
434 """Run command with arguments and return its output as a byte string.
435
436 If the exit code was non-zero it raises a CalledProcessError. The
437 CalledProcessError object will have the return code in the returncode
438 attribute and output in the output attribute.
439
440 The arguments are the same as for the Popen constructor. Example:
441
442 >>> check_output(["ls", "-l", "/dev/null"])
443 'crw-rw-rw- 1 root root 1, 3 Oct 18 2007 /dev/null\n'
444
445 The stdout argument is not allowed as it is used internally.
446 To capture standard error in the result, use stderr=subprocess.STDOUT.
447
448 >>> check_output(["/bin/sh", "-c",
449 "ls -l non_existant_file ; exit 0"],
450 stderr=subprocess.STDOUT)
451 'ls: non_existant_file: No such file or directory\n'
452 """
453 if 'stdout' in kwargs:
454 raise ValueError('stdout argument not allowed, it will be overridden.')
455 process = Popen(*popenargs, stdout=PIPE, **kwargs)
456 output, unused_err = process.communicate()
457 retcode = process.poll()
458 if retcode:
459 cmd = kwargs.get("args")
460 if cmd is None:
461 cmd = popenargs[0]
462 raise CalledProcessError(retcode, cmd, output=output)
463 return output
Peter Astrand454f7672005-01-01 09:36:35 +0000464
465
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000466def list2cmdline(seq):
467 """
468 Translate a sequence of arguments into a command line
469 string, using the same rules as the MS C runtime:
470
471 1) Arguments are delimited by white space, which is either a
472 space or a tab.
473
474 2) A string surrounded by double quotation marks is
475 interpreted as a single argument, regardless of white space
Christian Heimesfdab48e2008-01-20 09:06:41 +0000476 or pipe characters contained within. A quoted string can be
477 embedded in an argument.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000478
479 3) A double quotation mark preceded by a backslash is
480 interpreted as a literal double quotation mark.
481
482 4) Backslashes are interpreted literally, unless they
483 immediately precede a double quotation mark.
484
485 5) If backslashes immediately precede a double quotation mark,
486 every pair of backslashes is interpreted as a literal
487 backslash. If the number of backslashes is odd, the last
488 backslash escapes the next double quotation mark as
489 described in rule 3.
490 """
491
492 # See
493 # http://msdn.microsoft.com/library/en-us/vccelng/htm/progs_12.asp
494 result = []
495 needquote = False
496 for arg in seq:
497 bs_buf = []
498
499 # Add a space to separate this argument from the others
500 if result:
501 result.append(' ')
502
Christian Heimesfdab48e2008-01-20 09:06:41 +0000503 needquote = (" " in arg) or ("\t" in arg) or ("|" in arg) or not arg
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000504 if needquote:
505 result.append('"')
506
507 for c in arg:
508 if c == '\\':
Tim Peterse718f612004-10-12 21:51:32 +0000509 # Don't know if we need to double yet.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000510 bs_buf.append(c)
511 elif c == '"':
Christian Heimesfdab48e2008-01-20 09:06:41 +0000512 # Double backslashes.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000513 result.append('\\' * len(bs_buf)*2)
514 bs_buf = []
515 result.append('\\"')
516 else:
517 # Normal char
518 if bs_buf:
519 result.extend(bs_buf)
520 bs_buf = []
521 result.append(c)
522
Christian Heimesfdab48e2008-01-20 09:06:41 +0000523 # Add remaining backslashes, if any.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000524 if bs_buf:
525 result.extend(bs_buf)
526
527 if needquote:
Peter Astrand7e78ade2005-03-03 21:10:23 +0000528 result.extend(bs_buf)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000529 result.append('"')
530
531 return ''.join(result)
532
533
Brett Cannona23810f2008-05-26 19:04:21 +0000534# Various tools for executing commands and looking at their output and status.
535#
536# NB This only works (and is only relevant) for UNIX.
537
538def getstatusoutput(cmd):
539 """Return (status, output) of executing cmd in a shell.
540
541 Execute the string 'cmd' in a shell with os.popen() and return a 2-tuple
542 (status, output). cmd is actually run as '{ cmd ; } 2>&1', so that the
543 returned output will contain output or error messages. A trailing newline
544 is stripped from the output. The exit status for the command can be
545 interpreted according to the rules for the C function wait(). Example:
546
547 >>> import subprocess
548 >>> subprocess.getstatusoutput('ls /bin/ls')
549 (0, '/bin/ls')
550 >>> subprocess.getstatusoutput('cat /bin/junk')
551 (256, 'cat: /bin/junk: No such file or directory')
552 >>> subprocess.getstatusoutput('/bin/junk')
553 (256, 'sh: /bin/junk: not found')
554 """
555 pipe = os.popen('{ ' + cmd + '; } 2>&1', 'r')
556 text = pipe.read()
557 sts = pipe.close()
558 if sts is None: sts = 0
559 if text[-1:] == '\n': text = text[:-1]
560 return sts, text
561
562
563def getoutput(cmd):
564 """Return output (stdout or stderr) of executing cmd in a shell.
565
566 Like getstatusoutput(), except the exit status is ignored and the return
567 value is a string containing the command's output. Example:
568
569 >>> import subprocess
570 >>> subprocess.getoutput('ls /bin/ls')
571 '/bin/ls'
572 """
573 return getstatusoutput(cmd)[1]
574
575
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000576class Popen(object):
577 def __init__(self, args, bufsize=0, executable=None,
578 stdin=None, stdout=None, stderr=None,
579 preexec_fn=None, close_fds=False, shell=False,
580 cwd=None, env=None, universal_newlines=False,
581 startupinfo=None, creationflags=0):
582 """Create new Popen instance."""
583 _cleanup()
584
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000585 self._child_created = False
Guido van Rossum46a05a72007-06-07 21:56:45 +0000586 if bufsize is None:
587 bufsize = 0 # Restore default
Walter Dörwaldaa97f042007-05-03 21:05:51 +0000588 if not isinstance(bufsize, int):
Peter Astrand738131d2004-11-30 21:04:45 +0000589 raise TypeError("bufsize must be an integer")
590
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000591 if mswindows:
Tim Peterse8374a52004-10-13 03:15:00 +0000592 if preexec_fn is not None:
593 raise ValueError("preexec_fn is not supported on Windows "
594 "platforms")
Guido van Rossume7ba4952007-06-06 23:52:48 +0000595 if close_fds and (stdin is not None or stdout is not None or
596 stderr is not None):
Tim Peterse8374a52004-10-13 03:15:00 +0000597 raise ValueError("close_fds is not supported on Windows "
Guido van Rossume7ba4952007-06-06 23:52:48 +0000598 "platforms if you redirect stdin/stdout/stderr")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000599 else:
600 # POSIX
Tim Peterse8374a52004-10-13 03:15:00 +0000601 if startupinfo is not None:
602 raise ValueError("startupinfo is only supported on Windows "
603 "platforms")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000604 if creationflags != 0:
Tim Peterse8374a52004-10-13 03:15:00 +0000605 raise ValueError("creationflags is only supported on Windows "
606 "platforms")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000607
Tim Peterse718f612004-10-12 21:51:32 +0000608 self.stdin = None
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000609 self.stdout = None
610 self.stderr = None
611 self.pid = None
612 self.returncode = None
613 self.universal_newlines = universal_newlines
614
615 # Input and output objects. The general principle is like
616 # this:
617 #
618 # Parent Child
619 # ------ -----
620 # p2cwrite ---stdin---> p2cread
621 # c2pread <--stdout--- c2pwrite
622 # errread <--stderr--- errwrite
623 #
624 # On POSIX, the child objects are file descriptors. On
625 # Windows, these are Windows file handles. The parent objects
626 # are file descriptors on both platforms. The parent objects
627 # are None when not using PIPEs. The child objects are None
628 # when not redirecting.
Tim Peterse718f612004-10-12 21:51:32 +0000629
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000630 (p2cread, p2cwrite,
631 c2pread, c2pwrite,
632 errread, errwrite) = self._get_handles(stdin, stdout, stderr)
633
634 self._execute_child(args, executable, preexec_fn, close_fds,
635 cwd, env, universal_newlines,
636 startupinfo, creationflags, shell,
637 p2cread, p2cwrite,
638 c2pread, c2pwrite,
639 errread, errwrite)
640
Thomas Wouterscf297e42007-02-23 15:07:44 +0000641 # On Windows, you cannot just redirect one or two handles: You
642 # either have to redirect all three or none. If the subprocess
643 # user has only redirected one or two handles, we are
644 # automatically creating PIPEs for the rest. We should close
Guido van Rossumd8faa362007-04-27 19:54:29 +0000645 # these after the process is started. See bug #1124861.
Thomas Wouterscf297e42007-02-23 15:07:44 +0000646 if mswindows:
647 if stdin is None and p2cwrite is not None:
648 os.close(p2cwrite)
649 p2cwrite = None
650 if stdout is None and c2pread is not None:
651 os.close(c2pread)
652 c2pread = None
653 if stderr is None and errread is not None:
654 os.close(errread)
655 errread = None
656
Guido van Rossumc2f93dc2007-05-24 00:50:02 +0000657 if bufsize == 0:
658 bufsize = 1 # Nearly unbuffered (XXX for now)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000659 if p2cwrite is not None:
Guido van Rossumfa0054a2007-05-24 04:05:35 +0000660 self.stdin = io.open(p2cwrite, 'wb', bufsize)
661 if self.universal_newlines:
662 self.stdin = io.TextIOWrapper(self.stdin)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000663 if c2pread is not None:
Guido van Rossumfa0054a2007-05-24 04:05:35 +0000664 self.stdout = io.open(c2pread, 'rb', bufsize)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000665 if universal_newlines:
Guido van Rossumfa0054a2007-05-24 04:05:35 +0000666 self.stdout = io.TextIOWrapper(self.stdout)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000667 if errread is not None:
Guido van Rossumfa0054a2007-05-24 04:05:35 +0000668 self.stderr = io.open(errread, 'rb', bufsize)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000669 if universal_newlines:
Guido van Rossumfa0054a2007-05-24 04:05:35 +0000670 self.stderr = io.TextIOWrapper(self.stderr)
Tim Peterse718f612004-10-12 21:51:32 +0000671
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000672
Guido van Rossum98297ee2007-11-06 21:34:58 +0000673 def _translate_newlines(self, data, encoding):
674 data = data.replace(b"\r\n", b"\n").replace(b"\r", b"\n")
675 return data.decode(encoding)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000676
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000677
Guido van Rossumd8faa362007-04-27 19:54:29 +0000678 def __del__(self, sys=sys):
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000679 if not self._child_created:
680 # We didn't get to successfully create a child process.
681 return
682 # In case the child hasn't been waited on, check if it's done.
Georg Brandl6aa2d1f2008-08-12 08:35:52 +0000683 self._internal_poll(_deadstate=sys.maxsize)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000684 if self.returncode is None and _active is not None:
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000685 # Child is still running, keep us alive until we can wait on it.
686 _active.append(self)
687
688
Peter Astrand23109f02005-03-03 20:28:59 +0000689 def communicate(self, input=None):
690 """Interact with process: Send data to stdin. Read data from
691 stdout and stderr, until end-of-file is reached. Wait for
692 process to terminate. The optional input argument should be a
693 string to be sent to the child process, or None, if no data
694 should be sent to the child.
Tim Peterseba28be2005-03-28 01:08:02 +0000695
Peter Astrand23109f02005-03-03 20:28:59 +0000696 communicate() returns a tuple (stdout, stderr)."""
697
698 # Optimization: If we are only using one pipe, or no pipe at
699 # all, using select() or threads is unnecessary.
700 if [self.stdin, self.stdout, self.stderr].count(None) >= 2:
Tim Peterseba28be2005-03-28 01:08:02 +0000701 stdout = None
702 stderr = None
Peter Astrand23109f02005-03-03 20:28:59 +0000703 if self.stdin:
704 if input:
705 self.stdin.write(input)
706 self.stdin.close()
707 elif self.stdout:
708 stdout = self.stdout.read()
Georg Brandlf08a9dd2008-06-10 16:57:31 +0000709 self.stdout.close()
Peter Astrand23109f02005-03-03 20:28:59 +0000710 elif self.stderr:
711 stderr = self.stderr.read()
Georg Brandlf08a9dd2008-06-10 16:57:31 +0000712 self.stderr.close()
Peter Astrand23109f02005-03-03 20:28:59 +0000713 self.wait()
714 return (stdout, stderr)
Tim Peterseba28be2005-03-28 01:08:02 +0000715
Peter Astrand23109f02005-03-03 20:28:59 +0000716 return self._communicate(input)
717
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000718
Georg Brandl6aa2d1f2008-08-12 08:35:52 +0000719 def poll(self):
720 return self._internal_poll()
721
722
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000723 if mswindows:
724 #
725 # Windows methods
726 #
727 def _get_handles(self, stdin, stdout, stderr):
728 """Construct and return tupel with IO objects:
729 p2cread, p2cwrite, c2pread, c2pwrite, errread, errwrite
730 """
Peter Astrandd38ddf42005-02-10 08:32:50 +0000731 if stdin is None and stdout is None and stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000732 return (None, None, None, None, None, None)
Tim Peterse718f612004-10-12 21:51:32 +0000733
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000734 p2cread, p2cwrite = None, None
735 c2pread, c2pwrite = None, None
736 errread, errwrite = None, None
737
Peter Astrandd38ddf42005-02-10 08:32:50 +0000738 if stdin is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000739 p2cread = GetStdHandle(STD_INPUT_HANDLE)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000740 if p2cread is not None:
741 pass
742 elif stdin is None or stdin == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000743 p2cread, p2cwrite = CreatePipe(None, 0)
744 # Detach and turn into fd
745 p2cwrite = p2cwrite.Detach()
746 p2cwrite = msvcrt.open_osfhandle(p2cwrite, 0)
Peter Astrandd38ddf42005-02-10 08:32:50 +0000747 elif isinstance(stdin, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000748 p2cread = msvcrt.get_osfhandle(stdin)
749 else:
750 # Assuming file-like object
751 p2cread = msvcrt.get_osfhandle(stdin.fileno())
752 p2cread = self._make_inheritable(p2cread)
753
Peter Astrandd38ddf42005-02-10 08:32:50 +0000754 if stdout is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000755 c2pwrite = GetStdHandle(STD_OUTPUT_HANDLE)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000756 if c2pwrite is not None:
757 pass
758 elif stdout is None or stdout == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000759 c2pread, c2pwrite = CreatePipe(None, 0)
760 # Detach and turn into fd
761 c2pread = c2pread.Detach()
762 c2pread = msvcrt.open_osfhandle(c2pread, 0)
Peter Astrandd38ddf42005-02-10 08:32:50 +0000763 elif isinstance(stdout, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000764 c2pwrite = msvcrt.get_osfhandle(stdout)
765 else:
766 # Assuming file-like object
767 c2pwrite = msvcrt.get_osfhandle(stdout.fileno())
768 c2pwrite = self._make_inheritable(c2pwrite)
769
Peter Astrandd38ddf42005-02-10 08:32:50 +0000770 if stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000771 errwrite = GetStdHandle(STD_ERROR_HANDLE)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000772 if errwrite is not None:
773 pass
774 elif stderr is None or stderr == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000775 errread, errwrite = CreatePipe(None, 0)
776 # Detach and turn into fd
777 errread = errread.Detach()
778 errread = msvcrt.open_osfhandle(errread, 0)
779 elif stderr == STDOUT:
780 errwrite = c2pwrite
Peter Astrandd38ddf42005-02-10 08:32:50 +0000781 elif isinstance(stderr, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000782 errwrite = msvcrt.get_osfhandle(stderr)
783 else:
784 # Assuming file-like object
785 errwrite = msvcrt.get_osfhandle(stderr.fileno())
786 errwrite = self._make_inheritable(errwrite)
787
788 return (p2cread, p2cwrite,
789 c2pread, c2pwrite,
790 errread, errwrite)
791
792
793 def _make_inheritable(self, handle):
794 """Return a duplicate of handle, which is inheritable"""
795 return DuplicateHandle(GetCurrentProcess(), handle,
796 GetCurrentProcess(), 0, 1,
797 DUPLICATE_SAME_ACCESS)
798
799
800 def _find_w9xpopen(self):
801 """Find and return absolut path to w9xpopen.exe"""
Tim Peterse8374a52004-10-13 03:15:00 +0000802 w9xpopen = os.path.join(os.path.dirname(GetModuleFileName(0)),
803 "w9xpopen.exe")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000804 if not os.path.exists(w9xpopen):
805 # Eeek - file-not-found - possibly an embedding
806 # situation - see if we can locate it in sys.exec_prefix
Tim Peterse8374a52004-10-13 03:15:00 +0000807 w9xpopen = os.path.join(os.path.dirname(sys.exec_prefix),
808 "w9xpopen.exe")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000809 if not os.path.exists(w9xpopen):
Tim Peterse8374a52004-10-13 03:15:00 +0000810 raise RuntimeError("Cannot locate w9xpopen.exe, which is "
811 "needed for Popen to work with your "
812 "shell or platform.")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000813 return w9xpopen
814
Tim Peterse718f612004-10-12 21:51:32 +0000815
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000816 def _execute_child(self, args, executable, preexec_fn, close_fds,
817 cwd, env, universal_newlines,
818 startupinfo, creationflags, shell,
819 p2cread, p2cwrite,
820 c2pread, c2pwrite,
821 errread, errwrite):
822 """Execute program (MS Windows version)"""
823
Guido van Rossum3172c5d2007-10-16 18:12:55 +0000824 if not isinstance(args, str):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000825 args = list2cmdline(args)
826
Peter Astrandc1d65362004-11-07 14:30:34 +0000827 # Process startup details
Peter Astrandd38ddf42005-02-10 08:32:50 +0000828 if startupinfo is None:
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000829 startupinfo = STARTUPINFO()
830 if None not in (p2cread, c2pwrite, errwrite):
Peter Astrandc1d65362004-11-07 14:30:34 +0000831 startupinfo.dwFlags |= STARTF_USESTDHANDLES
832 startupinfo.hStdInput = p2cread
833 startupinfo.hStdOutput = c2pwrite
834 startupinfo.hStdError = errwrite
835
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000836 if shell:
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000837 startupinfo.dwFlags |= STARTF_USESHOWWINDOW
838 startupinfo.wShowWindow = SW_HIDE
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000839 comspec = os.environ.get("COMSPEC", "cmd.exe")
840 args = comspec + " /c " + args
Guido van Rossume2a383d2007-01-15 16:59:06 +0000841 if (GetVersion() >= 0x80000000 or
Tim Peterse8374a52004-10-13 03:15:00 +0000842 os.path.basename(comspec).lower() == "command.com"):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000843 # Win9x, or using command.com on NT. We need to
844 # use the w9xpopen intermediate program. For more
845 # information, see KB Q150956
846 # (http://web.archive.org/web/20011105084002/http://support.microsoft.com/support/kb/articles/Q150/9/56.asp)
847 w9xpopen = self._find_w9xpopen()
848 args = '"%s" %s' % (w9xpopen, args)
849 # Not passing CREATE_NEW_CONSOLE has been known to
850 # cause random failures on win9x. Specifically a
851 # dialog: "Your program accessed mem currently in
852 # use at xxx" and a hopeful warning about the
853 # stability of your system. Cost is Ctrl+C wont
854 # kill children.
855 creationflags |= CREATE_NEW_CONSOLE
856
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000857 # Start the process
858 try:
859 hp, ht, pid, tid = CreateProcess(executable, args,
Tim Peterse8374a52004-10-13 03:15:00 +0000860 # no special security
861 None, None,
Guido van Rossume7ba4952007-06-06 23:52:48 +0000862 int(not close_fds),
Tim Peterse8374a52004-10-13 03:15:00 +0000863 creationflags,
864 env,
865 cwd,
866 startupinfo)
Guido van Rossumb940e112007-01-10 16:19:56 +0000867 except pywintypes.error as e:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000868 # Translate pywintypes.error to WindowsError, which is
869 # a subclass of OSError. FIXME: We should really
870 # translate errno using _sys_errlist (or simliar), but
871 # how can this be done from Python?
872 raise WindowsError(*e.args)
873
874 # Retain the process handle, but close the thread handle
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000875 self._child_created = True
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000876 self._handle = hp
877 self.pid = pid
878 ht.Close()
879
Andrew M. Kuchling51ee66e2004-10-12 16:38:42 +0000880 # Child is launched. Close the parent's copy of those pipe
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000881 # handles that only the child should have open. You need
882 # to make sure that no handles to the write end of the
883 # output pipe are maintained in this process or else the
884 # pipe will not close when the child process exits and the
885 # ReadFile will hang.
Peter Astrandd38ddf42005-02-10 08:32:50 +0000886 if p2cread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000887 p2cread.Close()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000888 if c2pwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000889 c2pwrite.Close()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000890 if errwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000891 errwrite.Close()
892
Tim Peterse718f612004-10-12 21:51:32 +0000893
Georg Brandl6aa2d1f2008-08-12 08:35:52 +0000894 def _internal_poll(self, _deadstate=None):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000895 """Check if child process has terminated. Returns returncode
896 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +0000897 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000898 if WaitForSingleObject(self._handle, 0) == WAIT_OBJECT_0:
899 self.returncode = GetExitCodeProcess(self._handle)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000900 return self.returncode
901
902
903 def wait(self):
904 """Wait for child process to terminate. Returns returncode
905 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +0000906 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000907 obj = WaitForSingleObject(self._handle, INFINITE)
908 self.returncode = GetExitCodeProcess(self._handle)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000909 return self.returncode
910
911
912 def _readerthread(self, fh, buffer):
913 buffer.append(fh.read())
914
915
Peter Astrand23109f02005-03-03 20:28:59 +0000916 def _communicate(self, input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000917 stdout = None # Return
918 stderr = None # Return
919
920 if self.stdout:
921 stdout = []
Tim Peterse8374a52004-10-13 03:15:00 +0000922 stdout_thread = threading.Thread(target=self._readerthread,
923 args=(self.stdout, stdout))
Benjamin Peterson632e0362008-08-18 19:08:51 +0000924 stdout_thread.daemon = True
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000925 stdout_thread.start()
926 if self.stderr:
927 stderr = []
Tim Peterse8374a52004-10-13 03:15:00 +0000928 stderr_thread = threading.Thread(target=self._readerthread,
929 args=(self.stderr, stderr))
Benjamin Peterson632e0362008-08-18 19:08:51 +0000930 stderr_thread.daemon = True
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000931 stderr_thread.start()
932
933 if self.stdin:
Peter Astrandd38ddf42005-02-10 08:32:50 +0000934 if input is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000935 self.stdin.write(input)
936 self.stdin.close()
937
938 if self.stdout:
939 stdout_thread.join()
940 if self.stderr:
941 stderr_thread.join()
942
943 # All data exchanged. Translate lists into strings.
Peter Astrandd38ddf42005-02-10 08:32:50 +0000944 if stdout is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000945 stdout = stdout[0]
Peter Astrandd38ddf42005-02-10 08:32:50 +0000946 if stderr is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000947 stderr = stderr[0]
948
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000949 self.wait()
950 return (stdout, stderr)
951
Christian Heimesa342c012008-04-20 21:01:16 +0000952 def send_signal(self, sig):
953 """Send a signal to the process
954 """
955 if sig == signal.SIGTERM:
956 self.terminate()
957 else:
958 raise ValueError("Only SIGTERM is supported on Windows")
959
960 def terminate(self):
961 """Terminates the process
962 """
963 TerminateProcess(self._handle, 1)
964
965 kill = terminate
966
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000967 else:
968 #
969 # POSIX methods
970 #
971 def _get_handles(self, stdin, stdout, stderr):
972 """Construct and return tupel with IO objects:
973 p2cread, p2cwrite, c2pread, c2pwrite, errread, errwrite
974 """
975 p2cread, p2cwrite = None, None
976 c2pread, c2pwrite = None, None
977 errread, errwrite = None, None
978
Peter Astrandd38ddf42005-02-10 08:32:50 +0000979 if stdin is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000980 pass
981 elif stdin == PIPE:
982 p2cread, p2cwrite = os.pipe()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000983 elif isinstance(stdin, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000984 p2cread = stdin
985 else:
986 # Assuming file-like object
987 p2cread = stdin.fileno()
988
Peter Astrandd38ddf42005-02-10 08:32:50 +0000989 if stdout is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000990 pass
991 elif stdout == PIPE:
992 c2pread, c2pwrite = os.pipe()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000993 elif isinstance(stdout, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000994 c2pwrite = stdout
995 else:
996 # Assuming file-like object
Tim Peterse718f612004-10-12 21:51:32 +0000997 c2pwrite = stdout.fileno()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000998
Peter Astrandd38ddf42005-02-10 08:32:50 +0000999 if stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001000 pass
1001 elif stderr == PIPE:
1002 errread, errwrite = os.pipe()
1003 elif stderr == STDOUT:
1004 errwrite = c2pwrite
Peter Astrandd38ddf42005-02-10 08:32:50 +00001005 elif isinstance(stderr, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001006 errwrite = stderr
1007 else:
1008 # Assuming file-like object
1009 errwrite = stderr.fileno()
1010
1011 return (p2cread, p2cwrite,
1012 c2pread, c2pwrite,
1013 errread, errwrite)
1014
1015
1016 def _set_cloexec_flag(self, fd):
1017 try:
1018 cloexec_flag = fcntl.FD_CLOEXEC
1019 except AttributeError:
1020 cloexec_flag = 1
1021
1022 old = fcntl.fcntl(fd, fcntl.F_GETFD)
1023 fcntl.fcntl(fd, fcntl.F_SETFD, old | cloexec_flag)
1024
1025
1026 def _close_fds(self, but):
Christian Heimesfdab48e2008-01-20 09:06:41 +00001027 os.closerange(3, but)
1028 os.closerange(but + 1, MAXFD)
Tim Peterse718f612004-10-12 21:51:32 +00001029
1030
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001031 def _execute_child(self, args, executable, preexec_fn, close_fds,
1032 cwd, env, universal_newlines,
1033 startupinfo, creationflags, shell,
1034 p2cread, p2cwrite,
1035 c2pread, c2pwrite,
1036 errread, errwrite):
1037 """Execute program (POSIX version)"""
1038
Guido van Rossum3172c5d2007-10-16 18:12:55 +00001039 if isinstance(args, str):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001040 args = [args]
Thomas Wouters89f507f2006-12-13 04:49:30 +00001041 else:
1042 args = list(args)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001043
1044 if shell:
1045 args = ["/bin/sh", "-c"] + args
1046
Peter Astrandd38ddf42005-02-10 08:32:50 +00001047 if executable is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001048 executable = args[0]
1049
1050 # For transferring possible exec failure from child to parent
1051 # The first char specifies the exception type: 0 means
1052 # OSError, 1 means some other error.
1053 errpipe_read, errpipe_write = os.pipe()
1054 self._set_cloexec_flag(errpipe_write)
1055
Christian Heimesfdab48e2008-01-20 09:06:41 +00001056 gc_was_enabled = gc.isenabled()
1057 # Disable gc to avoid bug where gc -> file_dealloc ->
1058 # write to stderr -> hang. http://bugs.python.org/issue1336
1059 gc.disable()
1060 try:
1061 self.pid = os.fork()
1062 except:
1063 if gc_was_enabled:
1064 gc.enable()
1065 raise
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001066 self._child_created = True
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001067 if self.pid == 0:
1068 # Child
1069 try:
1070 # Close parent's pipe ends
Thomas Wouterscf297e42007-02-23 15:07:44 +00001071 if p2cwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001072 os.close(p2cwrite)
Thomas Wouterscf297e42007-02-23 15:07:44 +00001073 if c2pread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001074 os.close(c2pread)
Thomas Wouterscf297e42007-02-23 15:07:44 +00001075 if errread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001076 os.close(errread)
1077 os.close(errpipe_read)
1078
1079 # Dup fds for child
Thomas Wouterscf297e42007-02-23 15:07:44 +00001080 if p2cread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001081 os.dup2(p2cread, 0)
Thomas Wouterscf297e42007-02-23 15:07:44 +00001082 if c2pwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001083 os.dup2(c2pwrite, 1)
Thomas Wouterscf297e42007-02-23 15:07:44 +00001084 if errwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001085 os.dup2(errwrite, 2)
1086
Thomas Wouters89f507f2006-12-13 04:49:30 +00001087 # Close pipe fds. Make sure we don't close the same
1088 # fd more than once, or standard fds.
Thomas Wouterscf297e42007-02-23 15:07:44 +00001089 if p2cread is not None and p2cread not in (0,):
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001090 os.close(p2cread)
Thomas Wouterscf297e42007-02-23 15:07:44 +00001091 if c2pwrite is not None and c2pwrite not in (p2cread, 1):
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001092 os.close(c2pwrite)
Guido van Rossum98297ee2007-11-06 21:34:58 +00001093 if (errwrite is not None and
1094 errwrite not in (p2cread, c2pwrite, 2)):
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001095 os.close(errwrite)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001096
1097 # Close all other fds, if asked for
1098 if close_fds:
1099 self._close_fds(but=errpipe_write)
1100
Peter Astrandd38ddf42005-02-10 08:32:50 +00001101 if cwd is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001102 os.chdir(cwd)
1103
1104 if preexec_fn:
Neal Norwitzd9108552006-03-17 08:00:19 +00001105 preexec_fn()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001106
Peter Astrandd38ddf42005-02-10 08:32:50 +00001107 if env is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001108 os.execvp(executable, args)
1109 else:
1110 os.execvpe(executable, args, env)
1111
1112 except:
1113 exc_type, exc_value, tb = sys.exc_info()
1114 # Save the traceback and attach it to the exception object
Tim Peterse8374a52004-10-13 03:15:00 +00001115 exc_lines = traceback.format_exception(exc_type,
1116 exc_value,
1117 tb)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001118 exc_value.child_traceback = ''.join(exc_lines)
1119 os.write(errpipe_write, pickle.dumps(exc_value))
1120
1121 # This exitcode won't be reported to applications, so it
1122 # really doesn't matter what we return.
1123 os._exit(255)
1124
1125 # Parent
Christian Heimesfdab48e2008-01-20 09:06:41 +00001126 if gc_was_enabled:
1127 gc.enable()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001128 os.close(errpipe_write)
Thomas Wouterscf297e42007-02-23 15:07:44 +00001129 if p2cread is not None and p2cwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001130 os.close(p2cread)
Thomas Wouterscf297e42007-02-23 15:07:44 +00001131 if c2pwrite is not None and c2pread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001132 os.close(c2pwrite)
Thomas Wouterscf297e42007-02-23 15:07:44 +00001133 if errwrite is not None and errread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001134 os.close(errwrite)
1135
1136 # Wait for exec to fail or succeed; possibly raising exception
1137 data = os.read(errpipe_read, 1048576) # Exceptions limited to 1 MB
1138 os.close(errpipe_read)
Guido van Rossumaf2362a2007-05-15 22:32:02 +00001139 if data:
Peter Astrandf791d7a2005-01-01 09:38:57 +00001140 os.waitpid(self.pid, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001141 child_exception = pickle.loads(data)
Benjamin Petersond75fcb42009-02-19 04:22:03 +00001142 for fd in (p2cwrite, c2pread, errread):
1143 if fd is not None:
1144 os.close(fd)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001145 raise child_exception
1146
1147
1148 def _handle_exitstatus(self, sts):
1149 if os.WIFSIGNALED(sts):
1150 self.returncode = -os.WTERMSIG(sts)
1151 elif os.WIFEXITED(sts):
1152 self.returncode = os.WEXITSTATUS(sts)
1153 else:
1154 # Should never happen
1155 raise RuntimeError("Unknown child exit status!")
1156
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001157
Georg Brandl6aa2d1f2008-08-12 08:35:52 +00001158 def _internal_poll(self, _deadstate=None):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001159 """Check if child process has terminated. Returns returncode
1160 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +00001161 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001162 try:
1163 pid, sts = os.waitpid(self.pid, os.WNOHANG)
1164 if pid == self.pid:
1165 self._handle_exitstatus(sts)
1166 except os.error:
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001167 if _deadstate is not None:
1168 self.returncode = _deadstate
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001169 return self.returncode
1170
1171
1172 def wait(self):
1173 """Wait for child process to terminate. Returns returncode
1174 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +00001175 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001176 pid, sts = os.waitpid(self.pid, 0)
1177 self._handle_exitstatus(sts)
1178 return self.returncode
1179
1180
Peter Astrand23109f02005-03-03 20:28:59 +00001181 def _communicate(self, input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001182 read_set = []
1183 write_set = []
1184 stdout = None # Return
1185 stderr = None # Return
1186
1187 if self.stdin:
1188 # Flush stdio buffer. This might block, if the user has
1189 # been writing to .stdin in an uncontrolled fashion.
1190 self.stdin.flush()
1191 if input:
1192 write_set.append(self.stdin)
1193 else:
1194 self.stdin.close()
1195 if self.stdout:
1196 read_set.append(self.stdout)
Tim Peterse718f612004-10-12 21:51:32 +00001197 stdout = []
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001198 if self.stderr:
1199 read_set.append(self.stderr)
1200 stderr = []
1201
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001202 input_offset = 0
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001203 while read_set or write_set:
Georg Brandl86b2fb92008-07-16 03:43:04 +00001204 try:
1205 rlist, wlist, xlist = select.select(read_set, write_set, [])
1206 except select.error as e:
1207 if e.args[0] == errno.EINTR:
1208 continue
1209 raise
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001210
Guido van Rossum98297ee2007-11-06 21:34:58 +00001211 # XXX Rewrite these to use non-blocking I/O on the
1212 # file objects; they are no longer using C stdio!
1213
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001214 if self.stdin in wlist:
1215 # When select has indicated that the file is writable,
1216 # we can write up to PIPE_BUF bytes without risk
1217 # blocking. POSIX defines PIPE_BUF >= 512
Guido van Rossumbae07c92007-10-08 02:46:15 +00001218 chunk = input[input_offset : input_offset + 512]
1219 bytes_written = os.write(self.stdin.fileno(), chunk)
Thomas Wouters9fe394c2007-02-05 01:24:16 +00001220 input_offset += bytes_written
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001221 if input_offset >= len(input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001222 self.stdin.close()
1223 write_set.remove(self.stdin)
1224
1225 if self.stdout in rlist:
1226 data = os.read(self.stdout.fileno(), 1024)
Guido van Rossumc9e363c2007-05-15 23:18:55 +00001227 if not data:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001228 self.stdout.close()
1229 read_set.remove(self.stdout)
1230 stdout.append(data)
1231
1232 if self.stderr in rlist:
1233 data = os.read(self.stderr.fileno(), 1024)
Guido van Rossumc9e363c2007-05-15 23:18:55 +00001234 if not data:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001235 self.stderr.close()
1236 read_set.remove(self.stderr)
1237 stderr.append(data)
1238
1239 # All data exchanged. Translate lists into strings.
Peter Astrandd38ddf42005-02-10 08:32:50 +00001240 if stdout is not None:
Guido van Rossum98297ee2007-11-06 21:34:58 +00001241 stdout = b"".join(stdout)
Peter Astrandd38ddf42005-02-10 08:32:50 +00001242 if stderr is not None:
Guido van Rossum98297ee2007-11-06 21:34:58 +00001243 stderr = b"".join(stderr)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001244
Guido van Rossum98297ee2007-11-06 21:34:58 +00001245 # Translate newlines, if requested.
1246 # This also turns bytes into strings.
Guido van Rossumc9e363c2007-05-15 23:18:55 +00001247 if self.universal_newlines:
Guido van Rossumfa0054a2007-05-24 04:05:35 +00001248 if stdout is not None:
Guido van Rossum98297ee2007-11-06 21:34:58 +00001249 stdout = self._translate_newlines(stdout,
1250 self.stdout.encoding)
Guido van Rossumfa0054a2007-05-24 04:05:35 +00001251 if stderr is not None:
Guido van Rossum98297ee2007-11-06 21:34:58 +00001252 stderr = self._translate_newlines(stderr,
1253 self.stderr.encoding)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001254
1255 self.wait()
1256 return (stdout, stderr)
1257
Christian Heimesa342c012008-04-20 21:01:16 +00001258 def send_signal(self, sig):
1259 """Send a signal to the process
1260 """
1261 os.kill(self.pid, sig)
1262
1263 def terminate(self):
1264 """Terminate the process with SIGTERM
1265 """
1266 self.send_signal(signal.SIGTERM)
1267
1268 def kill(self):
1269 """Kill the process with SIGKILL
1270 """
1271 self.send_signal(signal.SIGKILL)
1272
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001273
1274def _demo_posix():
1275 #
1276 # Example 1: Simple redirection: Get process list
1277 #
1278 plist = Popen(["ps"], stdout=PIPE).communicate()[0]
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001279 print("Process list:")
1280 print(plist)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001281
1282 #
1283 # Example 2: Change uid before executing child
1284 #
1285 if os.getuid() == 0:
1286 p = Popen(["id"], preexec_fn=lambda: os.setuid(100))
1287 p.wait()
1288
1289 #
1290 # Example 3: Connecting several subprocesses
1291 #
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001292 print("Looking for 'hda'...")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001293 p1 = Popen(["dmesg"], stdout=PIPE)
1294 p2 = Popen(["grep", "hda"], stdin=p1.stdout, stdout=PIPE)
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001295 print(repr(p2.communicate()[0]))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001296
1297 #
1298 # Example 4: Catch execution error
1299 #
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001300 print()
1301 print("Trying a weird file...")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001302 try:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001303 print(Popen(["/this/path/does/not/exist"]).communicate())
Guido van Rossumb940e112007-01-10 16:19:56 +00001304 except OSError as e:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001305 if e.errno == errno.ENOENT:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001306 print("The file didn't exist. I thought so...")
1307 print("Child traceback:")
1308 print(e.child_traceback)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001309 else:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001310 print("Error", e.errno)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001311 else:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001312 print("Gosh. No error.", file=sys.stderr)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001313
1314
1315def _demo_windows():
1316 #
1317 # Example 1: Connecting several subprocesses
1318 #
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001319 print("Looking for 'PROMPT' in set output...")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001320 p1 = Popen("set", stdout=PIPE, shell=True)
1321 p2 = Popen('find "PROMPT"', stdin=p1.stdout, stdout=PIPE)
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001322 print(repr(p2.communicate()[0]))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001323
1324 #
1325 # Example 2: Simple execution of program
1326 #
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001327 print("Executing calc...")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001328 p = Popen("calc")
1329 p.wait()
1330
1331
1332if __name__ == "__main__":
1333 if mswindows:
1334 _demo_windows()
1335 else:
1336 _demo_posix()