blob: 2a4c1567aa830d9606eef2726ae7bd4b9be5eed4 [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 Astrandc26516b2005-02-21 08:13:02 +00005# This module should remain compatible with Python 2.2, see PEP 291.
6#
Peter Astrand3a708df2005-09-23 17:37:29 +00007# Copyright (c) 2003-2005 by Peter Astrand <astrand@lysator.liu.se>
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00008#
Peter Astrand69bf13f2005-02-14 08:56:32 +00009# Licensed to PSF under a Contributor Agreement.
Peter Astrand3a708df2005-09-23 17:37:29 +000010# See http://www.python.org/2.4/license for licensing details.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +000011
Raymond Hettinger837dd932004-10-17 16:36:53 +000012r"""subprocess - Subprocesses with accessible I/O streams
Fredrik Lundh5b3687d2004-10-12 15:26:28 +000013
Fredrik Lundh15aaacc2004-10-17 14:47:05 +000014This module allows you to spawn processes, connect to their
15input/output/error pipes, and obtain their return codes. This module
16intends to replace several other, older modules and functions, like:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +000017
18os.system
19os.spawn*
20os.popen*
21popen2.*
22commands.*
23
24Information about how the subprocess module can be used to replace these
25modules and functions can be found below.
26
27
28
29Using the subprocess module
30===========================
31This module defines one class called Popen:
32
33class Popen(args, bufsize=0, executable=None,
34 stdin=None, stdout=None, stderr=None,
35 preexec_fn=None, close_fds=False, shell=False,
36 cwd=None, env=None, universal_newlines=False,
37 startupinfo=None, creationflags=0):
38
39
40Arguments are:
41
42args should be a string, or a sequence of program arguments. The
43program to execute is normally the first item in the args sequence or
44string, but can be explicitly set by using the executable argument.
45
46On UNIX, with shell=False (default): In this case, the Popen class
47uses os.execvp() to execute the child program. args should normally
48be a sequence. A string will be treated as a sequence with the string
49as the only item (the program to execute).
50
51On UNIX, with shell=True: If args is a string, it specifies the
52command string to execute through the shell. If args is a sequence,
53the first item specifies the command string, and any additional items
54will be treated as additional shell arguments.
55
56On Windows: the Popen class uses CreateProcess() to execute the child
57program, which operates on strings. If args is a sequence, it will be
58converted to a string using the list2cmdline method. Please note that
59not all MS Windows applications interpret the command line the same
60way: The list2cmdline is designed for applications using the same
61rules as the MS C runtime.
62
63bufsize, if given, has the same meaning as the corresponding argument
64to the built-in open() function: 0 means unbuffered, 1 means line
65buffered, any other positive value means use a buffer of
66(approximately) that size. A negative bufsize means to use the system
67default, which usually means fully buffered. The default value for
68bufsize is 0 (unbuffered).
69
70stdin, stdout and stderr specify the executed programs' standard
71input, standard output and standard error file handles, respectively.
72Valid values are PIPE, an existing file descriptor (a positive
73integer), an existing file object, and None. PIPE indicates that a
74new pipe to the child should be created. With None, no redirection
75will occur; the child's file handles will be inherited from the
76parent. Additionally, stderr can be STDOUT, which indicates that the
77stderr data from the applications should be captured into the same
78file handle as for stdout.
79
80If preexec_fn is set to a callable object, this object will be called
81in the child process just before the child is executed.
82
83If close_fds is true, all file descriptors except 0, 1 and 2 will be
84closed before the child process is executed.
85
86if shell is true, the specified command will be executed through the
87shell.
88
89If cwd is not None, the current directory will be changed to cwd
90before the child is executed.
91
92If env is not None, it defines the environment variables for the new
93process.
94
95If universal_newlines is true, the file objects stdout and stderr are
96opened as a text files, but lines may be terminated by any of '\n',
97the Unix end-of-line convention, '\r', the Macintosh convention or
98'\r\n', the Windows convention. All of these external representations
99are seen as '\n' by the Python program. Note: This feature is only
100available if Python is built with universal newline support (the
101default). Also, the newlines attribute of the file objects stdout,
102stdin and stderr are not updated by the communicate() method.
103
104The startupinfo and creationflags, if given, will be passed to the
105underlying CreateProcess() function. They can specify things such as
106appearance of the main window and priority for the new process.
107(Windows only)
108
109
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000110This module also defines some shortcut functions:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000111
Peter Astrand5f5e1412004-12-05 20:15:36 +0000112call(*popenargs, **kwargs):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000113 Run command with arguments. Wait for command to complete, then
114 return the returncode attribute.
115
116 The arguments are the same as for the Popen constructor. Example:
117
118 retcode = call(["ls", "-l"])
119
Peter Astrand454f7672005-01-01 09:36:35 +0000120check_call(*popenargs, **kwargs):
121 Run command with arguments. Wait for command to complete. If the
122 exit code was zero then return, otherwise raise
123 CalledProcessError. The CalledProcessError object will have the
Peter Astrand7d1d4362006-07-14 14:04:45 +0000124 return code in the returncode attribute.
Peter Astrand454f7672005-01-01 09:36:35 +0000125
126 The arguments are the same as for the Popen constructor. Example:
127
128 check_call(["ls", "-l"])
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000129
Gregory P. Smith26576802008-12-05 02:27:01 +0000130check_output(*popenargs, **kwargs):
Georg Brandl6ab5d082009-12-20 14:33:20 +0000131 Run command with arguments and return its output as a byte string.
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000132
Georg Brandl6ab5d082009-12-20 14:33:20 +0000133 If the exit code was non-zero it raises a CalledProcessError. The
134 CalledProcessError object will have the return code in the returncode
135 attribute and output in the output attribute.
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000136
Georg Brandl6ab5d082009-12-20 14:33:20 +0000137 The arguments are the same as for the Popen constructor. Example:
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000138
Georg Brandl02e7dfd2009-12-28 08:09:32 +0000139 output = check_output(["ls", "-l", "/dev/null"])
140
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000141
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000142Exceptions
143----------
144Exceptions raised in the child process, before the new program has
145started to execute, will be re-raised in the parent. Additionally,
146the exception object will have one extra attribute called
147'child_traceback', which is a string containing traceback information
148from the childs point of view.
149
150The most common exception raised is OSError. This occurs, for
151example, when trying to execute a non-existent file. Applications
152should prepare for OSErrors.
153
154A ValueError will be raised if Popen is called with invalid arguments.
155
Gregory P. Smith26576802008-12-05 02:27:01 +0000156check_call() and check_output() will raise CalledProcessError, if the
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000157called process returns a non-zero return code.
Peter Astrand454f7672005-01-01 09:36:35 +0000158
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000159
160Security
161--------
162Unlike some other popen functions, this implementation will never call
163/bin/sh implicitly. This means that all characters, including shell
164metacharacters, can safely be passed to child processes.
165
166
167Popen objects
168=============
169Instances of the Popen class have the following methods:
170
171poll()
172 Check if child process has terminated. Returns returncode
173 attribute.
174
175wait()
176 Wait for child process to terminate. Returns returncode attribute.
177
178communicate(input=None)
179 Interact with process: Send data to stdin. Read data from stdout
180 and stderr, until end-of-file is reached. Wait for process to
Neal Norwitza186ee22006-12-29 03:01:53 +0000181 terminate. The optional input argument should be a string to be
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000182 sent to the child process, or None, if no data should be sent to
183 the child.
Tim Peterse718f612004-10-12 21:51:32 +0000184
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000185 communicate() returns a tuple (stdout, stderr).
186
187 Note: The data read is buffered in memory, so do not use this
188 method if the data size is large or unlimited.
189
190The following attributes are also available:
191
192stdin
193 If the stdin argument is PIPE, this attribute is a file object
194 that provides input to the child process. Otherwise, it is None.
195
196stdout
197 If the stdout argument is PIPE, this attribute is a file object
198 that provides output from the child process. Otherwise, it is
199 None.
200
201stderr
202 If the stderr argument is PIPE, this attribute is file object that
203 provides error output from the child process. Otherwise, it is
204 None.
205
206pid
207 The process ID of the child process.
208
209returncode
210 The child return code. A None value indicates that the process
211 hasn't terminated yet. A negative value -N indicates that the
212 child was terminated by signal N (UNIX only).
213
214
215Replacing older functions with the subprocess module
216====================================================
217In this section, "a ==> b" means that b can be used as a replacement
218for a.
219
220Note: All functions in this section fail (more or less) silently if
221the executed program cannot be found; this module raises an OSError
222exception.
223
224In the following examples, we assume that the subprocess module is
225imported with "from subprocess import *".
226
227
228Replacing /bin/sh shell backquote
229---------------------------------
230output=`mycmd myarg`
231==>
232output = Popen(["mycmd", "myarg"], stdout=PIPE).communicate()[0]
233
234
235Replacing shell pipe line
236-------------------------
237output=`dmesg | grep hda`
238==>
239p1 = Popen(["dmesg"], stdout=PIPE)
Peter Astrand6fdf3cb2004-11-30 18:06:42 +0000240p2 = Popen(["grep", "hda"], stdin=p1.stdout, stdout=PIPE)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000241output = p2.communicate()[0]
242
243
244Replacing os.system()
245---------------------
246sts = os.system("mycmd" + " myarg")
247==>
248p = Popen("mycmd" + " myarg", shell=True)
Neal Norwitz84404832006-07-10 00:05:34 +0000249pid, sts = os.waitpid(p.pid, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000250
251Note:
252
253* Calling the program through the shell is usually not required.
254
255* It's easier to look at the returncode attribute than the
256 exitstatus.
257
258A more real-world example would look like this:
259
260try:
261 retcode = call("mycmd" + " myarg", shell=True)
262 if retcode < 0:
263 print >>sys.stderr, "Child was terminated by signal", -retcode
264 else:
265 print >>sys.stderr, "Child returned", retcode
266except OSError, e:
267 print >>sys.stderr, "Execution failed:", e
268
269
270Replacing os.spawn*
271-------------------
Tim Peterse718f612004-10-12 21:51:32 +0000272P_NOWAIT example:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000273
274pid = os.spawnlp(os.P_NOWAIT, "/bin/mycmd", "mycmd", "myarg")
275==>
276pid = Popen(["/bin/mycmd", "myarg"]).pid
277
278
279P_WAIT example:
280
281retcode = os.spawnlp(os.P_WAIT, "/bin/mycmd", "mycmd", "myarg")
282==>
283retcode = call(["/bin/mycmd", "myarg"])
284
285
Tim Peterse718f612004-10-12 21:51:32 +0000286Vector example:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000287
288os.spawnvp(os.P_NOWAIT, path, args)
289==>
290Popen([path] + args[1:])
291
292
Tim Peterse718f612004-10-12 21:51:32 +0000293Environment example:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000294
295os.spawnlpe(os.P_NOWAIT, "/bin/mycmd", "mycmd", "myarg", env)
296==>
297Popen(["/bin/mycmd", "myarg"], env={"PATH": "/usr/bin"})
298
299
Tim Peterse718f612004-10-12 21:51:32 +0000300Replacing os.popen*
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000301-------------------
Philip Jenvey8b902042009-09-29 19:10:15 +0000302pipe = os.popen("cmd", mode='r', bufsize)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000303==>
Philip Jenvey8b902042009-09-29 19:10:15 +0000304pipe = Popen("cmd", shell=True, bufsize=bufsize, stdout=PIPE).stdout
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000305
Philip Jenvey8b902042009-09-29 19:10:15 +0000306pipe = os.popen("cmd", mode='w', bufsize)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000307==>
Philip Jenvey8b902042009-09-29 19:10:15 +0000308pipe = Popen("cmd", shell=True, bufsize=bufsize, stdin=PIPE).stdin
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000309
310
Philip Jenvey8b902042009-09-29 19:10:15 +0000311(child_stdin, child_stdout) = os.popen2("cmd", mode, bufsize)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000312==>
Philip Jenvey8b902042009-09-29 19:10:15 +0000313p = Popen("cmd", shell=True, bufsize=bufsize,
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000314 stdin=PIPE, stdout=PIPE, close_fds=True)
315(child_stdin, child_stdout) = (p.stdin, p.stdout)
316
317
318(child_stdin,
319 child_stdout,
Philip Jenvey8b902042009-09-29 19:10:15 +0000320 child_stderr) = os.popen3("cmd", mode, bufsize)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000321==>
Philip Jenvey8b902042009-09-29 19:10:15 +0000322p = Popen("cmd", shell=True, bufsize=bufsize,
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000323 stdin=PIPE, stdout=PIPE, stderr=PIPE, close_fds=True)
324(child_stdin,
325 child_stdout,
326 child_stderr) = (p.stdin, p.stdout, p.stderr)
327
328
Philip Jenvey8b902042009-09-29 19:10:15 +0000329(child_stdin, child_stdout_and_stderr) = os.popen4("cmd", mode,
330 bufsize)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000331==>
Philip Jenvey8b902042009-09-29 19:10:15 +0000332p = Popen("cmd", shell=True, bufsize=bufsize,
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000333 stdin=PIPE, stdout=PIPE, stderr=STDOUT, close_fds=True)
334(child_stdin, child_stdout_and_stderr) = (p.stdin, p.stdout)
335
Philip Jenvey8b902042009-09-29 19:10:15 +0000336On Unix, os.popen2, os.popen3 and os.popen4 also accept a sequence as
337the command to execute, in which case arguments will be passed
338directly to the program without shell intervention. This usage can be
339replaced as follows:
340
341(child_stdin, child_stdout) = os.popen2(["/bin/ls", "-l"], mode,
342 bufsize)
343==>
344p = Popen(["/bin/ls", "-l"], bufsize=bufsize, stdin=PIPE, stdout=PIPE)
345(child_stdin, child_stdout) = (p.stdin, p.stdout)
346
347Return code handling translates as follows:
348
349pipe = os.popen("cmd", 'w')
350...
351rc = pipe.close()
352if rc != None and rc % 256:
353 print "There were some errors"
354==>
355process = Popen("cmd", 'w', shell=True, stdin=PIPE)
356...
357process.stdin.close()
358if process.wait() != 0:
359 print "There were some errors"
360
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000361
362Replacing popen2.*
363------------------
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000364(child_stdout, child_stdin) = popen2.popen2("somestring", bufsize, mode)
365==>
366p = Popen(["somestring"], shell=True, bufsize=bufsize
367 stdin=PIPE, stdout=PIPE, close_fds=True)
368(child_stdout, child_stdin) = (p.stdout, p.stdin)
369
Philip Jenvey8b902042009-09-29 19:10:15 +0000370On Unix, popen2 also accepts a sequence as the command to execute, in
371which case arguments will be passed directly to the program without
372shell intervention. This usage can be replaced as follows:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000373
Philip Jenvey8b902042009-09-29 19:10:15 +0000374(child_stdout, child_stdin) = popen2.popen2(["mycmd", "myarg"], bufsize,
375 mode)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000376==>
377p = Popen(["mycmd", "myarg"], bufsize=bufsize,
378 stdin=PIPE, stdout=PIPE, close_fds=True)
379(child_stdout, child_stdin) = (p.stdout, p.stdin)
380
Neal Norwitzaa87fb62007-05-11 06:23:01 +0000381The popen2.Popen3 and popen2.Popen4 basically works as subprocess.Popen,
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000382except that:
383
384* subprocess.Popen raises an exception if the execution fails
385* the capturestderr argument is replaced with the stderr argument.
386* stdin=PIPE and stdout=PIPE must be specified.
387* popen2 closes all filedescriptors by default, but you have to specify
Tim Peterse718f612004-10-12 21:51:32 +0000388 close_fds=True with subprocess.Popen.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000389"""
390
391import sys
392mswindows = (sys.platform == "win32")
393
394import os
Peter Astrandc26516b2005-02-21 08:13:02 +0000395import types
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000396import traceback
Gregory P. Smith87d49792008-01-19 20:57:59 +0000397import gc
Christian Heimese74c8f22008-04-19 02:23:57 +0000398import signal
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000399
Peter Astrand454f7672005-01-01 09:36:35 +0000400# Exception classes used by this module.
Peter Astrand7d1d4362006-07-14 14:04:45 +0000401class CalledProcessError(Exception):
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000402 """This exception is raised when a process run by check_call() or
Gregory P. Smith26576802008-12-05 02:27:01 +0000403 check_output() returns a non-zero exit status.
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000404 The exit status will be stored in the returncode attribute;
Gregory P. Smith26576802008-12-05 02:27:01 +0000405 check_output() will also store the output in the output attribute.
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000406 """
407 def __init__(self, returncode, cmd, output=None):
Peter Astrand7d1d4362006-07-14 14:04:45 +0000408 self.returncode = returncode
409 self.cmd = cmd
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000410 self.output = output
Peter Astrand7d1d4362006-07-14 14:04:45 +0000411 def __str__(self):
412 return "Command '%s' returned non-zero exit status %d" % (self.cmd, self.returncode)
Tim Peters73a9ead2006-07-18 21:55:15 +0000413
Peter Astrand454f7672005-01-01 09:36:35 +0000414
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000415if mswindows:
416 import threading
417 import msvcrt
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000418 if 0: # <-- change this to use pywin32 instead of the _subprocess driver
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000419 import pywintypes
Tim Peterse8374a52004-10-13 03:15:00 +0000420 from win32api import GetStdHandle, STD_INPUT_HANDLE, \
421 STD_OUTPUT_HANDLE, STD_ERROR_HANDLE
422 from win32api import GetCurrentProcess, DuplicateHandle, \
423 GetModuleFileName, GetVersion
Peter Astrandc1d65362004-11-07 14:30:34 +0000424 from win32con import DUPLICATE_SAME_ACCESS, SW_HIDE
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000425 from win32pipe import CreatePipe
Tim Peterse8374a52004-10-13 03:15:00 +0000426 from win32process import CreateProcess, STARTUPINFO, \
427 GetExitCodeProcess, STARTF_USESTDHANDLES, \
Peter Astrandc1d65362004-11-07 14:30:34 +0000428 STARTF_USESHOWWINDOW, CREATE_NEW_CONSOLE
Christian Heimese74c8f22008-04-19 02:23:57 +0000429 from win32process import TerminateProcess
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000430 from win32event import WaitForSingleObject, INFINITE, WAIT_OBJECT_0
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000431 else:
432 from _subprocess import *
433 class STARTUPINFO:
434 dwFlags = 0
435 hStdInput = None
436 hStdOutput = None
437 hStdError = None
Georg Brandlad624892006-06-04 22:15:37 +0000438 wShowWindow = 0
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000439 class pywintypes:
440 error = IOError
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000441else:
442 import select
Gregory P. Smithdd7ca242009-07-04 01:49:29 +0000443 _has_poll = hasattr(select, 'poll')
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000444 import errno
445 import fcntl
446 import pickle
447
Amaury Forgeot d'Arcce32eb72009-07-09 22:37:22 +0000448 # When select or poll has indicated that the file is writable,
449 # we can write up to _PIPE_BUF bytes without risk of blocking.
450 # POSIX defines PIPE_BUF as >= 512.
451 _PIPE_BUF = getattr(select, 'PIPE_BUF', 512)
452
453
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000454__all__ = ["Popen", "PIPE", "STDOUT", "call", "check_call",
Gregory P. Smith26576802008-12-05 02:27:01 +0000455 "check_output", "CalledProcessError"]
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000456
457try:
458 MAXFD = os.sysconf("SC_OPEN_MAX")
459except:
460 MAXFD = 256
461
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000462_active = []
463
464def _cleanup():
465 for inst in _active[:]:
Georg Brandl02e7dfd2009-12-28 08:09:32 +0000466 res = inst._internal_poll(_deadstate=sys.maxint)
467 if res is not None and res >= 0:
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000468 try:
469 _active.remove(inst)
470 except ValueError:
471 # This can happen if two threads create a new Popen instance.
472 # It's harmless that it was already removed, so ignore.
473 pass
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000474
475PIPE = -1
476STDOUT = -2
477
478
Peter Astrand5f5e1412004-12-05 20:15:36 +0000479def call(*popenargs, **kwargs):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000480 """Run command with arguments. Wait for command to complete, then
481 return the returncode attribute.
482
483 The arguments are the same as for the Popen constructor. Example:
484
485 retcode = call(["ls", "-l"])
486 """
Peter Astrand5f5e1412004-12-05 20:15:36 +0000487 return Popen(*popenargs, **kwargs).wait()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000488
489
Peter Astrand454f7672005-01-01 09:36:35 +0000490def check_call(*popenargs, **kwargs):
491 """Run command with arguments. Wait for command to complete. If
492 the exit code was zero then return, otherwise raise
493 CalledProcessError. The CalledProcessError object will have the
Peter Astrand7d1d4362006-07-14 14:04:45 +0000494 return code in the returncode attribute.
Peter Astrand454f7672005-01-01 09:36:35 +0000495
496 The arguments are the same as for the Popen constructor. Example:
497
498 check_call(["ls", "-l"])
499 """
500 retcode = call(*popenargs, **kwargs)
Peter Astrand454f7672005-01-01 09:36:35 +0000501 if retcode:
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000502 cmd = kwargs.get("args")
503 if cmd is None:
504 cmd = popenargs[0]
Peter Astrand7d1d4362006-07-14 14:04:45 +0000505 raise CalledProcessError(retcode, cmd)
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000506 return 0
507
508
Gregory P. Smith26576802008-12-05 02:27:01 +0000509def check_output(*popenargs, **kwargs):
Georg Brandl6ab5d082009-12-20 14:33:20 +0000510 r"""Run command with arguments and return its output as a byte string.
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000511
512 If the exit code was non-zero it raises a CalledProcessError. The
513 CalledProcessError object will have the return code in the returncode
514 attribute and output in the output attribute.
515
516 The arguments are the same as for the Popen constructor. Example:
517
Gregory P. Smith26576802008-12-05 02:27:01 +0000518 >>> check_output(["ls", "-l", "/dev/null"])
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000519 'crw-rw-rw- 1 root root 1, 3 Oct 18 2007 /dev/null\n'
520
521 The stdout argument is not allowed as it is used internally.
Georg Brandl02e7dfd2009-12-28 08:09:32 +0000522 To capture standard error in the result, use stderr=STDOUT.
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000523
Gregory P. Smith26576802008-12-05 02:27:01 +0000524 >>> check_output(["/bin/sh", "-c",
Georg Brandl6ab5d082009-12-20 14:33:20 +0000525 ... "ls -l non_existent_file ; exit 0"],
Georg Brandl02e7dfd2009-12-28 08:09:32 +0000526 ... stderr=STDOUT)
Mark Dickinson3e4caeb2009-02-21 20:27:01 +0000527 'ls: non_existent_file: No such file or directory\n'
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000528 """
529 if 'stdout' in kwargs:
530 raise ValueError('stdout argument not allowed, it will be overridden.')
Amaury Forgeot d'Arc5fe420e2009-06-18 22:32:50 +0000531 process = Popen(stdout=PIPE, *popenargs, **kwargs)
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000532 output, unused_err = process.communicate()
533 retcode = process.poll()
534 if retcode:
535 cmd = kwargs.get("args")
536 if cmd is None:
537 cmd = popenargs[0]
538 raise CalledProcessError(retcode, cmd, output=output)
539 return output
Peter Astrand454f7672005-01-01 09:36:35 +0000540
541
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000542def list2cmdline(seq):
543 """
544 Translate a sequence of arguments into a command line
545 string, using the same rules as the MS C runtime:
546
547 1) Arguments are delimited by white space, which is either a
548 space or a tab.
549
550 2) A string surrounded by double quotation marks is
551 interpreted as a single argument, regardless of white space
Gregory P. Smith70eb2f92008-01-19 22:49:37 +0000552 or pipe characters contained within. A quoted string can be
553 embedded in an argument.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000554
555 3) A double quotation mark preceded by a backslash is
556 interpreted as a literal double quotation mark.
557
558 4) Backslashes are interpreted literally, unless they
559 immediately precede a double quotation mark.
560
561 5) If backslashes immediately precede a double quotation mark,
562 every pair of backslashes is interpreted as a literal
563 backslash. If the number of backslashes is odd, the last
564 backslash escapes the next double quotation mark as
565 described in rule 3.
566 """
567
568 # See
Eric Smithd19915e2009-11-09 15:16:23 +0000569 # http://msdn.microsoft.com/en-us/library/17w5ykft.aspx
570 # or search http://msdn.microsoft.com for
571 # "Parsing C++ Command-Line Arguments"
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000572 result = []
573 needquote = False
574 for arg in seq:
575 bs_buf = []
576
577 # Add a space to separate this argument from the others
578 if result:
579 result.append(' ')
580
Gregory P. Smith70eb2f92008-01-19 22:49:37 +0000581 needquote = (" " in arg) or ("\t" in arg) or ("|" in arg) or not arg
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000582 if needquote:
583 result.append('"')
584
585 for c in arg:
586 if c == '\\':
Tim Peterse718f612004-10-12 21:51:32 +0000587 # Don't know if we need to double yet.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000588 bs_buf.append(c)
589 elif c == '"':
Gregory P. Smithe047e6d2008-01-19 20:49:02 +0000590 # Double backslashes.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000591 result.append('\\' * len(bs_buf)*2)
592 bs_buf = []
593 result.append('\\"')
594 else:
595 # Normal char
596 if bs_buf:
597 result.extend(bs_buf)
598 bs_buf = []
599 result.append(c)
600
Gregory P. Smithe047e6d2008-01-19 20:49:02 +0000601 # Add remaining backslashes, if any.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000602 if bs_buf:
603 result.extend(bs_buf)
604
605 if needquote:
Peter Astrand7e78ade2005-03-03 21:10:23 +0000606 result.extend(bs_buf)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000607 result.append('"')
608
609 return ''.join(result)
610
611
612class Popen(object):
613 def __init__(self, args, bufsize=0, executable=None,
614 stdin=None, stdout=None, stderr=None,
615 preexec_fn=None, close_fds=False, shell=False,
616 cwd=None, env=None, universal_newlines=False,
617 startupinfo=None, creationflags=0):
618 """Create new Popen instance."""
619 _cleanup()
620
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000621 self._child_created = False
Peter Astrand738131d2004-11-30 21:04:45 +0000622 if not isinstance(bufsize, (int, long)):
623 raise TypeError("bufsize must be an integer")
624
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000625 if mswindows:
Tim Peterse8374a52004-10-13 03:15:00 +0000626 if preexec_fn is not None:
627 raise ValueError("preexec_fn is not supported on Windows "
628 "platforms")
Peter Astrand81a191b2007-05-26 22:18:20 +0000629 if close_fds and (stdin is not None or stdout is not None or
630 stderr is not None):
Tim Peterse8374a52004-10-13 03:15:00 +0000631 raise ValueError("close_fds is not supported on Windows "
Peter Astrand81a191b2007-05-26 22:18:20 +0000632 "platforms if you redirect stdin/stdout/stderr")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000633 else:
634 # POSIX
Tim Peterse8374a52004-10-13 03:15:00 +0000635 if startupinfo is not None:
636 raise ValueError("startupinfo is only supported on Windows "
637 "platforms")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000638 if creationflags != 0:
Tim Peterse8374a52004-10-13 03:15:00 +0000639 raise ValueError("creationflags is only supported on Windows "
640 "platforms")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000641
Tim Peterse718f612004-10-12 21:51:32 +0000642 self.stdin = None
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000643 self.stdout = None
644 self.stderr = None
645 self.pid = None
646 self.returncode = None
647 self.universal_newlines = universal_newlines
648
649 # Input and output objects. The general principle is like
650 # this:
651 #
652 # Parent Child
653 # ------ -----
654 # p2cwrite ---stdin---> p2cread
655 # c2pread <--stdout--- c2pwrite
656 # errread <--stderr--- errwrite
657 #
658 # On POSIX, the child objects are file descriptors. On
659 # Windows, these are Windows file handles. The parent objects
660 # are file descriptors on both platforms. The parent objects
661 # are None when not using PIPEs. The child objects are None
662 # when not redirecting.
Tim Peterse718f612004-10-12 21:51:32 +0000663
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000664 (p2cread, p2cwrite,
665 c2pread, c2pwrite,
666 errread, errwrite) = self._get_handles(stdin, stdout, stderr)
667
668 self._execute_child(args, executable, preexec_fn, close_fds,
669 cwd, env, universal_newlines,
670 startupinfo, creationflags, shell,
671 p2cread, p2cwrite,
672 c2pread, c2pwrite,
673 errread, errwrite)
674
Peter Astrand5f9c6ae2007-02-06 15:37:50 +0000675 if mswindows:
Hirokazu Yamamotoeacbbdf2009-03-03 22:18:14 +0000676 if p2cwrite is not None:
677 p2cwrite = msvcrt.open_osfhandle(p2cwrite.Detach(), 0)
678 if c2pread is not None:
679 c2pread = msvcrt.open_osfhandle(c2pread.Detach(), 0)
680 if errread is not None:
681 errread = msvcrt.open_osfhandle(errread.Detach(), 0)
Peter Astrand5f9c6ae2007-02-06 15:37:50 +0000682
Peter Astrandf5400032007-02-02 19:06:36 +0000683 if p2cwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000684 self.stdin = os.fdopen(p2cwrite, 'wb', bufsize)
Peter Astrandf5400032007-02-02 19:06:36 +0000685 if c2pread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000686 if universal_newlines:
687 self.stdout = os.fdopen(c2pread, 'rU', bufsize)
688 else:
689 self.stdout = os.fdopen(c2pread, 'rb', bufsize)
Peter Astrandf5400032007-02-02 19:06:36 +0000690 if errread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000691 if universal_newlines:
692 self.stderr = os.fdopen(errread, 'rU', bufsize)
693 else:
694 self.stderr = os.fdopen(errread, 'rb', bufsize)
Tim Peterse718f612004-10-12 21:51:32 +0000695
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000696
697 def _translate_newlines(self, data):
698 data = data.replace("\r\n", "\n")
699 data = data.replace("\r", "\n")
700 return data
701
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000702
Georg Brandl24522982007-04-21 20:35:38 +0000703 def __del__(self, sys=sys):
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000704 if not self._child_created:
705 # We didn't get to successfully create a child process.
706 return
707 # In case the child hasn't been waited on, check if it's done.
Gregory P. Smitha36f8fe2008-08-04 00:13:29 +0000708 self._internal_poll(_deadstate=sys.maxint)
Georg Brandl13cf38c2006-07-20 16:28:39 +0000709 if self.returncode is None and _active is not None:
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000710 # Child is still running, keep us alive until we can wait on it.
711 _active.append(self)
712
713
Peter Astrand23109f02005-03-03 20:28:59 +0000714 def communicate(self, input=None):
715 """Interact with process: Send data to stdin. Read data from
716 stdout and stderr, until end-of-file is reached. Wait for
717 process to terminate. The optional input argument should be a
718 string to be sent to the child process, or None, if no data
719 should be sent to the child.
Tim Peterseba28be2005-03-28 01:08:02 +0000720
Peter Astrand23109f02005-03-03 20:28:59 +0000721 communicate() returns a tuple (stdout, stderr)."""
722
723 # Optimization: If we are only using one pipe, or no pipe at
724 # all, using select() or threads is unnecessary.
725 if [self.stdin, self.stdout, self.stderr].count(None) >= 2:
Tim Peterseba28be2005-03-28 01:08:02 +0000726 stdout = None
727 stderr = None
Peter Astrand23109f02005-03-03 20:28:59 +0000728 if self.stdin:
729 if input:
730 self.stdin.write(input)
731 self.stdin.close()
732 elif self.stdout:
733 stdout = self.stdout.read()
Gregory P. Smith4036fd42008-05-26 20:22:14 +0000734 self.stdout.close()
Peter Astrand23109f02005-03-03 20:28:59 +0000735 elif self.stderr:
736 stderr = self.stderr.read()
Gregory P. Smith4036fd42008-05-26 20:22:14 +0000737 self.stderr.close()
Peter Astrand23109f02005-03-03 20:28:59 +0000738 self.wait()
739 return (stdout, stderr)
Tim Peterseba28be2005-03-28 01:08:02 +0000740
Peter Astrand23109f02005-03-03 20:28:59 +0000741 return self._communicate(input)
742
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000743
Gregory P. Smitha36f8fe2008-08-04 00:13:29 +0000744 def poll(self):
745 return self._internal_poll()
746
747
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000748 if mswindows:
749 #
750 # Windows methods
751 #
752 def _get_handles(self, stdin, stdout, stderr):
Amaury Forgeot d'Arc8318afa2009-07-10 16:47:42 +0000753 """Construct and return tuple with IO objects:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000754 p2cread, p2cwrite, c2pread, c2pwrite, errread, errwrite
755 """
Peter Astrandd38ddf42005-02-10 08:32:50 +0000756 if stdin is None and stdout is None and stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000757 return (None, None, None, None, None, None)
Tim Peterse718f612004-10-12 21:51:32 +0000758
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000759 p2cread, p2cwrite = None, None
760 c2pread, c2pwrite = None, None
761 errread, errwrite = None, None
762
Peter Astrandd38ddf42005-02-10 08:32:50 +0000763 if stdin is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000764 p2cread = GetStdHandle(STD_INPUT_HANDLE)
Hirokazu Yamamotoeacbbdf2009-03-03 22:18:14 +0000765 if p2cread is None:
766 p2cread, _ = CreatePipe(None, 0)
767 elif stdin == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000768 p2cread, p2cwrite = CreatePipe(None, 0)
Peter Astrandd38ddf42005-02-10 08:32:50 +0000769 elif isinstance(stdin, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000770 p2cread = msvcrt.get_osfhandle(stdin)
771 else:
772 # Assuming file-like object
773 p2cread = msvcrt.get_osfhandle(stdin.fileno())
774 p2cread = self._make_inheritable(p2cread)
775
Peter Astrandd38ddf42005-02-10 08:32:50 +0000776 if stdout is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000777 c2pwrite = GetStdHandle(STD_OUTPUT_HANDLE)
Hirokazu Yamamotoeacbbdf2009-03-03 22:18:14 +0000778 if c2pwrite is None:
779 _, c2pwrite = CreatePipe(None, 0)
780 elif stdout == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000781 c2pread, c2pwrite = CreatePipe(None, 0)
Peter Astrandd38ddf42005-02-10 08:32:50 +0000782 elif isinstance(stdout, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000783 c2pwrite = msvcrt.get_osfhandle(stdout)
784 else:
785 # Assuming file-like object
786 c2pwrite = msvcrt.get_osfhandle(stdout.fileno())
787 c2pwrite = self._make_inheritable(c2pwrite)
788
Peter Astrandd38ddf42005-02-10 08:32:50 +0000789 if stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000790 errwrite = GetStdHandle(STD_ERROR_HANDLE)
Hirokazu Yamamotoeacbbdf2009-03-03 22:18:14 +0000791 if errwrite is None:
792 _, errwrite = CreatePipe(None, 0)
793 elif stderr == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000794 errread, errwrite = CreatePipe(None, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000795 elif stderr == STDOUT:
796 errwrite = c2pwrite
Peter Astrandd38ddf42005-02-10 08:32:50 +0000797 elif isinstance(stderr, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000798 errwrite = msvcrt.get_osfhandle(stderr)
799 else:
800 # Assuming file-like object
801 errwrite = msvcrt.get_osfhandle(stderr.fileno())
802 errwrite = self._make_inheritable(errwrite)
803
804 return (p2cread, p2cwrite,
805 c2pread, c2pwrite,
806 errread, errwrite)
807
808
809 def _make_inheritable(self, handle):
810 """Return a duplicate of handle, which is inheritable"""
811 return DuplicateHandle(GetCurrentProcess(), handle,
812 GetCurrentProcess(), 0, 1,
813 DUPLICATE_SAME_ACCESS)
814
815
816 def _find_w9xpopen(self):
817 """Find and return absolut path to w9xpopen.exe"""
Tim Peterse8374a52004-10-13 03:15:00 +0000818 w9xpopen = os.path.join(os.path.dirname(GetModuleFileName(0)),
819 "w9xpopen.exe")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000820 if not os.path.exists(w9xpopen):
821 # Eeek - file-not-found - possibly an embedding
822 # situation - see if we can locate it in sys.exec_prefix
Tim Peterse8374a52004-10-13 03:15:00 +0000823 w9xpopen = os.path.join(os.path.dirname(sys.exec_prefix),
824 "w9xpopen.exe")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000825 if not os.path.exists(w9xpopen):
Tim Peterse8374a52004-10-13 03:15:00 +0000826 raise RuntimeError("Cannot locate w9xpopen.exe, which is "
827 "needed for Popen to work with your "
828 "shell or platform.")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000829 return w9xpopen
830
Tim Peterse718f612004-10-12 21:51:32 +0000831
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000832 def _execute_child(self, args, executable, preexec_fn, close_fds,
833 cwd, env, universal_newlines,
834 startupinfo, creationflags, shell,
835 p2cread, p2cwrite,
836 c2pread, c2pwrite,
837 errread, errwrite):
838 """Execute program (MS Windows version)"""
839
Peter Astrandc26516b2005-02-21 08:13:02 +0000840 if not isinstance(args, types.StringTypes):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000841 args = list2cmdline(args)
842
Peter Astrandc1d65362004-11-07 14:30:34 +0000843 # Process startup details
Peter Astrandd38ddf42005-02-10 08:32:50 +0000844 if startupinfo is None:
Georg Brandlad624892006-06-04 22:15:37 +0000845 startupinfo = STARTUPINFO()
846 if None not in (p2cread, c2pwrite, errwrite):
Peter Astrandc1d65362004-11-07 14:30:34 +0000847 startupinfo.dwFlags |= STARTF_USESTDHANDLES
848 startupinfo.hStdInput = p2cread
849 startupinfo.hStdOutput = c2pwrite
850 startupinfo.hStdError = errwrite
851
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000852 if shell:
Georg Brandlad624892006-06-04 22:15:37 +0000853 startupinfo.dwFlags |= STARTF_USESHOWWINDOW
854 startupinfo.wShowWindow = SW_HIDE
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000855 comspec = os.environ.get("COMSPEC", "cmd.exe")
856 args = comspec + " /c " + args
Tim Peterse8374a52004-10-13 03:15:00 +0000857 if (GetVersion() >= 0x80000000L or
858 os.path.basename(comspec).lower() == "command.com"):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000859 # Win9x, or using command.com on NT. We need to
860 # use the w9xpopen intermediate program. For more
861 # information, see KB Q150956
862 # (http://web.archive.org/web/20011105084002/http://support.microsoft.com/support/kb/articles/Q150/9/56.asp)
863 w9xpopen = self._find_w9xpopen()
864 args = '"%s" %s' % (w9xpopen, args)
865 # Not passing CREATE_NEW_CONSOLE has been known to
866 # cause random failures on win9x. Specifically a
867 # dialog: "Your program accessed mem currently in
868 # use at xxx" and a hopeful warning about the
869 # stability of your system. Cost is Ctrl+C wont
870 # kill children.
871 creationflags |= CREATE_NEW_CONSOLE
872
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000873 # Start the process
874 try:
875 hp, ht, pid, tid = CreateProcess(executable, args,
Tim Peterse8374a52004-10-13 03:15:00 +0000876 # no special security
877 None, None,
Peter Astrand81a191b2007-05-26 22:18:20 +0000878 int(not close_fds),
Tim Peterse8374a52004-10-13 03:15:00 +0000879 creationflags,
880 env,
881 cwd,
882 startupinfo)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000883 except pywintypes.error, e:
884 # Translate pywintypes.error to WindowsError, which is
885 # a subclass of OSError. FIXME: We should really
886 # translate errno using _sys_errlist (or simliar), but
887 # how can this be done from Python?
888 raise WindowsError(*e.args)
889
890 # Retain the process handle, but close the thread handle
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000891 self._child_created = True
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000892 self._handle = hp
893 self.pid = pid
894 ht.Close()
895
Andrew M. Kuchling51ee66e2004-10-12 16:38:42 +0000896 # Child is launched. Close the parent's copy of those pipe
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000897 # handles that only the child should have open. You need
898 # to make sure that no handles to the write end of the
899 # output pipe are maintained in this process or else the
900 # pipe will not close when the child process exits and the
901 # ReadFile will hang.
Peter Astrandd38ddf42005-02-10 08:32:50 +0000902 if p2cread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000903 p2cread.Close()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000904 if c2pwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000905 c2pwrite.Close()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000906 if errwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000907 errwrite.Close()
908
Tim Peterse718f612004-10-12 21:51:32 +0000909
Gregory P. Smitha36f8fe2008-08-04 00:13:29 +0000910 def _internal_poll(self, _deadstate=None):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000911 """Check if child process has terminated. Returns returncode
912 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +0000913 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000914 if WaitForSingleObject(self._handle, 0) == WAIT_OBJECT_0:
915 self.returncode = GetExitCodeProcess(self._handle)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000916 return self.returncode
917
918
919 def wait(self):
920 """Wait for child process to terminate. Returns returncode
921 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +0000922 if self.returncode is None:
Georg Brandl84fedf72010-02-06 22:59:15 +0000923 WaitForSingleObject(self._handle, INFINITE)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000924 self.returncode = GetExitCodeProcess(self._handle)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000925 return self.returncode
926
927
928 def _readerthread(self, fh, buffer):
929 buffer.append(fh.read())
930
931
Peter Astrand23109f02005-03-03 20:28:59 +0000932 def _communicate(self, input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000933 stdout = None # Return
934 stderr = None # Return
935
936 if self.stdout:
937 stdout = []
Tim Peterse8374a52004-10-13 03:15:00 +0000938 stdout_thread = threading.Thread(target=self._readerthread,
939 args=(self.stdout, stdout))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000940 stdout_thread.setDaemon(True)
941 stdout_thread.start()
942 if self.stderr:
943 stderr = []
Tim Peterse8374a52004-10-13 03:15:00 +0000944 stderr_thread = threading.Thread(target=self._readerthread,
945 args=(self.stderr, stderr))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000946 stderr_thread.setDaemon(True)
947 stderr_thread.start()
948
949 if self.stdin:
Peter Astrandd38ddf42005-02-10 08:32:50 +0000950 if input is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000951 self.stdin.write(input)
952 self.stdin.close()
953
954 if self.stdout:
955 stdout_thread.join()
956 if self.stderr:
957 stderr_thread.join()
958
959 # All data exchanged. Translate lists into strings.
Peter Astrandd38ddf42005-02-10 08:32:50 +0000960 if stdout is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000961 stdout = stdout[0]
Peter Astrandd38ddf42005-02-10 08:32:50 +0000962 if stderr is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000963 stderr = stderr[0]
964
965 # Translate newlines, if requested. We cannot let the file
966 # object do the translation: It is based on stdio, which is
967 # impossible to combine with select (unless forcing no
968 # buffering).
Neal Norwitza6d01ce2006-05-02 06:23:22 +0000969 if self.universal_newlines and hasattr(file, 'newlines'):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000970 if stdout:
971 stdout = self._translate_newlines(stdout)
972 if stderr:
973 stderr = self._translate_newlines(stderr)
974
975 self.wait()
976 return (stdout, stderr)
977
Christian Heimese74c8f22008-04-19 02:23:57 +0000978 def send_signal(self, sig):
979 """Send a signal to the process
980 """
981 if sig == signal.SIGTERM:
982 self.terminate()
983 else:
984 raise ValueError("Only SIGTERM is supported on Windows")
985
986 def terminate(self):
987 """Terminates the process
988 """
989 TerminateProcess(self._handle, 1)
990
991 kill = terminate
992
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000993 else:
994 #
995 # POSIX methods
996 #
997 def _get_handles(self, stdin, stdout, stderr):
Amaury Forgeot d'Arc8318afa2009-07-10 16:47:42 +0000998 """Construct and return tuple with IO objects:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000999 p2cread, p2cwrite, c2pread, c2pwrite, errread, errwrite
1000 """
1001 p2cread, p2cwrite = None, None
1002 c2pread, c2pwrite = None, None
1003 errread, errwrite = None, None
1004
Peter Astrandd38ddf42005-02-10 08:32:50 +00001005 if stdin is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001006 pass
1007 elif stdin == PIPE:
1008 p2cread, p2cwrite = os.pipe()
Peter Astrandd38ddf42005-02-10 08:32:50 +00001009 elif isinstance(stdin, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001010 p2cread = stdin
1011 else:
1012 # Assuming file-like object
1013 p2cread = stdin.fileno()
1014
Peter Astrandd38ddf42005-02-10 08:32:50 +00001015 if stdout is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001016 pass
1017 elif stdout == PIPE:
1018 c2pread, c2pwrite = os.pipe()
Peter Astrandd38ddf42005-02-10 08:32:50 +00001019 elif isinstance(stdout, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001020 c2pwrite = stdout
1021 else:
1022 # Assuming file-like object
Tim Peterse718f612004-10-12 21:51:32 +00001023 c2pwrite = stdout.fileno()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001024
Peter Astrandd38ddf42005-02-10 08:32:50 +00001025 if stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001026 pass
1027 elif stderr == PIPE:
1028 errread, errwrite = os.pipe()
1029 elif stderr == STDOUT:
1030 errwrite = c2pwrite
Peter Astrandd38ddf42005-02-10 08:32:50 +00001031 elif isinstance(stderr, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001032 errwrite = stderr
1033 else:
1034 # Assuming file-like object
1035 errwrite = stderr.fileno()
1036
1037 return (p2cread, p2cwrite,
1038 c2pread, c2pwrite,
1039 errread, errwrite)
1040
1041
1042 def _set_cloexec_flag(self, fd):
1043 try:
1044 cloexec_flag = fcntl.FD_CLOEXEC
1045 except AttributeError:
1046 cloexec_flag = 1
1047
1048 old = fcntl.fcntl(fd, fcntl.F_GETFD)
1049 fcntl.fcntl(fd, fcntl.F_SETFD, old | cloexec_flag)
1050
1051
1052 def _close_fds(self, but):
Amaury Forgeot d'Arc5fe420e2009-06-18 22:32:50 +00001053 if hasattr(os, 'closerange'):
1054 os.closerange(3, but)
1055 os.closerange(but + 1, MAXFD)
1056 else:
1057 for i in xrange(3, MAXFD):
1058 if i == but:
1059 continue
1060 try:
1061 os.close(i)
1062 except:
1063 pass
Tim Peterse718f612004-10-12 21:51:32 +00001064
1065
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001066 def _execute_child(self, args, executable, preexec_fn, close_fds,
1067 cwd, env, universal_newlines,
1068 startupinfo, creationflags, shell,
1069 p2cread, p2cwrite,
1070 c2pread, c2pwrite,
1071 errread, errwrite):
1072 """Execute program (POSIX version)"""
1073
Peter Astrandc26516b2005-02-21 08:13:02 +00001074 if isinstance(args, types.StringTypes):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001075 args = [args]
Georg Brandl6c0e1e82006-10-29 09:05:04 +00001076 else:
1077 args = list(args)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001078
1079 if shell:
1080 args = ["/bin/sh", "-c"] + args
1081
Peter Astrandd38ddf42005-02-10 08:32:50 +00001082 if executable is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001083 executable = args[0]
1084
1085 # For transferring possible exec failure from child to parent
1086 # The first char specifies the exception type: 0 means
1087 # OSError, 1 means some other error.
1088 errpipe_read, errpipe_write = os.pipe()
Gregory P. Smith87d49792008-01-19 20:57:59 +00001089 try:
Gregory P. Smith92ffc632008-01-19 22:23:56 +00001090 try:
Facundo Batista8c826b72009-06-19 18:02:28 +00001091 self._set_cloexec_flag(errpipe_write)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001092
Facundo Batista8c826b72009-06-19 18:02:28 +00001093 gc_was_enabled = gc.isenabled()
1094 # Disable gc to avoid bug where gc -> file_dealloc ->
1095 # write to stderr -> hang. http://bugs.python.org/issue1336
1096 gc.disable()
1097 try:
1098 self.pid = os.fork()
Georg Brandl3e8b8692009-07-16 21:47:51 +00001099 except:
Facundo Batista8c826b72009-06-19 18:02:28 +00001100 if gc_was_enabled:
1101 gc.enable()
Georg Brandl3e8b8692009-07-16 21:47:51 +00001102 raise
Facundo Batista8c826b72009-06-19 18:02:28 +00001103 self._child_created = True
1104 if self.pid == 0:
1105 # Child
1106 try:
1107 # Close parent's pipe ends
1108 if p2cwrite is not None:
1109 os.close(p2cwrite)
1110 if c2pread is not None:
1111 os.close(c2pread)
1112 if errread is not None:
1113 os.close(errread)
1114 os.close(errpipe_read)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001115
Facundo Batista8c826b72009-06-19 18:02:28 +00001116 # Dup fds for child
1117 if p2cread is not None:
1118 os.dup2(p2cread, 0)
1119 if c2pwrite is not None:
1120 os.dup2(c2pwrite, 1)
1121 if errwrite is not None:
1122 os.dup2(errwrite, 2)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001123
Facundo Batista8c826b72009-06-19 18:02:28 +00001124 # Close pipe fds. Make sure we don't close the same
1125 # fd more than once, or standard fds.
1126 if p2cread is not None and p2cread not in (0,):
1127 os.close(p2cread)
1128 if c2pwrite is not None and c2pwrite not in (p2cread, 1):
1129 os.close(c2pwrite)
1130 if errwrite is not None and errwrite not in (p2cread, c2pwrite, 2):
1131 os.close(errwrite)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001132
Facundo Batista8c826b72009-06-19 18:02:28 +00001133 # Close all other fds, if asked for
1134 if close_fds:
1135 self._close_fds(but=errpipe_write)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001136
Facundo Batista8c826b72009-06-19 18:02:28 +00001137 if cwd is not None:
1138 os.chdir(cwd)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001139
Facundo Batista8c826b72009-06-19 18:02:28 +00001140 if preexec_fn:
1141 preexec_fn()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001142
Facundo Batista8c826b72009-06-19 18:02:28 +00001143 if env is None:
1144 os.execvp(executable, args)
1145 else:
1146 os.execvpe(executable, args, env)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001147
Facundo Batista8c826b72009-06-19 18:02:28 +00001148 except:
1149 exc_type, exc_value, tb = sys.exc_info()
1150 # Save the traceback and attach it to the exception object
1151 exc_lines = traceback.format_exception(exc_type,
1152 exc_value,
1153 tb)
1154 exc_value.child_traceback = ''.join(exc_lines)
1155 os.write(errpipe_write, pickle.dumps(exc_value))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001156
Facundo Batista8c826b72009-06-19 18:02:28 +00001157 # This exitcode won't be reported to applications, so it
1158 # really doesn't matter what we return.
1159 os._exit(255)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001160
Facundo Batista8c826b72009-06-19 18:02:28 +00001161 # Parent
1162 if gc_was_enabled:
1163 gc.enable()
1164 finally:
1165 # be sure the FD is closed no matter what
1166 os.close(errpipe_write)
1167
1168 if p2cread is not None and p2cwrite is not None:
1169 os.close(p2cread)
1170 if c2pwrite is not None and c2pread is not None:
1171 os.close(c2pwrite)
1172 if errwrite is not None and errread is not None:
1173 os.close(errwrite)
1174
1175 # Wait for exec to fail or succeed; possibly raising exception
1176 data = os.read(errpipe_read, 1048576) # Exception limited to 1M
1177 finally:
1178 # be sure the FD is closed no matter what
1179 os.close(errpipe_read)
1180
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001181 if data != "":
Peter Astrandf791d7a2005-01-01 09:38:57 +00001182 os.waitpid(self.pid, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001183 child_exception = pickle.loads(data)
Georg Brandlf3715d22009-02-14 17:01:36 +00001184 for fd in (p2cwrite, c2pread, errread):
1185 if fd is not None:
1186 os.close(fd)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001187 raise child_exception
1188
1189
1190 def _handle_exitstatus(self, sts):
1191 if os.WIFSIGNALED(sts):
1192 self.returncode = -os.WTERMSIG(sts)
1193 elif os.WIFEXITED(sts):
1194 self.returncode = os.WEXITSTATUS(sts)
1195 else:
1196 # Should never happen
1197 raise RuntimeError("Unknown child exit status!")
1198
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001199
Gregory P. Smitha36f8fe2008-08-04 00:13:29 +00001200 def _internal_poll(self, _deadstate=None):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001201 """Check if child process has terminated. Returns returncode
1202 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +00001203 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001204 try:
1205 pid, sts = os.waitpid(self.pid, os.WNOHANG)
1206 if pid == self.pid:
1207 self._handle_exitstatus(sts)
1208 except os.error:
Martin v. Löwis17de8ff2006-04-10 15:55:37 +00001209 if _deadstate is not None:
1210 self.returncode = _deadstate
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001211 return self.returncode
1212
1213
1214 def wait(self):
1215 """Wait for child process to terminate. Returns returncode
1216 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +00001217 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001218 pid, sts = os.waitpid(self.pid, 0)
1219 self._handle_exitstatus(sts)
1220 return self.returncode
1221
1222
Peter Astrand23109f02005-03-03 20:28:59 +00001223 def _communicate(self, input):
Gregory P. Smithdd7ca242009-07-04 01:49:29 +00001224 if self.stdin:
1225 # Flush stdio buffer. This might block, if the user has
1226 # been writing to .stdin in an uncontrolled fashion.
1227 self.stdin.flush()
1228 if not input:
1229 self.stdin.close()
1230
1231 if _has_poll:
1232 stdout, stderr = self._communicate_with_poll(input)
1233 else:
1234 stdout, stderr = self._communicate_with_select(input)
1235
1236 # All data exchanged. Translate lists into strings.
1237 if stdout is not None:
1238 stdout = ''.join(stdout)
1239 if stderr is not None:
1240 stderr = ''.join(stderr)
1241
1242 # Translate newlines, if requested. We cannot let the file
1243 # object do the translation: It is based on stdio, which is
1244 # impossible to combine with select (unless forcing no
1245 # buffering).
1246 if self.universal_newlines and hasattr(file, 'newlines'):
1247 if stdout:
1248 stdout = self._translate_newlines(stdout)
1249 if stderr:
1250 stderr = self._translate_newlines(stderr)
1251
1252 self.wait()
1253 return (stdout, stderr)
1254
1255
1256 def _communicate_with_poll(self, input):
1257 stdout = None # Return
1258 stderr = None # Return
1259 fd2file = {}
1260 fd2output = {}
1261
1262 poller = select.poll()
1263 def register_and_append(file_obj, eventmask):
1264 poller.register(file_obj.fileno(), eventmask)
1265 fd2file[file_obj.fileno()] = file_obj
1266
1267 def close_unregister_and_remove(fd):
1268 poller.unregister(fd)
1269 fd2file[fd].close()
1270 fd2file.pop(fd)
1271
1272 if self.stdin and input:
1273 register_and_append(self.stdin, select.POLLOUT)
1274
1275 select_POLLIN_POLLPRI = select.POLLIN | select.POLLPRI
1276 if self.stdout:
1277 register_and_append(self.stdout, select_POLLIN_POLLPRI)
1278 fd2output[self.stdout.fileno()] = stdout = []
1279 if self.stderr:
1280 register_and_append(self.stderr, select_POLLIN_POLLPRI)
1281 fd2output[self.stderr.fileno()] = stderr = []
1282
1283 input_offset = 0
1284 while fd2file:
1285 try:
1286 ready = poller.poll()
1287 except select.error, e:
1288 if e.args[0] == errno.EINTR:
1289 continue
1290 raise
1291
1292 for fd, mode in ready:
1293 if mode & select.POLLOUT:
1294 chunk = input[input_offset : input_offset + _PIPE_BUF]
1295 input_offset += os.write(fd, chunk)
1296 if input_offset >= len(input):
1297 close_unregister_and_remove(fd)
1298 elif mode & select_POLLIN_POLLPRI:
1299 data = os.read(fd, 4096)
1300 if not data:
1301 close_unregister_and_remove(fd)
1302 fd2output[fd].append(data)
1303 else:
1304 # Ignore hang up or errors.
1305 close_unregister_and_remove(fd)
1306
1307 return (stdout, stderr)
1308
1309
1310 def _communicate_with_select(self, input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001311 read_set = []
1312 write_set = []
1313 stdout = None # Return
1314 stderr = None # Return
1315
Gregory P. Smithdd7ca242009-07-04 01:49:29 +00001316 if self.stdin and input:
1317 write_set.append(self.stdin)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001318 if self.stdout:
1319 read_set.append(self.stdout)
Tim Peterse718f612004-10-12 21:51:32 +00001320 stdout = []
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001321 if self.stderr:
1322 read_set.append(self.stderr)
1323 stderr = []
1324
Peter Astrand1812f8c2007-01-07 14:34:16 +00001325 input_offset = 0
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001326 while read_set or write_set:
Gregory P. Smithf4140642008-07-06 07:16:40 +00001327 try:
1328 rlist, wlist, xlist = select.select(read_set, write_set, [])
1329 except select.error, e:
1330 if e.args[0] == errno.EINTR:
1331 continue
1332 raise
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001333
1334 if self.stdin in wlist:
Gregory P. Smithdd7ca242009-07-04 01:49:29 +00001335 chunk = input[input_offset : input_offset + _PIPE_BUF]
Brett Cannon03446c42008-08-08 04:19:32 +00001336 bytes_written = os.write(self.stdin.fileno(), chunk)
Tim Petersf733abb2007-01-30 03:03:46 +00001337 input_offset += bytes_written
Peter Astrand1812f8c2007-01-07 14:34:16 +00001338 if input_offset >= len(input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001339 self.stdin.close()
1340 write_set.remove(self.stdin)
1341
1342 if self.stdout in rlist:
1343 data = os.read(self.stdout.fileno(), 1024)
1344 if data == "":
1345 self.stdout.close()
1346 read_set.remove(self.stdout)
1347 stdout.append(data)
1348
1349 if self.stderr in rlist:
1350 data = os.read(self.stderr.fileno(), 1024)
1351 if data == "":
1352 self.stderr.close()
1353 read_set.remove(self.stderr)
1354 stderr.append(data)
1355
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001356 return (stdout, stderr)
1357
Gregory P. Smithdd7ca242009-07-04 01:49:29 +00001358
Christian Heimese74c8f22008-04-19 02:23:57 +00001359 def send_signal(self, sig):
1360 """Send a signal to the process
1361 """
1362 os.kill(self.pid, sig)
1363
1364 def terminate(self):
1365 """Terminate the process with SIGTERM
1366 """
1367 self.send_signal(signal.SIGTERM)
1368
1369 def kill(self):
1370 """Kill the process with SIGKILL
1371 """
1372 self.send_signal(signal.SIGKILL)
1373
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001374
1375def _demo_posix():
1376 #
1377 # Example 1: Simple redirection: Get process list
1378 #
1379 plist = Popen(["ps"], stdout=PIPE).communicate()[0]
1380 print "Process list:"
1381 print plist
1382
1383 #
1384 # Example 2: Change uid before executing child
1385 #
1386 if os.getuid() == 0:
1387 p = Popen(["id"], preexec_fn=lambda: os.setuid(100))
1388 p.wait()
1389
1390 #
1391 # Example 3: Connecting several subprocesses
1392 #
1393 print "Looking for 'hda'..."
1394 p1 = Popen(["dmesg"], stdout=PIPE)
1395 p2 = Popen(["grep", "hda"], stdin=p1.stdout, stdout=PIPE)
1396 print repr(p2.communicate()[0])
1397
1398 #
1399 # Example 4: Catch execution error
1400 #
1401 print
1402 print "Trying a weird file..."
1403 try:
1404 print Popen(["/this/path/does/not/exist"]).communicate()
1405 except OSError, e:
1406 if e.errno == errno.ENOENT:
1407 print "The file didn't exist. I thought so..."
1408 print "Child traceback:"
1409 print e.child_traceback
1410 else:
1411 print "Error", e.errno
1412 else:
1413 print >>sys.stderr, "Gosh. No error."
1414
1415
1416def _demo_windows():
1417 #
1418 # Example 1: Connecting several subprocesses
1419 #
1420 print "Looking for 'PROMPT' in set output..."
1421 p1 = Popen("set", stdout=PIPE, shell=True)
1422 p2 = Popen('find "PROMPT"', stdin=p1.stdout, stdout=PIPE)
1423 print repr(p2.communicate()[0])
1424
1425 #
1426 # Example 2: Simple execution of program
1427 #
1428 print "Executing calc..."
1429 p = Popen("calc")
1430 p.wait()
1431
1432
1433if __name__ == "__main__":
1434 if mswindows:
1435 _demo_windows()
1436 else:
1437 _demo_posix()