blob: fa54bcaa660bb200098b087de5788899ec7bfab6 [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
Georg Brandlae83d6e2009-08-13 09:04:31 +0000374 _has_poll = hasattr(select, 'poll')
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000375 import errno
376 import fcntl
377 import pickle
378
Gregory P. Smith10d29522009-08-13 18:33:30 +0000379 # When select or poll has indicated that the file is writable,
380 # we can write up to _PIPE_BUF bytes without risk of blocking.
381 # POSIX defines PIPE_BUF as >= 512.
382 _PIPE_BUF = getattr(select, 'PIPE_BUF', 512)
383
384
Brett Cannona23810f2008-05-26 19:04:21 +0000385__all__ = ["Popen", "PIPE", "STDOUT", "call", "check_call", "getstatusoutput",
Georg Brandlf9734072008-12-07 15:30:06 +0000386 "getoutput", "check_output", "CalledProcessError"]
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000387
388try:
389 MAXFD = os.sysconf("SC_OPEN_MAX")
390except:
391 MAXFD = 256
392
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000393_active = []
394
395def _cleanup():
396 for inst in _active[:]:
Georg Brandl6aa2d1f2008-08-12 08:35:52 +0000397 res = inst._internal_poll(_deadstate=sys.maxsize)
Guido van Rossumb5d47ef2006-08-24 02:27:45 +0000398 if res is not None and res >= 0:
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000399 try:
400 _active.remove(inst)
401 except ValueError:
402 # This can happen if two threads create a new Popen instance.
403 # It's harmless that it was already removed, so ignore.
404 pass
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000405
406PIPE = -1
407STDOUT = -2
408
409
Gregory P. Smith3fff44d2010-03-01 00:43:08 +0000410def _eintr_retry_call(func, *args):
411 while True:
412 try:
413 return func(*args)
414 except OSError as e:
415 if e.errno == errno.EINTR:
416 continue
417 raise
418
419
Peter Astrand5f5e1412004-12-05 20:15:36 +0000420def call(*popenargs, **kwargs):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000421 """Run command with arguments. Wait for command to complete, then
422 return the returncode attribute.
423
424 The arguments are the same as for the Popen constructor. Example:
425
426 retcode = call(["ls", "-l"])
427 """
Peter Astrand5f5e1412004-12-05 20:15:36 +0000428 return Popen(*popenargs, **kwargs).wait()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000429
430
Peter Astrand454f7672005-01-01 09:36:35 +0000431def check_call(*popenargs, **kwargs):
432 """Run command with arguments. Wait for command to complete. If
433 the exit code was zero then return, otherwise raise
434 CalledProcessError. The CalledProcessError object will have the
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000435 return code in the returncode attribute.
Peter Astrand454f7672005-01-01 09:36:35 +0000436
437 The arguments are the same as for the Popen constructor. Example:
438
439 check_call(["ls", "-l"])
440 """
441 retcode = call(*popenargs, **kwargs)
Peter Astrand454f7672005-01-01 09:36:35 +0000442 if retcode:
Georg Brandlf9734072008-12-07 15:30:06 +0000443 cmd = kwargs.get("args")
444 if cmd is None:
445 cmd = popenargs[0]
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000446 raise CalledProcessError(retcode, cmd)
Georg Brandlf9734072008-12-07 15:30:06 +0000447 return 0
448
449
450def check_output(*popenargs, **kwargs):
451 """Run command with arguments and return its output as a byte string.
452
453 If the exit code was non-zero it raises a CalledProcessError. The
454 CalledProcessError object will have the return code in the returncode
455 attribute and output in the output attribute.
456
457 The arguments are the same as for the Popen constructor. Example:
458
459 >>> check_output(["ls", "-l", "/dev/null"])
460 'crw-rw-rw- 1 root root 1, 3 Oct 18 2007 /dev/null\n'
461
462 The stdout argument is not allowed as it is used internally.
463 To capture standard error in the result, use stderr=subprocess.STDOUT.
464
465 >>> check_output(["/bin/sh", "-c",
Mark Dickinson934896d2009-02-21 20:59:32 +0000466 "ls -l non_existent_file ; exit 0"],
Georg Brandlf9734072008-12-07 15:30:06 +0000467 stderr=subprocess.STDOUT)
Mark Dickinson934896d2009-02-21 20:59:32 +0000468 'ls: non_existent_file: No such file or directory\n'
Georg Brandlf9734072008-12-07 15:30:06 +0000469 """
470 if 'stdout' in kwargs:
471 raise ValueError('stdout argument not allowed, it will be overridden.')
472 process = Popen(*popenargs, stdout=PIPE, **kwargs)
473 output, unused_err = process.communicate()
474 retcode = process.poll()
475 if retcode:
476 cmd = kwargs.get("args")
477 if cmd is None:
478 cmd = popenargs[0]
479 raise CalledProcessError(retcode, cmd, output=output)
480 return output
Peter Astrand454f7672005-01-01 09:36:35 +0000481
482
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000483def list2cmdline(seq):
484 """
485 Translate a sequence of arguments into a command line
486 string, using the same rules as the MS C runtime:
487
488 1) Arguments are delimited by white space, which is either a
489 space or a tab.
490
491 2) A string surrounded by double quotation marks is
492 interpreted as a single argument, regardless of white space
Christian Heimesfdab48e2008-01-20 09:06:41 +0000493 or pipe characters contained within. A quoted string can be
494 embedded in an argument.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000495
496 3) A double quotation mark preceded by a backslash is
497 interpreted as a literal double quotation mark.
498
499 4) Backslashes are interpreted literally, unless they
500 immediately precede a double quotation mark.
501
502 5) If backslashes immediately precede a double quotation mark,
503 every pair of backslashes is interpreted as a literal
504 backslash. If the number of backslashes is odd, the last
505 backslash escapes the next double quotation mark as
506 described in rule 3.
507 """
508
509 # See
Eric Smith536d2992009-11-09 15:24:55 +0000510 # http://msdn.microsoft.com/en-us/library/17w5ykft.aspx
511 # or search http://msdn.microsoft.com for
512 # "Parsing C++ Command-Line Arguments"
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000513 result = []
514 needquote = False
515 for arg in seq:
516 bs_buf = []
517
518 # Add a space to separate this argument from the others
519 if result:
520 result.append(' ')
521
Christian Heimesfdab48e2008-01-20 09:06:41 +0000522 needquote = (" " in arg) or ("\t" in arg) or ("|" in arg) or not arg
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000523 if needquote:
524 result.append('"')
525
526 for c in arg:
527 if c == '\\':
Tim Peterse718f612004-10-12 21:51:32 +0000528 # Don't know if we need to double yet.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000529 bs_buf.append(c)
530 elif c == '"':
Christian Heimesfdab48e2008-01-20 09:06:41 +0000531 # Double backslashes.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000532 result.append('\\' * len(bs_buf)*2)
533 bs_buf = []
534 result.append('\\"')
535 else:
536 # Normal char
537 if bs_buf:
538 result.extend(bs_buf)
539 bs_buf = []
540 result.append(c)
541
Christian Heimesfdab48e2008-01-20 09:06:41 +0000542 # Add remaining backslashes, if any.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000543 if bs_buf:
544 result.extend(bs_buf)
545
546 if needquote:
Peter Astrand7e78ade2005-03-03 21:10:23 +0000547 result.extend(bs_buf)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000548 result.append('"')
549
550 return ''.join(result)
551
552
Brett Cannona23810f2008-05-26 19:04:21 +0000553# Various tools for executing commands and looking at their output and status.
554#
555# NB This only works (and is only relevant) for UNIX.
556
557def getstatusoutput(cmd):
558 """Return (status, output) of executing cmd in a shell.
559
560 Execute the string 'cmd' in a shell with os.popen() and return a 2-tuple
561 (status, output). cmd is actually run as '{ cmd ; } 2>&1', so that the
562 returned output will contain output or error messages. A trailing newline
563 is stripped from the output. The exit status for the command can be
564 interpreted according to the rules for the C function wait(). Example:
565
566 >>> import subprocess
567 >>> subprocess.getstatusoutput('ls /bin/ls')
568 (0, '/bin/ls')
569 >>> subprocess.getstatusoutput('cat /bin/junk')
570 (256, 'cat: /bin/junk: No such file or directory')
571 >>> subprocess.getstatusoutput('/bin/junk')
572 (256, 'sh: /bin/junk: not found')
573 """
574 pipe = os.popen('{ ' + cmd + '; } 2>&1', 'r')
575 text = pipe.read()
576 sts = pipe.close()
577 if sts is None: sts = 0
578 if text[-1:] == '\n': text = text[:-1]
579 return sts, text
580
581
582def getoutput(cmd):
583 """Return output (stdout or stderr) of executing cmd in a shell.
584
585 Like getstatusoutput(), except the exit status is ignored and the return
586 value is a string containing the command's output. Example:
587
588 >>> import subprocess
589 >>> subprocess.getoutput('ls /bin/ls')
590 '/bin/ls'
591 """
592 return getstatusoutput(cmd)[1]
593
594
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000595class Popen(object):
596 def __init__(self, args, bufsize=0, executable=None,
597 stdin=None, stdout=None, stderr=None,
598 preexec_fn=None, close_fds=False, shell=False,
599 cwd=None, env=None, universal_newlines=False,
600 startupinfo=None, creationflags=0):
601 """Create new Popen instance."""
602 _cleanup()
603
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000604 self._child_created = False
Guido van Rossum46a05a72007-06-07 21:56:45 +0000605 if bufsize is None:
606 bufsize = 0 # Restore default
Walter Dörwaldaa97f042007-05-03 21:05:51 +0000607 if not isinstance(bufsize, int):
Peter Astrand738131d2004-11-30 21:04:45 +0000608 raise TypeError("bufsize must be an integer")
609
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000610 if mswindows:
Tim Peterse8374a52004-10-13 03:15:00 +0000611 if preexec_fn is not None:
612 raise ValueError("preexec_fn is not supported on Windows "
613 "platforms")
Guido van Rossume7ba4952007-06-06 23:52:48 +0000614 if close_fds and (stdin is not None or stdout is not None or
615 stderr is not None):
Tim Peterse8374a52004-10-13 03:15:00 +0000616 raise ValueError("close_fds is not supported on Windows "
Guido van Rossume7ba4952007-06-06 23:52:48 +0000617 "platforms if you redirect stdin/stdout/stderr")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000618 else:
619 # POSIX
Tim Peterse8374a52004-10-13 03:15:00 +0000620 if startupinfo is not None:
621 raise ValueError("startupinfo is only supported on Windows "
622 "platforms")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000623 if creationflags != 0:
Tim Peterse8374a52004-10-13 03:15:00 +0000624 raise ValueError("creationflags is only supported on Windows "
625 "platforms")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000626
Tim Peterse718f612004-10-12 21:51:32 +0000627 self.stdin = None
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000628 self.stdout = None
629 self.stderr = None
630 self.pid = None
631 self.returncode = None
632 self.universal_newlines = universal_newlines
633
634 # Input and output objects. The general principle is like
635 # this:
636 #
637 # Parent Child
638 # ------ -----
639 # p2cwrite ---stdin---> p2cread
640 # c2pread <--stdout--- c2pwrite
641 # errread <--stderr--- errwrite
642 #
643 # On POSIX, the child objects are file descriptors. On
644 # Windows, these are Windows file handles. The parent objects
645 # are file descriptors on both platforms. The parent objects
646 # are None when not using PIPEs. The child objects are None
647 # when not redirecting.
Tim Peterse718f612004-10-12 21:51:32 +0000648
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000649 (p2cread, p2cwrite,
650 c2pread, c2pwrite,
651 errread, errwrite) = self._get_handles(stdin, stdout, stderr)
652
653 self._execute_child(args, executable, preexec_fn, close_fds,
654 cwd, env, universal_newlines,
655 startupinfo, creationflags, shell,
656 p2cread, p2cwrite,
657 c2pread, c2pwrite,
658 errread, errwrite)
659
Thomas Wouterscf297e42007-02-23 15:07:44 +0000660 if mswindows:
Hirokazu Yamamoto0c988172009-03-03 22:41:26 +0000661 if p2cwrite is not None:
662 p2cwrite = msvcrt.open_osfhandle(p2cwrite.Detach(), 0)
663 if c2pread is not None:
664 c2pread = msvcrt.open_osfhandle(c2pread.Detach(), 0)
665 if errread is not None:
666 errread = msvcrt.open_osfhandle(errread.Detach(), 0)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000667
Guido van Rossumc2f93dc2007-05-24 00:50:02 +0000668 if bufsize == 0:
669 bufsize = 1 # Nearly unbuffered (XXX for now)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000670 if p2cwrite is not None:
Guido van Rossumfa0054a2007-05-24 04:05:35 +0000671 self.stdin = io.open(p2cwrite, 'wb', bufsize)
672 if self.universal_newlines:
673 self.stdin = io.TextIOWrapper(self.stdin)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000674 if c2pread is not None:
Guido van Rossumfa0054a2007-05-24 04:05:35 +0000675 self.stdout = io.open(c2pread, 'rb', bufsize)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000676 if universal_newlines:
Guido van Rossumfa0054a2007-05-24 04:05:35 +0000677 self.stdout = io.TextIOWrapper(self.stdout)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000678 if errread is not None:
Guido van Rossumfa0054a2007-05-24 04:05:35 +0000679 self.stderr = io.open(errread, 'rb', bufsize)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000680 if universal_newlines:
Guido van Rossumfa0054a2007-05-24 04:05:35 +0000681 self.stderr = io.TextIOWrapper(self.stderr)
Tim Peterse718f612004-10-12 21:51:32 +0000682
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000683
Guido van Rossum98297ee2007-11-06 21:34:58 +0000684 def _translate_newlines(self, data, encoding):
685 data = data.replace(b"\r\n", b"\n").replace(b"\r", b"\n")
686 return data.decode(encoding)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000687
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000688
Guido van Rossumd8faa362007-04-27 19:54:29 +0000689 def __del__(self, sys=sys):
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000690 if not self._child_created:
691 # We didn't get to successfully create a child process.
692 return
693 # In case the child hasn't been waited on, check if it's done.
Georg Brandl6aa2d1f2008-08-12 08:35:52 +0000694 self._internal_poll(_deadstate=sys.maxsize)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000695 if self.returncode is None and _active is not None:
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000696 # Child is still running, keep us alive until we can wait on it.
697 _active.append(self)
698
699
Peter Astrand23109f02005-03-03 20:28:59 +0000700 def communicate(self, input=None):
701 """Interact with process: Send data to stdin. Read data from
702 stdout and stderr, until end-of-file is reached. Wait for
703 process to terminate. The optional input argument should be a
704 string to be sent to the child process, or None, if no data
705 should be sent to the child.
Tim Peterseba28be2005-03-28 01:08:02 +0000706
Peter Astrand23109f02005-03-03 20:28:59 +0000707 communicate() returns a tuple (stdout, stderr)."""
708
709 # Optimization: If we are only using one pipe, or no pipe at
710 # all, using select() or threads is unnecessary.
711 if [self.stdin, self.stdout, self.stderr].count(None) >= 2:
Tim Peterseba28be2005-03-28 01:08:02 +0000712 stdout = None
713 stderr = None
Peter Astrand23109f02005-03-03 20:28:59 +0000714 if self.stdin:
715 if input:
716 self.stdin.write(input)
717 self.stdin.close()
718 elif self.stdout:
719 stdout = self.stdout.read()
Georg Brandlf08a9dd2008-06-10 16:57:31 +0000720 self.stdout.close()
Peter Astrand23109f02005-03-03 20:28:59 +0000721 elif self.stderr:
722 stderr = self.stderr.read()
Georg Brandlf08a9dd2008-06-10 16:57:31 +0000723 self.stderr.close()
Peter Astrand23109f02005-03-03 20:28:59 +0000724 self.wait()
725 return (stdout, stderr)
Tim Peterseba28be2005-03-28 01:08:02 +0000726
Peter Astrand23109f02005-03-03 20:28:59 +0000727 return self._communicate(input)
728
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000729
Georg Brandl6aa2d1f2008-08-12 08:35:52 +0000730 def poll(self):
731 return self._internal_poll()
732
733
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000734 if mswindows:
735 #
736 # Windows methods
737 #
738 def _get_handles(self, stdin, stdout, stderr):
Georg Brandla85ee5c2009-08-13 12:13:42 +0000739 """Construct and return tuple with IO objects:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000740 p2cread, p2cwrite, c2pread, c2pwrite, errread, errwrite
741 """
Peter Astrandd38ddf42005-02-10 08:32:50 +0000742 if stdin is None and stdout is None and stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000743 return (None, None, None, None, None, None)
Tim Peterse718f612004-10-12 21:51:32 +0000744
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000745 p2cread, p2cwrite = None, None
746 c2pread, c2pwrite = None, None
747 errread, errwrite = None, None
748
Peter Astrandd38ddf42005-02-10 08:32:50 +0000749 if stdin is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000750 p2cread = GetStdHandle(STD_INPUT_HANDLE)
Hirokazu Yamamoto0c988172009-03-03 22:41:26 +0000751 if p2cread is None:
752 p2cread, _ = CreatePipe(None, 0)
753 elif stdin == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000754 p2cread, p2cwrite = CreatePipe(None, 0)
Peter Astrandd38ddf42005-02-10 08:32:50 +0000755 elif isinstance(stdin, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000756 p2cread = msvcrt.get_osfhandle(stdin)
757 else:
758 # Assuming file-like object
759 p2cread = msvcrt.get_osfhandle(stdin.fileno())
760 p2cread = self._make_inheritable(p2cread)
761
Peter Astrandd38ddf42005-02-10 08:32:50 +0000762 if stdout is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000763 c2pwrite = GetStdHandle(STD_OUTPUT_HANDLE)
Hirokazu Yamamoto0c988172009-03-03 22:41:26 +0000764 if c2pwrite is None:
765 _, c2pwrite = CreatePipe(None, 0)
766 elif stdout == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000767 c2pread, c2pwrite = CreatePipe(None, 0)
Peter Astrandd38ddf42005-02-10 08:32:50 +0000768 elif isinstance(stdout, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000769 c2pwrite = msvcrt.get_osfhandle(stdout)
770 else:
771 # Assuming file-like object
772 c2pwrite = msvcrt.get_osfhandle(stdout.fileno())
773 c2pwrite = self._make_inheritable(c2pwrite)
774
Peter Astrandd38ddf42005-02-10 08:32:50 +0000775 if stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000776 errwrite = GetStdHandle(STD_ERROR_HANDLE)
Hirokazu Yamamoto0c988172009-03-03 22:41:26 +0000777 if errwrite is None:
778 _, errwrite = CreatePipe(None, 0)
779 elif stderr == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000780 errread, errwrite = CreatePipe(None, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000781 elif stderr == STDOUT:
782 errwrite = c2pwrite
Peter Astrandd38ddf42005-02-10 08:32:50 +0000783 elif isinstance(stderr, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000784 errwrite = msvcrt.get_osfhandle(stderr)
785 else:
786 # Assuming file-like object
787 errwrite = msvcrt.get_osfhandle(stderr.fileno())
788 errwrite = self._make_inheritable(errwrite)
789
790 return (p2cread, p2cwrite,
791 c2pread, c2pwrite,
792 errread, errwrite)
793
794
795 def _make_inheritable(self, handle):
796 """Return a duplicate of handle, which is inheritable"""
797 return DuplicateHandle(GetCurrentProcess(), handle,
798 GetCurrentProcess(), 0, 1,
799 DUPLICATE_SAME_ACCESS)
800
801
802 def _find_w9xpopen(self):
803 """Find and return absolut path to w9xpopen.exe"""
Tim Peterse8374a52004-10-13 03:15:00 +0000804 w9xpopen = os.path.join(os.path.dirname(GetModuleFileName(0)),
805 "w9xpopen.exe")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000806 if not os.path.exists(w9xpopen):
807 # Eeek - file-not-found - possibly an embedding
808 # situation - see if we can locate it in sys.exec_prefix
Tim Peterse8374a52004-10-13 03:15:00 +0000809 w9xpopen = os.path.join(os.path.dirname(sys.exec_prefix),
810 "w9xpopen.exe")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000811 if not os.path.exists(w9xpopen):
Tim Peterse8374a52004-10-13 03:15:00 +0000812 raise RuntimeError("Cannot locate w9xpopen.exe, which is "
813 "needed for Popen to work with your "
814 "shell or platform.")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000815 return w9xpopen
816
Tim Peterse718f612004-10-12 21:51:32 +0000817
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000818 def _execute_child(self, args, executable, preexec_fn, close_fds,
819 cwd, env, universal_newlines,
820 startupinfo, creationflags, shell,
821 p2cread, p2cwrite,
822 c2pread, c2pwrite,
823 errread, errwrite):
824 """Execute program (MS Windows version)"""
825
Guido van Rossum3172c5d2007-10-16 18:12:55 +0000826 if not isinstance(args, str):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000827 args = list2cmdline(args)
828
Peter Astrandc1d65362004-11-07 14:30:34 +0000829 # Process startup details
Peter Astrandd38ddf42005-02-10 08:32:50 +0000830 if startupinfo is None:
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000831 startupinfo = STARTUPINFO()
832 if None not in (p2cread, c2pwrite, errwrite):
Peter Astrandc1d65362004-11-07 14:30:34 +0000833 startupinfo.dwFlags |= STARTF_USESTDHANDLES
834 startupinfo.hStdInput = p2cread
835 startupinfo.hStdOutput = c2pwrite
836 startupinfo.hStdError = errwrite
837
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000838 if shell:
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000839 startupinfo.dwFlags |= STARTF_USESHOWWINDOW
840 startupinfo.wShowWindow = SW_HIDE
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000841 comspec = os.environ.get("COMSPEC", "cmd.exe")
842 args = comspec + " /c " + args
Guido van Rossume2a383d2007-01-15 16:59:06 +0000843 if (GetVersion() >= 0x80000000 or
Tim Peterse8374a52004-10-13 03:15:00 +0000844 os.path.basename(comspec).lower() == "command.com"):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000845 # Win9x, or using command.com on NT. We need to
846 # use the w9xpopen intermediate program. For more
847 # information, see KB Q150956
848 # (http://web.archive.org/web/20011105084002/http://support.microsoft.com/support/kb/articles/Q150/9/56.asp)
849 w9xpopen = self._find_w9xpopen()
850 args = '"%s" %s' % (w9xpopen, args)
851 # Not passing CREATE_NEW_CONSOLE has been known to
852 # cause random failures on win9x. Specifically a
853 # dialog: "Your program accessed mem currently in
854 # use at xxx" and a hopeful warning about the
Mark Dickinson934896d2009-02-21 20:59:32 +0000855 # stability of your system. Cost is Ctrl+C won't
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000856 # kill children.
857 creationflags |= CREATE_NEW_CONSOLE
858
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000859 # Start the process
860 try:
861 hp, ht, pid, tid = CreateProcess(executable, args,
Tim Peterse8374a52004-10-13 03:15:00 +0000862 # no special security
863 None, None,
Guido van Rossume7ba4952007-06-06 23:52:48 +0000864 int(not close_fds),
Tim Peterse8374a52004-10-13 03:15:00 +0000865 creationflags,
866 env,
867 cwd,
868 startupinfo)
Guido van Rossumb940e112007-01-10 16:19:56 +0000869 except pywintypes.error as e:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000870 # Translate pywintypes.error to WindowsError, which is
871 # a subclass of OSError. FIXME: We should really
872 # translate errno using _sys_errlist (or simliar), but
873 # how can this be done from Python?
874 raise WindowsError(*e.args)
875
876 # Retain the process handle, but close the thread handle
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000877 self._child_created = True
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000878 self._handle = hp
879 self.pid = pid
880 ht.Close()
881
Andrew M. Kuchling51ee66e2004-10-12 16:38:42 +0000882 # Child is launched. Close the parent's copy of those pipe
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000883 # handles that only the child should have open. You need
884 # to make sure that no handles to the write end of the
885 # output pipe are maintained in this process or else the
886 # pipe will not close when the child process exits and the
887 # ReadFile will hang.
Peter Astrandd38ddf42005-02-10 08:32:50 +0000888 if p2cread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000889 p2cread.Close()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000890 if c2pwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000891 c2pwrite.Close()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000892 if errwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000893 errwrite.Close()
894
Tim Peterse718f612004-10-12 21:51:32 +0000895
Georg Brandl6aa2d1f2008-08-12 08:35:52 +0000896 def _internal_poll(self, _deadstate=None):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000897 """Check if child process has terminated. Returns returncode
898 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +0000899 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000900 if WaitForSingleObject(self._handle, 0) == WAIT_OBJECT_0:
901 self.returncode = GetExitCodeProcess(self._handle)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000902 return self.returncode
903
904
905 def wait(self):
906 """Wait for child process to terminate. Returns returncode
907 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +0000908 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000909 obj = WaitForSingleObject(self._handle, INFINITE)
910 self.returncode = GetExitCodeProcess(self._handle)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000911 return self.returncode
912
913
914 def _readerthread(self, fh, buffer):
915 buffer.append(fh.read())
916
917
Peter Astrand23109f02005-03-03 20:28:59 +0000918 def _communicate(self, input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000919 stdout = None # Return
920 stderr = None # Return
921
922 if self.stdout:
923 stdout = []
Tim Peterse8374a52004-10-13 03:15:00 +0000924 stdout_thread = threading.Thread(target=self._readerthread,
925 args=(self.stdout, stdout))
Benjamin Peterson632e0362008-08-18 19:08:51 +0000926 stdout_thread.daemon = True
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000927 stdout_thread.start()
928 if self.stderr:
929 stderr = []
Tim Peterse8374a52004-10-13 03:15:00 +0000930 stderr_thread = threading.Thread(target=self._readerthread,
931 args=(self.stderr, stderr))
Benjamin Peterson632e0362008-08-18 19:08:51 +0000932 stderr_thread.daemon = True
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000933 stderr_thread.start()
934
935 if self.stdin:
Peter Astrandd38ddf42005-02-10 08:32:50 +0000936 if input is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000937 self.stdin.write(input)
938 self.stdin.close()
939
940 if self.stdout:
941 stdout_thread.join()
942 if self.stderr:
943 stderr_thread.join()
944
945 # All data exchanged. Translate lists into strings.
Peter Astrandd38ddf42005-02-10 08:32:50 +0000946 if stdout is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000947 stdout = stdout[0]
Peter Astrandd38ddf42005-02-10 08:32:50 +0000948 if stderr is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000949 stderr = stderr[0]
950
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000951 self.wait()
952 return (stdout, stderr)
953
Christian Heimesa342c012008-04-20 21:01:16 +0000954 def send_signal(self, sig):
955 """Send a signal to the process
956 """
957 if sig == signal.SIGTERM:
958 self.terminate()
959 else:
960 raise ValueError("Only SIGTERM is supported on Windows")
961
962 def terminate(self):
963 """Terminates the process
964 """
965 TerminateProcess(self._handle, 1)
966
967 kill = terminate
968
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000969 else:
970 #
971 # POSIX methods
972 #
973 def _get_handles(self, stdin, stdout, stderr):
Georg Brandla85ee5c2009-08-13 12:13:42 +0000974 """Construct and return tuple with IO objects:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000975 p2cread, p2cwrite, c2pread, c2pwrite, errread, errwrite
976 """
977 p2cread, p2cwrite = None, None
978 c2pread, c2pwrite = None, None
979 errread, errwrite = None, None
980
Peter Astrandd38ddf42005-02-10 08:32:50 +0000981 if stdin is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000982 pass
983 elif stdin == PIPE:
984 p2cread, p2cwrite = os.pipe()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000985 elif isinstance(stdin, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000986 p2cread = stdin
987 else:
988 # Assuming file-like object
989 p2cread = stdin.fileno()
990
Peter Astrandd38ddf42005-02-10 08:32:50 +0000991 if stdout is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000992 pass
993 elif stdout == PIPE:
994 c2pread, c2pwrite = os.pipe()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000995 elif isinstance(stdout, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000996 c2pwrite = stdout
997 else:
998 # Assuming file-like object
Tim Peterse718f612004-10-12 21:51:32 +0000999 c2pwrite = stdout.fileno()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001000
Peter Astrandd38ddf42005-02-10 08:32:50 +00001001 if stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001002 pass
1003 elif stderr == PIPE:
1004 errread, errwrite = os.pipe()
1005 elif stderr == STDOUT:
1006 errwrite = c2pwrite
Peter Astrandd38ddf42005-02-10 08:32:50 +00001007 elif isinstance(stderr, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001008 errwrite = stderr
1009 else:
1010 # Assuming file-like object
1011 errwrite = stderr.fileno()
1012
1013 return (p2cread, p2cwrite,
1014 c2pread, c2pwrite,
1015 errread, errwrite)
1016
1017
1018 def _set_cloexec_flag(self, fd):
1019 try:
1020 cloexec_flag = fcntl.FD_CLOEXEC
1021 except AttributeError:
1022 cloexec_flag = 1
1023
1024 old = fcntl.fcntl(fd, fcntl.F_GETFD)
1025 fcntl.fcntl(fd, fcntl.F_SETFD, old | cloexec_flag)
1026
1027
1028 def _close_fds(self, but):
Christian Heimesfdab48e2008-01-20 09:06:41 +00001029 os.closerange(3, but)
1030 os.closerange(but + 1, MAXFD)
Tim Peterse718f612004-10-12 21:51:32 +00001031
1032
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001033 def _execute_child(self, args, executable, preexec_fn, close_fds,
1034 cwd, env, universal_newlines,
1035 startupinfo, creationflags, shell,
1036 p2cread, p2cwrite,
1037 c2pread, c2pwrite,
1038 errread, errwrite):
1039 """Execute program (POSIX version)"""
1040
Guido van Rossum3172c5d2007-10-16 18:12:55 +00001041 if isinstance(args, str):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001042 args = [args]
Thomas Wouters89f507f2006-12-13 04:49:30 +00001043 else:
1044 args = list(args)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001045
1046 if shell:
1047 args = ["/bin/sh", "-c"] + args
1048
Peter Astrandd38ddf42005-02-10 08:32:50 +00001049 if executable is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001050 executable = args[0]
1051
1052 # For transferring possible exec failure from child to parent
1053 # The first char specifies the exception type: 0 means
1054 # OSError, 1 means some other error.
1055 errpipe_read, errpipe_write = os.pipe()
Christian Heimesfdab48e2008-01-20 09:06:41 +00001056 try:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001057 try:
Facundo Batista10706e22009-06-19 20:34:30 +00001058 self._set_cloexec_flag(errpipe_write)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001059
Facundo Batista10706e22009-06-19 20:34:30 +00001060 gc_was_enabled = gc.isenabled()
1061 # Disable gc to avoid bug where gc -> file_dealloc ->
1062 # write to stderr -> hang. http://bugs.python.org/issue1336
1063 gc.disable()
1064 try:
1065 self.pid = os.fork()
1066 except:
1067 if gc_was_enabled:
1068 gc.enable()
1069 raise
1070 self._child_created = True
1071 if self.pid == 0:
1072 # Child
1073 try:
1074 # Close parent's pipe ends
1075 if p2cwrite is not None:
1076 os.close(p2cwrite)
1077 if c2pread is not None:
1078 os.close(c2pread)
1079 if errread is not None:
1080 os.close(errread)
1081 os.close(errpipe_read)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001082
Facundo Batista10706e22009-06-19 20:34:30 +00001083 # Dup fds for child
1084 if p2cread is not None:
1085 os.dup2(p2cread, 0)
1086 if c2pwrite is not None:
1087 os.dup2(c2pwrite, 1)
1088 if errwrite is not None:
1089 os.dup2(errwrite, 2)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001090
Facundo Batista10706e22009-06-19 20:34:30 +00001091 # Close pipe fds. Make sure we don't close the
1092 # same fd more than once, or standard fds.
1093 if p2cread is not None and p2cread not in (0,):
1094 os.close(p2cread)
1095 if c2pwrite is not None and \
1096 c2pwrite not in (p2cread, 1):
1097 os.close(c2pwrite)
1098 if (errwrite is not None and
1099 errwrite not in (p2cread, c2pwrite, 2)):
1100 os.close(errwrite)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001101
Facundo Batista10706e22009-06-19 20:34:30 +00001102 # Close all other fds, if asked for
1103 if close_fds:
1104 self._close_fds(but=errpipe_write)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001105
Facundo Batista10706e22009-06-19 20:34:30 +00001106 if cwd is not None:
1107 os.chdir(cwd)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001108
Facundo Batista10706e22009-06-19 20:34:30 +00001109 if preexec_fn:
1110 preexec_fn()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001111
Facundo Batista10706e22009-06-19 20:34:30 +00001112 if env is None:
1113 os.execvp(executable, args)
1114 else:
1115 os.execvpe(executable, args, env)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001116
Facundo Batista10706e22009-06-19 20:34:30 +00001117 except:
1118 exc_type, exc_value, tb = sys.exc_info()
1119 # Save the traceback and attach it to the exception
1120 # object
1121 exc_lines = traceback.format_exception(exc_type,
1122 exc_value,
1123 tb)
1124 exc_value.child_traceback = ''.join(exc_lines)
1125 os.write(errpipe_write, pickle.dumps(exc_value))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001126
Facundo Batista10706e22009-06-19 20:34:30 +00001127 # This exitcode won't be reported to applications, so
1128 # it really doesn't matter what we return.
1129 os._exit(255)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001130
Facundo Batista10706e22009-06-19 20:34:30 +00001131 # Parent
1132 if gc_was_enabled:
1133 gc.enable()
1134 finally:
1135 # be sure the FD is closed no matter what
1136 os.close(errpipe_write)
1137
1138 if p2cread is not None and p2cwrite is not None:
1139 os.close(p2cread)
1140 if c2pwrite is not None and c2pread is not None:
1141 os.close(c2pwrite)
1142 if errwrite is not None and errread is not None:
1143 os.close(errwrite)
1144
1145 # Wait for exec to fail or succeed; possibly raising an
1146 # exception (limited to 1 MB)
Gregory P. Smith3fff44d2010-03-01 00:43:08 +00001147 data = _eintr_retry_call(os.read, errpipe_read, 1048576)
Facundo Batista10706e22009-06-19 20:34:30 +00001148 finally:
1149 # be sure the FD is closed no matter what
1150 os.close(errpipe_read)
1151
Guido van Rossumaf2362a2007-05-15 22:32:02 +00001152 if data:
Gregory P. Smith3fff44d2010-03-01 00:43:08 +00001153 _eintr_retry_call(os.waitpid, self.pid, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001154 child_exception = pickle.loads(data)
Benjamin Petersond75fcb42009-02-19 04:22:03 +00001155 for fd in (p2cwrite, c2pread, errread):
1156 if fd is not None:
1157 os.close(fd)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001158 raise child_exception
1159
1160
1161 def _handle_exitstatus(self, sts):
1162 if os.WIFSIGNALED(sts):
1163 self.returncode = -os.WTERMSIG(sts)
1164 elif os.WIFEXITED(sts):
1165 self.returncode = os.WEXITSTATUS(sts)
1166 else:
1167 # Should never happen
1168 raise RuntimeError("Unknown child exit status!")
1169
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001170
Georg Brandl6aa2d1f2008-08-12 08:35:52 +00001171 def _internal_poll(self, _deadstate=None):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001172 """Check if child process has terminated. Returns returncode
1173 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +00001174 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001175 try:
1176 pid, sts = os.waitpid(self.pid, os.WNOHANG)
1177 if pid == self.pid:
1178 self._handle_exitstatus(sts)
1179 except os.error:
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001180 if _deadstate is not None:
1181 self.returncode = _deadstate
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001182 return self.returncode
1183
1184
1185 def wait(self):
1186 """Wait for child process to terminate. Returns returncode
1187 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +00001188 if self.returncode is None:
Gregory P. Smith3fff44d2010-03-01 00:43:08 +00001189 pid, sts = _eintr_retry_call(os.waitpid, self.pid, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001190 self._handle_exitstatus(sts)
1191 return self.returncode
1192
1193
Peter Astrand23109f02005-03-03 20:28:59 +00001194 def _communicate(self, input):
Georg Brandlae83d6e2009-08-13 09:04:31 +00001195 if self.stdin:
1196 # Flush stdio buffer. This might block, if the user has
1197 # been writing to .stdin in an uncontrolled fashion.
1198 self.stdin.flush()
1199 if not input:
1200 self.stdin.close()
1201
1202 if _has_poll:
1203 stdout, stderr = self._communicate_with_poll(input)
1204 else:
1205 stdout, stderr = self._communicate_with_select(input)
1206
1207 # All data exchanged. Translate lists into strings.
1208 if stdout is not None:
1209 stdout = b''.join(stdout)
1210 if stderr is not None:
1211 stderr = b''.join(stderr)
1212
1213 # Translate newlines, if requested.
1214 # This also turns bytes into strings.
1215 if self.universal_newlines:
1216 if stdout is not None:
1217 stdout = self._translate_newlines(stdout,
1218 self.stdout.encoding)
1219 if stderr is not None:
1220 stderr = self._translate_newlines(stderr,
1221 self.stderr.encoding)
1222
1223 self.wait()
1224 return (stdout, stderr)
1225
1226
1227 def _communicate_with_poll(self, input):
1228 stdout = None # Return
1229 stderr = None # Return
1230 fd2file = {}
1231 fd2output = {}
1232
1233 poller = select.poll()
1234 def register_and_append(file_obj, eventmask):
1235 poller.register(file_obj.fileno(), eventmask)
1236 fd2file[file_obj.fileno()] = file_obj
1237
1238 def close_unregister_and_remove(fd):
1239 poller.unregister(fd)
1240 fd2file[fd].close()
1241 fd2file.pop(fd)
1242
1243 if self.stdin and input:
1244 register_and_append(self.stdin, select.POLLOUT)
1245
1246 select_POLLIN_POLLPRI = select.POLLIN | select.POLLPRI
1247 if self.stdout:
1248 register_and_append(self.stdout, select_POLLIN_POLLPRI)
1249 fd2output[self.stdout.fileno()] = stdout = []
1250 if self.stderr:
1251 register_and_append(self.stderr, select_POLLIN_POLLPRI)
1252 fd2output[self.stderr.fileno()] = stderr = []
1253
1254 input_offset = 0
1255 while fd2file:
1256 try:
1257 ready = poller.poll()
1258 except select.error as e:
1259 if e.args[0] == errno.EINTR:
1260 continue
1261 raise
1262
1263 # XXX Rewrite these to use non-blocking I/O on the
1264 # file objects; they are no longer using C stdio!
1265
1266 for fd, mode in ready:
1267 if mode & select.POLLOUT:
1268 chunk = input[input_offset : input_offset + _PIPE_BUF]
1269 input_offset += os.write(fd, chunk)
1270 if input_offset >= len(input):
1271 close_unregister_and_remove(fd)
1272 elif mode & select_POLLIN_POLLPRI:
1273 data = os.read(fd, 4096)
1274 if not data:
1275 close_unregister_and_remove(fd)
1276 fd2output[fd].append(data)
1277 else:
1278 # Ignore hang up or errors.
1279 close_unregister_and_remove(fd)
1280
1281 return (stdout, stderr)
1282
1283
1284 def _communicate_with_select(self, input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001285 read_set = []
1286 write_set = []
1287 stdout = None # Return
1288 stderr = None # Return
1289
Georg Brandlae83d6e2009-08-13 09:04:31 +00001290 if self.stdin and input:
1291 write_set.append(self.stdin)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001292 if self.stdout:
1293 read_set.append(self.stdout)
Tim Peterse718f612004-10-12 21:51:32 +00001294 stdout = []
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001295 if self.stderr:
1296 read_set.append(self.stderr)
1297 stderr = []
1298
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001299 input_offset = 0
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001300 while read_set or write_set:
Georg Brandl86b2fb92008-07-16 03:43:04 +00001301 try:
1302 rlist, wlist, xlist = select.select(read_set, write_set, [])
1303 except select.error as e:
1304 if e.args[0] == errno.EINTR:
1305 continue
1306 raise
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001307
Guido van Rossum98297ee2007-11-06 21:34:58 +00001308 # XXX Rewrite these to use non-blocking I/O on the
1309 # file objects; they are no longer using C stdio!
1310
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001311 if self.stdin in wlist:
Georg Brandlae83d6e2009-08-13 09:04:31 +00001312 chunk = input[input_offset : input_offset + _PIPE_BUF]
Guido van Rossumbae07c92007-10-08 02:46:15 +00001313 bytes_written = os.write(self.stdin.fileno(), chunk)
Thomas Wouters9fe394c2007-02-05 01:24:16 +00001314 input_offset += bytes_written
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001315 if input_offset >= len(input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001316 self.stdin.close()
1317 write_set.remove(self.stdin)
1318
1319 if self.stdout in rlist:
1320 data = os.read(self.stdout.fileno(), 1024)
Guido van Rossumc9e363c2007-05-15 23:18:55 +00001321 if not data:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001322 self.stdout.close()
1323 read_set.remove(self.stdout)
1324 stdout.append(data)
1325
1326 if self.stderr in rlist:
1327 data = os.read(self.stderr.fileno(), 1024)
Guido van Rossumc9e363c2007-05-15 23:18:55 +00001328 if not data:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001329 self.stderr.close()
1330 read_set.remove(self.stderr)
1331 stderr.append(data)
1332
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001333 return (stdout, stderr)
1334
Georg Brandlae83d6e2009-08-13 09:04:31 +00001335
Christian Heimesa342c012008-04-20 21:01:16 +00001336 def send_signal(self, sig):
1337 """Send a signal to the process
1338 """
1339 os.kill(self.pid, sig)
1340
1341 def terminate(self):
1342 """Terminate the process with SIGTERM
1343 """
1344 self.send_signal(signal.SIGTERM)
1345
1346 def kill(self):
1347 """Kill the process with SIGKILL
1348 """
1349 self.send_signal(signal.SIGKILL)
1350
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001351
1352def _demo_posix():
1353 #
1354 # Example 1: Simple redirection: Get process list
1355 #
1356 plist = Popen(["ps"], stdout=PIPE).communicate()[0]
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001357 print("Process list:")
1358 print(plist)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001359
1360 #
1361 # Example 2: Change uid before executing child
1362 #
1363 if os.getuid() == 0:
1364 p = Popen(["id"], preexec_fn=lambda: os.setuid(100))
1365 p.wait()
1366
1367 #
1368 # Example 3: Connecting several subprocesses
1369 #
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001370 print("Looking for 'hda'...")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001371 p1 = Popen(["dmesg"], stdout=PIPE)
1372 p2 = Popen(["grep", "hda"], stdin=p1.stdout, stdout=PIPE)
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001373 print(repr(p2.communicate()[0]))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001374
1375 #
1376 # Example 4: Catch execution error
1377 #
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001378 print()
1379 print("Trying a weird file...")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001380 try:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001381 print(Popen(["/this/path/does/not/exist"]).communicate())
Guido van Rossumb940e112007-01-10 16:19:56 +00001382 except OSError as e:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001383 if e.errno == errno.ENOENT:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001384 print("The file didn't exist. I thought so...")
1385 print("Child traceback:")
1386 print(e.child_traceback)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001387 else:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001388 print("Error", e.errno)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001389 else:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001390 print("Gosh. No error.", file=sys.stderr)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001391
1392
1393def _demo_windows():
1394 #
1395 # Example 1: Connecting several subprocesses
1396 #
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001397 print("Looking for 'PROMPT' in set output...")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001398 p1 = Popen("set", stdout=PIPE, shell=True)
1399 p2 = Popen('find "PROMPT"', stdin=p1.stdout, stdout=PIPE)
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001400 print(repr(p2.communicate()[0]))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001401
1402 #
1403 # Example 2: Simple execution of program
1404 #
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001405 print("Executing calc...")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001406 p = Popen("calc")
1407 p.wait()
1408
1409
1410if __name__ == "__main__":
1411 if mswindows:
1412 _demo_windows()
1413 else:
1414 _demo_posix()