blob: 6e753cb7126d5232e9d8f2ca00d4c972eaef3b02 [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
Georg Brandl2708f3a2009-12-20 14:38:23 +0000113 >>> retcode = call(["ls", "-l"])
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000114
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
Georg Brandl2708f3a2009-12-20 14:38:23 +0000123 >>> check_call(["ls", "-l"])
124 0
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000125
Brett Cannona23810f2008-05-26 19:04:21 +0000126getstatusoutput(cmd):
127 Return (status, output) of executing cmd in a shell.
128
129 Execute the string 'cmd' in a shell with os.popen() and return a 2-tuple
130 (status, output). cmd is actually run as '{ cmd ; } 2>&1', so that the
131 returned output will contain output or error messages. A trailing newline
132 is stripped from the output. The exit status for the command can be
133 interpreted according to the rules for the C function wait(). Example:
134
Brett Cannona23810f2008-05-26 19:04:21 +0000135 >>> 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
Brett Cannona23810f2008-05-26 19:04:21 +0000148 >>> subprocess.getoutput('ls /bin/ls')
149 '/bin/ls'
150
Georg Brandlf9734072008-12-07 15:30:06 +0000151check_output(*popenargs, **kwargs):
Georg Brandl2708f3a2009-12-20 14:38:23 +0000152 Run command with arguments and return its output as a byte string.
Georg Brandlf9734072008-12-07 15:30:06 +0000153
Georg Brandl2708f3a2009-12-20 14:38:23 +0000154 If the exit code was non-zero it raises a CalledProcessError. The
155 CalledProcessError object will have the return code in the returncode
156 attribute and output in the output attribute.
Georg Brandlf9734072008-12-07 15:30:06 +0000157
Georg Brandl2708f3a2009-12-20 14:38:23 +0000158 The arguments are the same as for the Popen constructor. Example:
Georg Brandlf9734072008-12-07 15:30:06 +0000159
Georg Brandl2708f3a2009-12-20 14:38:23 +0000160 >>> output = subprocess.check_output(["ls", "-l", "/dev/null"])
Georg Brandlf9734072008-12-07 15:30:06 +0000161
Brett Cannona23810f2008-05-26 19:04:21 +0000162
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000163Exceptions
164----------
165Exceptions raised in the child process, before the new program has
166started to execute, will be re-raised in the parent. Additionally,
167the exception object will have one extra attribute called
168'child_traceback', which is a string containing traceback information
169from the childs point of view.
170
171The most common exception raised is OSError. This occurs, for
172example, when trying to execute a non-existent file. Applications
173should prepare for OSErrors.
174
175A ValueError will be raised if Popen is called with invalid arguments.
176
Georg Brandlf9734072008-12-07 15:30:06 +0000177check_call() and check_output() will raise CalledProcessError, if the
178called process returns a non-zero return code.
Peter Astrand454f7672005-01-01 09:36:35 +0000179
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000180
181Security
182--------
183Unlike some other popen functions, this implementation will never call
184/bin/sh implicitly. This means that all characters, including shell
185metacharacters, can safely be passed to child processes.
186
187
188Popen objects
189=============
190Instances of the Popen class have the following methods:
191
192poll()
193 Check if child process has terminated. Returns returncode
194 attribute.
195
196wait()
197 Wait for child process to terminate. Returns returncode attribute.
198
199communicate(input=None)
200 Interact with process: Send data to stdin. Read data from stdout
201 and stderr, until end-of-file is reached. Wait for process to
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000202 terminate. The optional input argument should be a string to be
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000203 sent to the child process, or None, if no data should be sent to
204 the child.
Tim Peterse718f612004-10-12 21:51:32 +0000205
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000206 communicate() returns a tuple (stdout, stderr).
207
208 Note: The data read is buffered in memory, so do not use this
209 method if the data size is large or unlimited.
210
211The following attributes are also available:
212
213stdin
214 If the stdin argument is PIPE, this attribute is a file object
215 that provides input to the child process. Otherwise, it is None.
216
217stdout
218 If the stdout argument is PIPE, this attribute is a file object
219 that provides output from the child process. Otherwise, it is
220 None.
221
222stderr
223 If the stderr argument is PIPE, this attribute is file object that
224 provides error output from the child process. Otherwise, it is
225 None.
226
227pid
228 The process ID of the child process.
229
230returncode
231 The child return code. A None value indicates that the process
232 hasn't terminated yet. A negative value -N indicates that the
233 child was terminated by signal N (UNIX only).
234
235
236Replacing older functions with the subprocess module
237====================================================
238In this section, "a ==> b" means that b can be used as a replacement
239for a.
240
241Note: All functions in this section fail (more or less) silently if
242the executed program cannot be found; this module raises an OSError
243exception.
244
245In the following examples, we assume that the subprocess module is
246imported with "from subprocess import *".
247
248
249Replacing /bin/sh shell backquote
250---------------------------------
251output=`mycmd myarg`
252==>
253output = Popen(["mycmd", "myarg"], stdout=PIPE).communicate()[0]
254
255
256Replacing shell pipe line
257-------------------------
258output=`dmesg | grep hda`
259==>
260p1 = Popen(["dmesg"], stdout=PIPE)
Peter Astrand6fdf3cb2004-11-30 18:06:42 +0000261p2 = Popen(["grep", "hda"], stdin=p1.stdout, stdout=PIPE)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000262output = p2.communicate()[0]
263
264
265Replacing os.system()
266---------------------
267sts = os.system("mycmd" + " myarg")
268==>
269p = Popen("mycmd" + " myarg", shell=True)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000270pid, sts = os.waitpid(p.pid, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000271
272Note:
273
274* Calling the program through the shell is usually not required.
275
276* It's easier to look at the returncode attribute than the
277 exitstatus.
278
279A more real-world example would look like this:
280
281try:
282 retcode = call("mycmd" + " myarg", shell=True)
283 if retcode < 0:
Guido van Rossumc2f93dc2007-05-24 00:50:02 +0000284 print("Child was terminated by signal", -retcode, file=sys.stderr)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000285 else:
Guido van Rossumc2f93dc2007-05-24 00:50:02 +0000286 print("Child returned", retcode, file=sys.stderr)
287except OSError as e:
288 print("Execution failed:", e, file=sys.stderr)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000289
290
291Replacing os.spawn*
292-------------------
Tim Peterse718f612004-10-12 21:51:32 +0000293P_NOWAIT example:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000294
295pid = os.spawnlp(os.P_NOWAIT, "/bin/mycmd", "mycmd", "myarg")
296==>
297pid = Popen(["/bin/mycmd", "myarg"]).pid
298
299
300P_WAIT example:
301
302retcode = os.spawnlp(os.P_WAIT, "/bin/mycmd", "mycmd", "myarg")
303==>
304retcode = call(["/bin/mycmd", "myarg"])
305
306
Tim Peterse718f612004-10-12 21:51:32 +0000307Vector example:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000308
309os.spawnvp(os.P_NOWAIT, path, args)
310==>
311Popen([path] + args[1:])
312
313
Tim Peterse718f612004-10-12 21:51:32 +0000314Environment example:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000315
316os.spawnlpe(os.P_NOWAIT, "/bin/mycmd", "mycmd", "myarg", env)
317==>
318Popen(["/bin/mycmd", "myarg"], env={"PATH": "/usr/bin"})
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000319"""
320
321import sys
322mswindows = (sys.platform == "win32")
323
Guido van Rossumfa0054a2007-05-24 04:05:35 +0000324import io
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000325import os
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000326import traceback
Christian Heimesfdab48e2008-01-20 09:06:41 +0000327import gc
Christian Heimesa342c012008-04-20 21:01:16 +0000328import signal
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000329
Peter Astrand454f7672005-01-01 09:36:35 +0000330# Exception classes used by this module.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000331class CalledProcessError(Exception):
Georg Brandlf9734072008-12-07 15:30:06 +0000332 """This exception is raised when a process run by check_call() or
333 check_output() returns a non-zero exit status.
334 The exit status will be stored in the returncode attribute;
335 check_output() will also store the output in the output attribute.
336 """
337 def __init__(self, returncode, cmd, output=None):
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000338 self.returncode = returncode
339 self.cmd = cmd
Georg Brandlf9734072008-12-07 15:30:06 +0000340 self.output = output
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000341 def __str__(self):
342 return "Command '%s' returned non-zero exit status %d" % (self.cmd, self.returncode)
343
Peter Astrand454f7672005-01-01 09:36:35 +0000344
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000345if mswindows:
346 import threading
347 import msvcrt
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000348 if 0: # <-- change this to use pywin32 instead of the _subprocess driver
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000349 import pywintypes
Tim Peterse8374a52004-10-13 03:15:00 +0000350 from win32api import GetStdHandle, STD_INPUT_HANDLE, \
351 STD_OUTPUT_HANDLE, STD_ERROR_HANDLE
352 from win32api import GetCurrentProcess, DuplicateHandle, \
353 GetModuleFileName, GetVersion
Peter Astrandc1d65362004-11-07 14:30:34 +0000354 from win32con import DUPLICATE_SAME_ACCESS, SW_HIDE
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000355 from win32pipe import CreatePipe
Tim Peterse8374a52004-10-13 03:15:00 +0000356 from win32process import CreateProcess, STARTUPINFO, \
357 GetExitCodeProcess, STARTF_USESTDHANDLES, \
Peter Astrandc1d65362004-11-07 14:30:34 +0000358 STARTF_USESHOWWINDOW, CREATE_NEW_CONSOLE
Christian Heimesa342c012008-04-20 21:01:16 +0000359 from win32process import TerminateProcess
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000360 from win32event import WaitForSingleObject, INFINITE, WAIT_OBJECT_0
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000361 else:
362 from _subprocess import *
363 class STARTUPINFO:
364 dwFlags = 0
365 hStdInput = None
366 hStdOutput = None
367 hStdError = None
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000368 wShowWindow = 0
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000369 class pywintypes:
370 error = IOError
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000371else:
372 import select
Gregory P. Smithd06fa472009-07-04 02:46:54 +0000373 _has_poll = hasattr(select, 'poll')
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000374 import errno
375 import fcntl
376 import pickle
377
Amaury Forgeot d'Arcace31022009-07-09 22:44:11 +0000378 # When select or poll has indicated that the file is writable,
379 # we can write up to _PIPE_BUF bytes without risk of blocking.
380 # POSIX defines PIPE_BUF as >= 512.
381 _PIPE_BUF = getattr(select, 'PIPE_BUF', 512)
382
383
Brett Cannona23810f2008-05-26 19:04:21 +0000384__all__ = ["Popen", "PIPE", "STDOUT", "call", "check_call", "getstatusoutput",
Georg Brandlf9734072008-12-07 15:30:06 +0000385 "getoutput", "check_output", "CalledProcessError"]
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000386
387try:
388 MAXFD = os.sysconf("SC_OPEN_MAX")
389except:
390 MAXFD = 256
391
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000392_active = []
393
394def _cleanup():
395 for inst in _active[:]:
Georg Brandl6aa2d1f2008-08-12 08:35:52 +0000396 res = inst._internal_poll(_deadstate=sys.maxsize)
Guido van Rossumb5d47ef2006-08-24 02:27:45 +0000397 if res is not None and res >= 0:
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000398 try:
399 _active.remove(inst)
400 except ValueError:
401 # This can happen if two threads create a new Popen instance.
402 # It's harmless that it was already removed, so ignore.
403 pass
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000404
405PIPE = -1
406STDOUT = -2
407
408
Peter Astrand5f5e1412004-12-05 20:15:36 +0000409def call(*popenargs, **kwargs):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000410 """Run command with arguments. Wait for command to complete, then
411 return the returncode attribute.
412
413 The arguments are the same as for the Popen constructor. Example:
414
415 retcode = call(["ls", "-l"])
416 """
Peter Astrand5f5e1412004-12-05 20:15:36 +0000417 return Popen(*popenargs, **kwargs).wait()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000418
419
Peter Astrand454f7672005-01-01 09:36:35 +0000420def check_call(*popenargs, **kwargs):
421 """Run command with arguments. Wait for command to complete. If
422 the exit code was zero then return, otherwise raise
423 CalledProcessError. The CalledProcessError object will have the
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000424 return code in the returncode attribute.
Peter Astrand454f7672005-01-01 09:36:35 +0000425
426 The arguments are the same as for the Popen constructor. Example:
427
428 check_call(["ls", "-l"])
429 """
430 retcode = call(*popenargs, **kwargs)
Peter Astrand454f7672005-01-01 09:36:35 +0000431 if retcode:
Georg Brandlf9734072008-12-07 15:30:06 +0000432 cmd = kwargs.get("args")
433 if cmd is None:
434 cmd = popenargs[0]
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000435 raise CalledProcessError(retcode, cmd)
Georg Brandlf9734072008-12-07 15:30:06 +0000436 return 0
437
438
439def check_output(*popenargs, **kwargs):
Georg Brandl2708f3a2009-12-20 14:38:23 +0000440 r"""Run command with arguments and return its output as a byte string.
Georg Brandlf9734072008-12-07 15:30:06 +0000441
442 If the exit code was non-zero it raises a CalledProcessError. The
443 CalledProcessError object will have the return code in the returncode
444 attribute and output in the output attribute.
445
446 The arguments are the same as for the Popen constructor. Example:
447
448 >>> check_output(["ls", "-l", "/dev/null"])
Georg Brandl2708f3a2009-12-20 14:38:23 +0000449 b'crw-rw-rw- 1 root root 1, 3 Oct 18 2007 /dev/null\n'
Georg Brandlf9734072008-12-07 15:30:06 +0000450
451 The stdout argument is not allowed as it is used internally.
Georg Brandl127d4702009-12-28 08:10:38 +0000452 To capture standard error in the result, use stderr=STDOUT.
Georg Brandlf9734072008-12-07 15:30:06 +0000453
454 >>> check_output(["/bin/sh", "-c",
Georg Brandl2708f3a2009-12-20 14:38:23 +0000455 ... "ls -l non_existent_file ; exit 0"],
Georg Brandl127d4702009-12-28 08:10:38 +0000456 ... stderr=STDOUT)
Georg Brandl2708f3a2009-12-20 14:38:23 +0000457 b'ls: non_existent_file: No such file or directory\n'
Georg Brandlf9734072008-12-07 15:30:06 +0000458 """
459 if 'stdout' in kwargs:
460 raise ValueError('stdout argument not allowed, it will be overridden.')
461 process = Popen(*popenargs, stdout=PIPE, **kwargs)
462 output, unused_err = process.communicate()
463 retcode = process.poll()
464 if retcode:
465 cmd = kwargs.get("args")
466 if cmd is None:
467 cmd = popenargs[0]
468 raise CalledProcessError(retcode, cmd, output=output)
469 return output
Peter Astrand454f7672005-01-01 09:36:35 +0000470
471
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000472def list2cmdline(seq):
473 """
474 Translate a sequence of arguments into a command line
475 string, using the same rules as the MS C runtime:
476
477 1) Arguments are delimited by white space, which is either a
478 space or a tab.
479
480 2) A string surrounded by double quotation marks is
481 interpreted as a single argument, regardless of white space
Christian Heimesfdab48e2008-01-20 09:06:41 +0000482 or pipe characters contained within. A quoted string can be
483 embedded in an argument.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000484
485 3) A double quotation mark preceded by a backslash is
486 interpreted as a literal double quotation mark.
487
488 4) Backslashes are interpreted literally, unless they
489 immediately precede a double quotation mark.
490
491 5) If backslashes immediately precede a double quotation mark,
492 every pair of backslashes is interpreted as a literal
493 backslash. If the number of backslashes is odd, the last
494 backslash escapes the next double quotation mark as
495 described in rule 3.
496 """
497
498 # See
Eric Smith3c573af2009-11-09 15:23:15 +0000499 # http://msdn.microsoft.com/en-us/library/17w5ykft.aspx
500 # or search http://msdn.microsoft.com for
501 # "Parsing C++ Command-Line Arguments"
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000502 result = []
503 needquote = False
504 for arg in seq:
505 bs_buf = []
506
507 # Add a space to separate this argument from the others
508 if result:
509 result.append(' ')
510
Christian Heimesfdab48e2008-01-20 09:06:41 +0000511 needquote = (" " in arg) or ("\t" in arg) or ("|" in arg) or not arg
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000512 if needquote:
513 result.append('"')
514
515 for c in arg:
516 if c == '\\':
Tim Peterse718f612004-10-12 21:51:32 +0000517 # Don't know if we need to double yet.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000518 bs_buf.append(c)
519 elif c == '"':
Christian Heimesfdab48e2008-01-20 09:06:41 +0000520 # Double backslashes.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000521 result.append('\\' * len(bs_buf)*2)
522 bs_buf = []
523 result.append('\\"')
524 else:
525 # Normal char
526 if bs_buf:
527 result.extend(bs_buf)
528 bs_buf = []
529 result.append(c)
530
Christian Heimesfdab48e2008-01-20 09:06:41 +0000531 # Add remaining backslashes, if any.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000532 if bs_buf:
533 result.extend(bs_buf)
534
535 if needquote:
Peter Astrand7e78ade2005-03-03 21:10:23 +0000536 result.extend(bs_buf)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000537 result.append('"')
538
539 return ''.join(result)
540
541
Brett Cannona23810f2008-05-26 19:04:21 +0000542# Various tools for executing commands and looking at their output and status.
543#
544# NB This only works (and is only relevant) for UNIX.
545
546def getstatusoutput(cmd):
547 """Return (status, output) of executing cmd in a shell.
548
549 Execute the string 'cmd' in a shell with os.popen() and return a 2-tuple
550 (status, output). cmd is actually run as '{ cmd ; } 2>&1', so that the
551 returned output will contain output or error messages. A trailing newline
552 is stripped from the output. The exit status for the command can be
553 interpreted according to the rules for the C function wait(). Example:
554
555 >>> import subprocess
556 >>> subprocess.getstatusoutput('ls /bin/ls')
557 (0, '/bin/ls')
558 >>> subprocess.getstatusoutput('cat /bin/junk')
559 (256, 'cat: /bin/junk: No such file or directory')
560 >>> subprocess.getstatusoutput('/bin/junk')
561 (256, 'sh: /bin/junk: not found')
562 """
563 pipe = os.popen('{ ' + cmd + '; } 2>&1', 'r')
564 text = pipe.read()
565 sts = pipe.close()
566 if sts is None: sts = 0
567 if text[-1:] == '\n': text = text[:-1]
568 return sts, text
569
570
571def getoutput(cmd):
572 """Return output (stdout or stderr) of executing cmd in a shell.
573
574 Like getstatusoutput(), except the exit status is ignored and the return
575 value is a string containing the command's output. Example:
576
577 >>> import subprocess
578 >>> subprocess.getoutput('ls /bin/ls')
579 '/bin/ls'
580 """
581 return getstatusoutput(cmd)[1]
582
583
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000584class Popen(object):
585 def __init__(self, args, bufsize=0, executable=None,
586 stdin=None, stdout=None, stderr=None,
587 preexec_fn=None, close_fds=False, shell=False,
588 cwd=None, env=None, universal_newlines=False,
589 startupinfo=None, creationflags=0):
590 """Create new Popen instance."""
591 _cleanup()
592
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000593 self._child_created = False
Guido van Rossum46a05a72007-06-07 21:56:45 +0000594 if bufsize is None:
595 bufsize = 0 # Restore default
Walter Dörwaldaa97f042007-05-03 21:05:51 +0000596 if not isinstance(bufsize, int):
Peter Astrand738131d2004-11-30 21:04:45 +0000597 raise TypeError("bufsize must be an integer")
598
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000599 if mswindows:
Tim Peterse8374a52004-10-13 03:15:00 +0000600 if preexec_fn is not None:
601 raise ValueError("preexec_fn is not supported on Windows "
602 "platforms")
Guido van Rossume7ba4952007-06-06 23:52:48 +0000603 if close_fds and (stdin is not None or stdout is not None or
604 stderr is not None):
Tim Peterse8374a52004-10-13 03:15:00 +0000605 raise ValueError("close_fds is not supported on Windows "
Guido van Rossume7ba4952007-06-06 23:52:48 +0000606 "platforms if you redirect stdin/stdout/stderr")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000607 else:
608 # POSIX
Tim Peterse8374a52004-10-13 03:15:00 +0000609 if startupinfo is not None:
610 raise ValueError("startupinfo is only supported on Windows "
611 "platforms")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000612 if creationflags != 0:
Tim Peterse8374a52004-10-13 03:15:00 +0000613 raise ValueError("creationflags is only supported on Windows "
614 "platforms")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000615
Tim Peterse718f612004-10-12 21:51:32 +0000616 self.stdin = None
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000617 self.stdout = None
618 self.stderr = None
619 self.pid = None
620 self.returncode = None
621 self.universal_newlines = universal_newlines
622
623 # Input and output objects. The general principle is like
624 # this:
625 #
626 # Parent Child
627 # ------ -----
628 # p2cwrite ---stdin---> p2cread
629 # c2pread <--stdout--- c2pwrite
630 # errread <--stderr--- errwrite
631 #
632 # On POSIX, the child objects are file descriptors. On
633 # Windows, these are Windows file handles. The parent objects
634 # are file descriptors on both platforms. The parent objects
635 # are None when not using PIPEs. The child objects are None
636 # when not redirecting.
Tim Peterse718f612004-10-12 21:51:32 +0000637
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000638 (p2cread, p2cwrite,
639 c2pread, c2pwrite,
640 errread, errwrite) = self._get_handles(stdin, stdout, stderr)
641
642 self._execute_child(args, executable, preexec_fn, close_fds,
643 cwd, env, universal_newlines,
644 startupinfo, creationflags, shell,
645 p2cread, p2cwrite,
646 c2pread, c2pwrite,
647 errread, errwrite)
648
Thomas Wouterscf297e42007-02-23 15:07:44 +0000649 if mswindows:
Hirokazu Yamamoto0c988172009-03-03 22:41:26 +0000650 if p2cwrite is not None:
651 p2cwrite = msvcrt.open_osfhandle(p2cwrite.Detach(), 0)
652 if c2pread is not None:
653 c2pread = msvcrt.open_osfhandle(c2pread.Detach(), 0)
654 if errread is not None:
655 errread = msvcrt.open_osfhandle(errread.Detach(), 0)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000656
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):
Alexandre Vassalotti711ed4a2009-07-17 10:42:05 +0000728 """Construct and return tuple with IO objects:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000729 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)
Hirokazu Yamamoto0c988172009-03-03 22:41:26 +0000740 if p2cread is None:
741 p2cread, _ = CreatePipe(None, 0)
742 elif stdin == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000743 p2cread, p2cwrite = CreatePipe(None, 0)
Peter Astrandd38ddf42005-02-10 08:32:50 +0000744 elif isinstance(stdin, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000745 p2cread = msvcrt.get_osfhandle(stdin)
746 else:
747 # Assuming file-like object
748 p2cread = msvcrt.get_osfhandle(stdin.fileno())
749 p2cread = self._make_inheritable(p2cread)
750
Peter Astrandd38ddf42005-02-10 08:32:50 +0000751 if stdout is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000752 c2pwrite = GetStdHandle(STD_OUTPUT_HANDLE)
Hirokazu Yamamoto0c988172009-03-03 22:41:26 +0000753 if c2pwrite is None:
754 _, c2pwrite = CreatePipe(None, 0)
755 elif stdout == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000756 c2pread, c2pwrite = CreatePipe(None, 0)
Peter Astrandd38ddf42005-02-10 08:32:50 +0000757 elif isinstance(stdout, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000758 c2pwrite = msvcrt.get_osfhandle(stdout)
759 else:
760 # Assuming file-like object
761 c2pwrite = msvcrt.get_osfhandle(stdout.fileno())
762 c2pwrite = self._make_inheritable(c2pwrite)
763
Peter Astrandd38ddf42005-02-10 08:32:50 +0000764 if stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000765 errwrite = GetStdHandle(STD_ERROR_HANDLE)
Hirokazu Yamamoto0c988172009-03-03 22:41:26 +0000766 if errwrite is None:
767 _, errwrite = CreatePipe(None, 0)
768 elif stderr == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000769 errread, errwrite = CreatePipe(None, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000770 elif stderr == STDOUT:
771 errwrite = c2pwrite
Peter Astrandd38ddf42005-02-10 08:32:50 +0000772 elif isinstance(stderr, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000773 errwrite = msvcrt.get_osfhandle(stderr)
774 else:
775 # Assuming file-like object
776 errwrite = msvcrt.get_osfhandle(stderr.fileno())
777 errwrite = self._make_inheritable(errwrite)
778
779 return (p2cread, p2cwrite,
780 c2pread, c2pwrite,
781 errread, errwrite)
782
783
784 def _make_inheritable(self, handle):
785 """Return a duplicate of handle, which is inheritable"""
786 return DuplicateHandle(GetCurrentProcess(), handle,
787 GetCurrentProcess(), 0, 1,
788 DUPLICATE_SAME_ACCESS)
789
790
791 def _find_w9xpopen(self):
792 """Find and return absolut path to w9xpopen.exe"""
Tim Peterse8374a52004-10-13 03:15:00 +0000793 w9xpopen = os.path.join(os.path.dirname(GetModuleFileName(0)),
794 "w9xpopen.exe")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000795 if not os.path.exists(w9xpopen):
796 # Eeek - file-not-found - possibly an embedding
797 # situation - see if we can locate it in sys.exec_prefix
Tim Peterse8374a52004-10-13 03:15:00 +0000798 w9xpopen = os.path.join(os.path.dirname(sys.exec_prefix),
799 "w9xpopen.exe")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000800 if not os.path.exists(w9xpopen):
Tim Peterse8374a52004-10-13 03:15:00 +0000801 raise RuntimeError("Cannot locate w9xpopen.exe, which is "
802 "needed for Popen to work with your "
803 "shell or platform.")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000804 return w9xpopen
805
Tim Peterse718f612004-10-12 21:51:32 +0000806
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000807 def _execute_child(self, args, executable, preexec_fn, close_fds,
808 cwd, env, universal_newlines,
809 startupinfo, creationflags, shell,
810 p2cread, p2cwrite,
811 c2pread, c2pwrite,
812 errread, errwrite):
813 """Execute program (MS Windows version)"""
814
Guido van Rossum3172c5d2007-10-16 18:12:55 +0000815 if not isinstance(args, str):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000816 args = list2cmdline(args)
817
Peter Astrandc1d65362004-11-07 14:30:34 +0000818 # Process startup details
Peter Astrandd38ddf42005-02-10 08:32:50 +0000819 if startupinfo is None:
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000820 startupinfo = STARTUPINFO()
821 if None not in (p2cread, c2pwrite, errwrite):
Peter Astrandc1d65362004-11-07 14:30:34 +0000822 startupinfo.dwFlags |= STARTF_USESTDHANDLES
823 startupinfo.hStdInput = p2cread
824 startupinfo.hStdOutput = c2pwrite
825 startupinfo.hStdError = errwrite
826
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000827 if shell:
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000828 startupinfo.dwFlags |= STARTF_USESHOWWINDOW
829 startupinfo.wShowWindow = SW_HIDE
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000830 comspec = os.environ.get("COMSPEC", "cmd.exe")
831 args = comspec + " /c " + args
Guido van Rossume2a383d2007-01-15 16:59:06 +0000832 if (GetVersion() >= 0x80000000 or
Tim Peterse8374a52004-10-13 03:15:00 +0000833 os.path.basename(comspec).lower() == "command.com"):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000834 # Win9x, or using command.com on NT. We need to
835 # use the w9xpopen intermediate program. For more
836 # information, see KB Q150956
837 # (http://web.archive.org/web/20011105084002/http://support.microsoft.com/support/kb/articles/Q150/9/56.asp)
838 w9xpopen = self._find_w9xpopen()
839 args = '"%s" %s' % (w9xpopen, args)
840 # Not passing CREATE_NEW_CONSOLE has been known to
841 # cause random failures on win9x. Specifically a
842 # dialog: "Your program accessed mem currently in
843 # use at xxx" and a hopeful warning about the
Mark Dickinson934896d2009-02-21 20:59:32 +0000844 # stability of your system. Cost is Ctrl+C won't
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000845 # kill children.
846 creationflags |= CREATE_NEW_CONSOLE
847
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000848 # Start the process
849 try:
850 hp, ht, pid, tid = CreateProcess(executable, args,
Tim Peterse8374a52004-10-13 03:15:00 +0000851 # no special security
852 None, None,
Guido van Rossume7ba4952007-06-06 23:52:48 +0000853 int(not close_fds),
Tim Peterse8374a52004-10-13 03:15:00 +0000854 creationflags,
855 env,
856 cwd,
857 startupinfo)
Guido van Rossumb940e112007-01-10 16:19:56 +0000858 except pywintypes.error as e:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000859 # Translate pywintypes.error to WindowsError, which is
860 # a subclass of OSError. FIXME: We should really
861 # translate errno using _sys_errlist (or simliar), but
862 # how can this be done from Python?
863 raise WindowsError(*e.args)
864
865 # Retain the process handle, but close the thread handle
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000866 self._child_created = True
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000867 self._handle = hp
868 self.pid = pid
869 ht.Close()
870
Andrew M. Kuchling51ee66e2004-10-12 16:38:42 +0000871 # Child is launched. Close the parent's copy of those pipe
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000872 # handles that only the child should have open. You need
873 # to make sure that no handles to the write end of the
874 # output pipe are maintained in this process or else the
875 # pipe will not close when the child process exits and the
876 # ReadFile will hang.
Peter Astrandd38ddf42005-02-10 08:32:50 +0000877 if p2cread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000878 p2cread.Close()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000879 if c2pwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000880 c2pwrite.Close()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000881 if errwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000882 errwrite.Close()
883
Tim Peterse718f612004-10-12 21:51:32 +0000884
Georg Brandl6aa2d1f2008-08-12 08:35:52 +0000885 def _internal_poll(self, _deadstate=None):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000886 """Check if child process has terminated. Returns returncode
887 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +0000888 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000889 if WaitForSingleObject(self._handle, 0) == WAIT_OBJECT_0:
890 self.returncode = GetExitCodeProcess(self._handle)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000891 return self.returncode
892
893
894 def wait(self):
895 """Wait for child process to terminate. 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 obj = WaitForSingleObject(self._handle, INFINITE)
899 self.returncode = GetExitCodeProcess(self._handle)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000900 return self.returncode
901
902
903 def _readerthread(self, fh, buffer):
904 buffer.append(fh.read())
905
906
Peter Astrand23109f02005-03-03 20:28:59 +0000907 def _communicate(self, input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000908 stdout = None # Return
909 stderr = None # Return
910
911 if self.stdout:
912 stdout = []
Tim Peterse8374a52004-10-13 03:15:00 +0000913 stdout_thread = threading.Thread(target=self._readerthread,
914 args=(self.stdout, stdout))
Benjamin Peterson632e0362008-08-18 19:08:51 +0000915 stdout_thread.daemon = True
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000916 stdout_thread.start()
917 if self.stderr:
918 stderr = []
Tim Peterse8374a52004-10-13 03:15:00 +0000919 stderr_thread = threading.Thread(target=self._readerthread,
920 args=(self.stderr, stderr))
Benjamin Peterson632e0362008-08-18 19:08:51 +0000921 stderr_thread.daemon = True
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000922 stderr_thread.start()
923
924 if self.stdin:
Peter Astrandd38ddf42005-02-10 08:32:50 +0000925 if input is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000926 self.stdin.write(input)
927 self.stdin.close()
928
929 if self.stdout:
930 stdout_thread.join()
931 if self.stderr:
932 stderr_thread.join()
933
934 # All data exchanged. Translate lists into strings.
Peter Astrandd38ddf42005-02-10 08:32:50 +0000935 if stdout is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000936 stdout = stdout[0]
Peter Astrandd38ddf42005-02-10 08:32:50 +0000937 if stderr is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000938 stderr = stderr[0]
939
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000940 self.wait()
941 return (stdout, stderr)
942
Christian Heimesa342c012008-04-20 21:01:16 +0000943 def send_signal(self, sig):
944 """Send a signal to the process
945 """
946 if sig == signal.SIGTERM:
947 self.terminate()
948 else:
949 raise ValueError("Only SIGTERM is supported on Windows")
950
951 def terminate(self):
952 """Terminates the process
953 """
954 TerminateProcess(self._handle, 1)
955
956 kill = terminate
957
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000958 else:
959 #
960 # POSIX methods
961 #
962 def _get_handles(self, stdin, stdout, stderr):
Alexandre Vassalotti711ed4a2009-07-17 10:42:05 +0000963 """Construct and return tuple with IO objects:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000964 p2cread, p2cwrite, c2pread, c2pwrite, errread, errwrite
965 """
966 p2cread, p2cwrite = None, None
967 c2pread, c2pwrite = None, None
968 errread, errwrite = None, None
969
Peter Astrandd38ddf42005-02-10 08:32:50 +0000970 if stdin is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000971 pass
972 elif stdin == PIPE:
973 p2cread, p2cwrite = os.pipe()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000974 elif isinstance(stdin, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000975 p2cread = stdin
976 else:
977 # Assuming file-like object
978 p2cread = stdin.fileno()
979
Peter Astrandd38ddf42005-02-10 08:32:50 +0000980 if stdout is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000981 pass
982 elif stdout == PIPE:
983 c2pread, c2pwrite = os.pipe()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000984 elif isinstance(stdout, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000985 c2pwrite = stdout
986 else:
987 # Assuming file-like object
Tim Peterse718f612004-10-12 21:51:32 +0000988 c2pwrite = stdout.fileno()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000989
Peter Astrandd38ddf42005-02-10 08:32:50 +0000990 if stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000991 pass
992 elif stderr == PIPE:
993 errread, errwrite = os.pipe()
994 elif stderr == STDOUT:
995 errwrite = c2pwrite
Peter Astrandd38ddf42005-02-10 08:32:50 +0000996 elif isinstance(stderr, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000997 errwrite = stderr
998 else:
999 # Assuming file-like object
1000 errwrite = stderr.fileno()
1001
1002 return (p2cread, p2cwrite,
1003 c2pread, c2pwrite,
1004 errread, errwrite)
1005
1006
1007 def _set_cloexec_flag(self, fd):
1008 try:
1009 cloexec_flag = fcntl.FD_CLOEXEC
1010 except AttributeError:
1011 cloexec_flag = 1
1012
1013 old = fcntl.fcntl(fd, fcntl.F_GETFD)
1014 fcntl.fcntl(fd, fcntl.F_SETFD, old | cloexec_flag)
1015
1016
1017 def _close_fds(self, but):
Christian Heimesfdab48e2008-01-20 09:06:41 +00001018 os.closerange(3, but)
1019 os.closerange(but + 1, MAXFD)
Tim Peterse718f612004-10-12 21:51:32 +00001020
1021
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001022 def _execute_child(self, args, executable, preexec_fn, close_fds,
1023 cwd, env, universal_newlines,
1024 startupinfo, creationflags, shell,
1025 p2cread, p2cwrite,
1026 c2pread, c2pwrite,
1027 errread, errwrite):
1028 """Execute program (POSIX version)"""
1029
Guido van Rossum3172c5d2007-10-16 18:12:55 +00001030 if isinstance(args, str):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001031 args = [args]
Thomas Wouters89f507f2006-12-13 04:49:30 +00001032 else:
1033 args = list(args)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001034
1035 if shell:
1036 args = ["/bin/sh", "-c"] + args
1037
Peter Astrandd38ddf42005-02-10 08:32:50 +00001038 if executable is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001039 executable = args[0]
1040
1041 # For transferring possible exec failure from child to parent
1042 # The first char specifies the exception type: 0 means
1043 # OSError, 1 means some other error.
1044 errpipe_read, errpipe_write = os.pipe()
Christian Heimesfdab48e2008-01-20 09:06:41 +00001045 try:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001046 try:
Facundo Batista10706e22009-06-19 20:34:30 +00001047 self._set_cloexec_flag(errpipe_write)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001048
Facundo Batista10706e22009-06-19 20:34:30 +00001049 gc_was_enabled = gc.isenabled()
1050 # Disable gc to avoid bug where gc -> file_dealloc ->
1051 # write to stderr -> hang. http://bugs.python.org/issue1336
1052 gc.disable()
1053 try:
1054 self.pid = os.fork()
1055 except:
1056 if gc_was_enabled:
1057 gc.enable()
1058 raise
1059 self._child_created = True
1060 if self.pid == 0:
1061 # Child
1062 try:
1063 # Close parent's pipe ends
1064 if p2cwrite is not None:
1065 os.close(p2cwrite)
1066 if c2pread is not None:
1067 os.close(c2pread)
1068 if errread is not None:
1069 os.close(errread)
1070 os.close(errpipe_read)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001071
Facundo Batista10706e22009-06-19 20:34:30 +00001072 # Dup fds for child
1073 if p2cread is not None:
1074 os.dup2(p2cread, 0)
1075 if c2pwrite is not None:
1076 os.dup2(c2pwrite, 1)
1077 if errwrite is not None:
1078 os.dup2(errwrite, 2)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001079
Facundo Batista10706e22009-06-19 20:34:30 +00001080 # Close pipe fds. Make sure we don't close the
1081 # same fd more than once, or standard fds.
1082 if p2cread is not None and p2cread not in (0,):
1083 os.close(p2cread)
1084 if c2pwrite is not None and \
1085 c2pwrite not in (p2cread, 1):
1086 os.close(c2pwrite)
1087 if (errwrite is not None and
1088 errwrite not in (p2cread, c2pwrite, 2)):
1089 os.close(errwrite)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001090
Facundo Batista10706e22009-06-19 20:34:30 +00001091 # Close all other fds, if asked for
1092 if close_fds:
1093 self._close_fds(but=errpipe_write)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001094
Facundo Batista10706e22009-06-19 20:34:30 +00001095 if cwd is not None:
1096 os.chdir(cwd)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001097
Facundo Batista10706e22009-06-19 20:34:30 +00001098 if preexec_fn:
1099 preexec_fn()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001100
Facundo Batista10706e22009-06-19 20:34:30 +00001101 if env is None:
1102 os.execvp(executable, args)
1103 else:
1104 os.execvpe(executable, args, env)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001105
Facundo Batista10706e22009-06-19 20:34:30 +00001106 except:
1107 exc_type, exc_value, tb = sys.exc_info()
1108 # Save the traceback and attach it to the exception
1109 # object
1110 exc_lines = traceback.format_exception(exc_type,
1111 exc_value,
1112 tb)
1113 exc_value.child_traceback = ''.join(exc_lines)
1114 os.write(errpipe_write, pickle.dumps(exc_value))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001115
Facundo Batista10706e22009-06-19 20:34:30 +00001116 # This exitcode won't be reported to applications, so
1117 # it really doesn't matter what we return.
1118 os._exit(255)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001119
Facundo Batista10706e22009-06-19 20:34:30 +00001120 # Parent
1121 if gc_was_enabled:
1122 gc.enable()
1123 finally:
1124 # be sure the FD is closed no matter what
1125 os.close(errpipe_write)
1126
1127 if p2cread is not None and p2cwrite is not None:
1128 os.close(p2cread)
1129 if c2pwrite is not None and c2pread is not None:
1130 os.close(c2pwrite)
1131 if errwrite is not None and errread is not None:
1132 os.close(errwrite)
1133
1134 # Wait for exec to fail or succeed; possibly raising an
1135 # exception (limited to 1 MB)
1136 data = os.read(errpipe_read, 1048576)
1137 finally:
1138 # be sure the FD is closed no matter what
1139 os.close(errpipe_read)
1140
Guido van Rossumaf2362a2007-05-15 22:32:02 +00001141 if data:
Peter Astrandf791d7a2005-01-01 09:38:57 +00001142 os.waitpid(self.pid, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001143 child_exception = pickle.loads(data)
Benjamin Petersond75fcb42009-02-19 04:22:03 +00001144 for fd in (p2cwrite, c2pread, errread):
1145 if fd is not None:
1146 os.close(fd)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001147 raise child_exception
1148
1149
1150 def _handle_exitstatus(self, sts):
1151 if os.WIFSIGNALED(sts):
1152 self.returncode = -os.WTERMSIG(sts)
1153 elif os.WIFEXITED(sts):
1154 self.returncode = os.WEXITSTATUS(sts)
1155 else:
1156 # Should never happen
1157 raise RuntimeError("Unknown child exit status!")
1158
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001159
Georg Brandl6aa2d1f2008-08-12 08:35:52 +00001160 def _internal_poll(self, _deadstate=None):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001161 """Check if child process has terminated. Returns returncode
1162 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +00001163 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001164 try:
1165 pid, sts = os.waitpid(self.pid, os.WNOHANG)
1166 if pid == self.pid:
1167 self._handle_exitstatus(sts)
1168 except os.error:
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001169 if _deadstate is not None:
1170 self.returncode = _deadstate
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001171 return self.returncode
1172
1173
1174 def wait(self):
1175 """Wait for child process to terminate. Returns returncode
1176 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +00001177 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001178 pid, sts = os.waitpid(self.pid, 0)
1179 self._handle_exitstatus(sts)
1180 return self.returncode
1181
1182
Peter Astrand23109f02005-03-03 20:28:59 +00001183 def _communicate(self, input):
Gregory P. Smithd06fa472009-07-04 02:46:54 +00001184 if self.stdin:
1185 # Flush stdio buffer. This might block, if the user has
1186 # been writing to .stdin in an uncontrolled fashion.
1187 self.stdin.flush()
1188 if not input:
1189 self.stdin.close()
1190
1191 if _has_poll:
1192 stdout, stderr = self._communicate_with_poll(input)
1193 else:
1194 stdout, stderr = self._communicate_with_select(input)
1195
1196 # All data exchanged. Translate lists into strings.
1197 if stdout is not None:
1198 stdout = b''.join(stdout)
1199 if stderr is not None:
1200 stderr = b''.join(stderr)
1201
1202 # Translate newlines, if requested.
1203 # This also turns bytes into strings.
1204 if self.universal_newlines:
1205 if stdout is not None:
1206 stdout = self._translate_newlines(stdout,
1207 self.stdout.encoding)
1208 if stderr is not None:
1209 stderr = self._translate_newlines(stderr,
1210 self.stderr.encoding)
1211
1212 self.wait()
1213 return (stdout, stderr)
1214
1215
1216 def _communicate_with_poll(self, input):
1217 stdout = None # Return
1218 stderr = None # Return
1219 fd2file = {}
1220 fd2output = {}
1221
1222 poller = select.poll()
1223 def register_and_append(file_obj, eventmask):
1224 poller.register(file_obj.fileno(), eventmask)
1225 fd2file[file_obj.fileno()] = file_obj
1226
1227 def close_unregister_and_remove(fd):
1228 poller.unregister(fd)
1229 fd2file[fd].close()
1230 fd2file.pop(fd)
1231
1232 if self.stdin and input:
1233 register_and_append(self.stdin, select.POLLOUT)
1234
1235 select_POLLIN_POLLPRI = select.POLLIN | select.POLLPRI
1236 if self.stdout:
1237 register_and_append(self.stdout, select_POLLIN_POLLPRI)
1238 fd2output[self.stdout.fileno()] = stdout = []
1239 if self.stderr:
1240 register_and_append(self.stderr, select_POLLIN_POLLPRI)
1241 fd2output[self.stderr.fileno()] = stderr = []
1242
1243 input_offset = 0
1244 while fd2file:
1245 try:
1246 ready = poller.poll()
1247 except select.error as e:
1248 if e.args[0] == errno.EINTR:
1249 continue
1250 raise
1251
1252 # XXX Rewrite these to use non-blocking I/O on the
1253 # file objects; they are no longer using C stdio!
1254
1255 for fd, mode in ready:
1256 if mode & select.POLLOUT:
1257 chunk = input[input_offset : input_offset + _PIPE_BUF]
1258 input_offset += os.write(fd, chunk)
1259 if input_offset >= len(input):
1260 close_unregister_and_remove(fd)
1261 elif mode & select_POLLIN_POLLPRI:
1262 data = os.read(fd, 4096)
1263 if not data:
1264 close_unregister_and_remove(fd)
1265 fd2output[fd].append(data)
1266 else:
1267 # Ignore hang up or errors.
1268 close_unregister_and_remove(fd)
1269
1270 return (stdout, stderr)
1271
1272
1273 def _communicate_with_select(self, input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001274 read_set = []
1275 write_set = []
1276 stdout = None # Return
1277 stderr = None # Return
1278
Gregory P. Smithd06fa472009-07-04 02:46:54 +00001279 if self.stdin and input:
1280 write_set.append(self.stdin)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001281 if self.stdout:
1282 read_set.append(self.stdout)
Tim Peterse718f612004-10-12 21:51:32 +00001283 stdout = []
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001284 if self.stderr:
1285 read_set.append(self.stderr)
1286 stderr = []
1287
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001288 input_offset = 0
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001289 while read_set or write_set:
Georg Brandl86b2fb92008-07-16 03:43:04 +00001290 try:
1291 rlist, wlist, xlist = select.select(read_set, write_set, [])
1292 except select.error as e:
1293 if e.args[0] == errno.EINTR:
1294 continue
1295 raise
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001296
Guido van Rossum98297ee2007-11-06 21:34:58 +00001297 # XXX Rewrite these to use non-blocking I/O on the
1298 # file objects; they are no longer using C stdio!
1299
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001300 if self.stdin in wlist:
Gregory P. Smithd06fa472009-07-04 02:46:54 +00001301 chunk = input[input_offset : input_offset + _PIPE_BUF]
Guido van Rossumbae07c92007-10-08 02:46:15 +00001302 bytes_written = os.write(self.stdin.fileno(), chunk)
Thomas Wouters9fe394c2007-02-05 01:24:16 +00001303 input_offset += bytes_written
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001304 if input_offset >= len(input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001305 self.stdin.close()
1306 write_set.remove(self.stdin)
1307
1308 if self.stdout in rlist:
1309 data = os.read(self.stdout.fileno(), 1024)
Guido van Rossumc9e363c2007-05-15 23:18:55 +00001310 if not data:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001311 self.stdout.close()
1312 read_set.remove(self.stdout)
1313 stdout.append(data)
1314
1315 if self.stderr in rlist:
1316 data = os.read(self.stderr.fileno(), 1024)
Guido van Rossumc9e363c2007-05-15 23:18:55 +00001317 if not data:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001318 self.stderr.close()
1319 read_set.remove(self.stderr)
1320 stderr.append(data)
1321
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001322 return (stdout, stderr)
1323
Gregory P. Smithd06fa472009-07-04 02:46:54 +00001324
Christian Heimesa342c012008-04-20 21:01:16 +00001325 def send_signal(self, sig):
1326 """Send a signal to the process
1327 """
1328 os.kill(self.pid, sig)
1329
1330 def terminate(self):
1331 """Terminate the process with SIGTERM
1332 """
1333 self.send_signal(signal.SIGTERM)
1334
1335 def kill(self):
1336 """Kill the process with SIGKILL
1337 """
1338 self.send_signal(signal.SIGKILL)
1339
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001340
1341def _demo_posix():
1342 #
1343 # Example 1: Simple redirection: Get process list
1344 #
1345 plist = Popen(["ps"], stdout=PIPE).communicate()[0]
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001346 print("Process list:")
1347 print(plist)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001348
1349 #
1350 # Example 2: Change uid before executing child
1351 #
1352 if os.getuid() == 0:
1353 p = Popen(["id"], preexec_fn=lambda: os.setuid(100))
1354 p.wait()
1355
1356 #
1357 # Example 3: Connecting several subprocesses
1358 #
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001359 print("Looking for 'hda'...")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001360 p1 = Popen(["dmesg"], stdout=PIPE)
1361 p2 = Popen(["grep", "hda"], stdin=p1.stdout, stdout=PIPE)
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001362 print(repr(p2.communicate()[0]))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001363
1364 #
1365 # Example 4: Catch execution error
1366 #
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001367 print()
1368 print("Trying a weird file...")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001369 try:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001370 print(Popen(["/this/path/does/not/exist"]).communicate())
Guido van Rossumb940e112007-01-10 16:19:56 +00001371 except OSError as e:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001372 if e.errno == errno.ENOENT:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001373 print("The file didn't exist. I thought so...")
1374 print("Child traceback:")
1375 print(e.child_traceback)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001376 else:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001377 print("Error", e.errno)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001378 else:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001379 print("Gosh. No error.", file=sys.stderr)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001380
1381
1382def _demo_windows():
1383 #
1384 # Example 1: Connecting several subprocesses
1385 #
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001386 print("Looking for 'PROMPT' in set output...")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001387 p1 = Popen("set", stdout=PIPE, shell=True)
1388 p2 = Popen('find "PROMPT"', stdin=p1.stdout, stdout=PIPE)
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001389 print(repr(p2.communicate()[0]))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001390
1391 #
1392 # Example 2: Simple execution of program
1393 #
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001394 print("Executing calc...")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001395 p = Popen("calc")
1396 p.wait()
1397
1398
1399if __name__ == "__main__":
1400 if mswindows:
1401 _demo_windows()
1402 else:
1403 _demo_posix()