blob: 496430789936dfd97c35e0220d2ba20cc1ef31b7 [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
110This module also defines two shortcut functions:
111
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
130Exceptions
131----------
132Exceptions raised in the child process, before the new program has
133started to execute, will be re-raised in the parent. Additionally,
134the exception object will have one extra attribute called
135'child_traceback', which is a string containing traceback information
136from the childs point of view.
137
138The most common exception raised is OSError. This occurs, for
139example, when trying to execute a non-existent file. Applications
140should prepare for OSErrors.
141
142A ValueError will be raised if Popen is called with invalid arguments.
143
Peter Astrand7d1d4362006-07-14 14:04:45 +0000144check_call() will raise CalledProcessError, if the called process
145returns a non-zero return code.
Peter Astrand454f7672005-01-01 09:36:35 +0000146
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000147
148Security
149--------
150Unlike some other popen functions, this implementation will never call
151/bin/sh implicitly. This means that all characters, including shell
152metacharacters, can safely be passed to child processes.
153
154
155Popen objects
156=============
157Instances of the Popen class have the following methods:
158
159poll()
160 Check if child process has terminated. Returns returncode
161 attribute.
162
163wait()
164 Wait for child process to terminate. Returns returncode attribute.
165
166communicate(input=None)
167 Interact with process: Send data to stdin. Read data from stdout
168 and stderr, until end-of-file is reached. Wait for process to
Neal Norwitza186ee22006-12-29 03:01:53 +0000169 terminate. The optional input argument should be a string to be
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000170 sent to the child process, or None, if no data should be sent to
171 the child.
Tim Peterse718f612004-10-12 21:51:32 +0000172
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000173 communicate() returns a tuple (stdout, stderr).
174
175 Note: The data read is buffered in memory, so do not use this
176 method if the data size is large or unlimited.
177
178The following attributes are also available:
179
180stdin
181 If the stdin argument is PIPE, this attribute is a file object
182 that provides input to the child process. Otherwise, it is None.
183
184stdout
185 If the stdout argument is PIPE, this attribute is a file object
186 that provides output from the child process. Otherwise, it is
187 None.
188
189stderr
190 If the stderr argument is PIPE, this attribute is file object that
191 provides error output from the child process. Otherwise, it is
192 None.
193
194pid
195 The process ID of the child process.
196
197returncode
198 The child return code. A None value indicates that the process
199 hasn't terminated yet. A negative value -N indicates that the
200 child was terminated by signal N (UNIX only).
201
202
203Replacing older functions with the subprocess module
204====================================================
205In this section, "a ==> b" means that b can be used as a replacement
206for a.
207
208Note: All functions in this section fail (more or less) silently if
209the executed program cannot be found; this module raises an OSError
210exception.
211
212In the following examples, we assume that the subprocess module is
213imported with "from subprocess import *".
214
215
216Replacing /bin/sh shell backquote
217---------------------------------
218output=`mycmd myarg`
219==>
220output = Popen(["mycmd", "myarg"], stdout=PIPE).communicate()[0]
221
222
223Replacing shell pipe line
224-------------------------
225output=`dmesg | grep hda`
226==>
227p1 = Popen(["dmesg"], stdout=PIPE)
Peter Astrand6fdf3cb2004-11-30 18:06:42 +0000228p2 = Popen(["grep", "hda"], stdin=p1.stdout, stdout=PIPE)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000229output = p2.communicate()[0]
230
231
232Replacing os.system()
233---------------------
234sts = os.system("mycmd" + " myarg")
235==>
236p = Popen("mycmd" + " myarg", shell=True)
Neal Norwitz84404832006-07-10 00:05:34 +0000237pid, sts = os.waitpid(p.pid, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000238
239Note:
240
241* Calling the program through the shell is usually not required.
242
243* It's easier to look at the returncode attribute than the
244 exitstatus.
245
246A more real-world example would look like this:
247
248try:
249 retcode = call("mycmd" + " myarg", shell=True)
250 if retcode < 0:
251 print >>sys.stderr, "Child was terminated by signal", -retcode
252 else:
253 print >>sys.stderr, "Child returned", retcode
254except OSError, e:
255 print >>sys.stderr, "Execution failed:", e
256
257
258Replacing os.spawn*
259-------------------
Tim Peterse718f612004-10-12 21:51:32 +0000260P_NOWAIT example:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000261
262pid = os.spawnlp(os.P_NOWAIT, "/bin/mycmd", "mycmd", "myarg")
263==>
264pid = Popen(["/bin/mycmd", "myarg"]).pid
265
266
267P_WAIT example:
268
269retcode = os.spawnlp(os.P_WAIT, "/bin/mycmd", "mycmd", "myarg")
270==>
271retcode = call(["/bin/mycmd", "myarg"])
272
273
Tim Peterse718f612004-10-12 21:51:32 +0000274Vector example:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000275
276os.spawnvp(os.P_NOWAIT, path, args)
277==>
278Popen([path] + args[1:])
279
280
Tim Peterse718f612004-10-12 21:51:32 +0000281Environment example:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000282
283os.spawnlpe(os.P_NOWAIT, "/bin/mycmd", "mycmd", "myarg", env)
284==>
285Popen(["/bin/mycmd", "myarg"], env={"PATH": "/usr/bin"})
286
287
Tim Peterse718f612004-10-12 21:51:32 +0000288Replacing os.popen*
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000289-------------------
Philip Jenvey78652962009-09-29 19:18:11 +0000290pipe = os.popen("cmd", mode='r', bufsize)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000291==>
Philip Jenvey78652962009-09-29 19:18:11 +0000292pipe = Popen("cmd", shell=True, bufsize=bufsize, stdout=PIPE).stdout
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000293
Philip Jenvey78652962009-09-29 19:18:11 +0000294pipe = os.popen("cmd", mode='w', bufsize)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000295==>
Philip Jenvey78652962009-09-29 19:18:11 +0000296pipe = Popen("cmd", shell=True, bufsize=bufsize, stdin=PIPE).stdin
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000297
298
Philip Jenvey78652962009-09-29 19:18:11 +0000299(child_stdin, child_stdout) = os.popen2("cmd", mode, bufsize)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000300==>
Philip Jenvey78652962009-09-29 19:18:11 +0000301p = Popen("cmd", shell=True, bufsize=bufsize,
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000302 stdin=PIPE, stdout=PIPE, close_fds=True)
303(child_stdin, child_stdout) = (p.stdin, p.stdout)
304
305
306(child_stdin,
307 child_stdout,
Philip Jenvey78652962009-09-29 19:18:11 +0000308 child_stderr) = os.popen3("cmd", mode, bufsize)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000309==>
Philip Jenvey78652962009-09-29 19:18:11 +0000310p = Popen("cmd", shell=True, bufsize=bufsize,
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000311 stdin=PIPE, stdout=PIPE, stderr=PIPE, close_fds=True)
312(child_stdin,
313 child_stdout,
314 child_stderr) = (p.stdin, p.stdout, p.stderr)
315
316
Philip Jenvey78652962009-09-29 19:18:11 +0000317(child_stdin, child_stdout_and_stderr) = os.popen4("cmd", mode,
318 bufsize)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000319==>
Philip Jenvey78652962009-09-29 19:18:11 +0000320p = Popen("cmd", shell=True, bufsize=bufsize,
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000321 stdin=PIPE, stdout=PIPE, stderr=STDOUT, close_fds=True)
322(child_stdin, child_stdout_and_stderr) = (p.stdin, p.stdout)
323
Philip Jenvey78652962009-09-29 19:18:11 +0000324On Unix, os.popen2, os.popen3 and os.popen4 also accept a sequence as
325the command to execute, in which case arguments will be passed
326directly to the program without shell intervention. This usage can be
327replaced as follows:
328
329(child_stdin, child_stdout) = os.popen2(["/bin/ls", "-l"], mode,
330 bufsize)
331==>
332p = Popen(["/bin/ls", "-l"], bufsize=bufsize, stdin=PIPE, stdout=PIPE)
333(child_stdin, child_stdout) = (p.stdin, p.stdout)
334
335Return code handling translates as follows:
336
337pipe = os.popen("cmd", 'w')
338...
339rc = pipe.close()
340if rc != None and rc % 256:
341 print "There were some errors"
342==>
343process = Popen("cmd", 'w', shell=True, stdin=PIPE)
344...
345process.stdin.close()
346if process.wait() != 0:
347 print "There were some errors"
348
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000349
350Replacing popen2.*
351------------------
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000352(child_stdout, child_stdin) = popen2.popen2("somestring", bufsize, mode)
353==>
354p = Popen(["somestring"], shell=True, bufsize=bufsize
355 stdin=PIPE, stdout=PIPE, close_fds=True)
356(child_stdout, child_stdin) = (p.stdout, p.stdin)
357
Philip Jenvey78652962009-09-29 19:18:11 +0000358On Unix, popen2 also accepts a sequence as the command to execute, in
359which case arguments will be passed directly to the program without
360shell intervention. This usage can be replaced as follows:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000361
Philip Jenvey78652962009-09-29 19:18:11 +0000362(child_stdout, child_stdin) = popen2.popen2(["mycmd", "myarg"], bufsize,
363 mode)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000364==>
365p = Popen(["mycmd", "myarg"], bufsize=bufsize,
366 stdin=PIPE, stdout=PIPE, close_fds=True)
367(child_stdout, child_stdin) = (p.stdout, p.stdin)
368
Neal Norwitzaa87fb62007-05-11 06:23:01 +0000369The popen2.Popen3 and popen2.Popen4 basically works as subprocess.Popen,
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000370except that:
371
372* subprocess.Popen raises an exception if the execution fails
373* the capturestderr argument is replaced with the stderr argument.
374* stdin=PIPE and stdout=PIPE must be specified.
375* popen2 closes all filedescriptors by default, but you have to specify
Tim Peterse718f612004-10-12 21:51:32 +0000376 close_fds=True with subprocess.Popen.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000377"""
378
379import sys
380mswindows = (sys.platform == "win32")
381
382import os
Peter Astrandc26516b2005-02-21 08:13:02 +0000383import types
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000384import traceback
Gregory P. Smith87d49792008-01-19 20:57:59 +0000385import gc
Christian Heimese74c8f22008-04-19 02:23:57 +0000386import signal
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000387
Peter Astrand454f7672005-01-01 09:36:35 +0000388# Exception classes used by this module.
Peter Astrand7d1d4362006-07-14 14:04:45 +0000389class CalledProcessError(Exception):
Peter Astrand454f7672005-01-01 09:36:35 +0000390 """This exception is raised when a process run by check_call() returns
391 a non-zero exit status. The exit status will be stored in the
Peter Astrand7d1d4362006-07-14 14:04:45 +0000392 returncode attribute."""
393 def __init__(self, returncode, cmd):
394 self.returncode = returncode
395 self.cmd = cmd
396 def __str__(self):
397 return "Command '%s' returned non-zero exit status %d" % (self.cmd, self.returncode)
Tim Peters73a9ead2006-07-18 21:55:15 +0000398
Peter Astrand454f7672005-01-01 09:36:35 +0000399
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000400if mswindows:
401 import threading
402 import msvcrt
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000403 if 0: # <-- change this to use pywin32 instead of the _subprocess driver
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000404 import pywintypes
Tim Peterse8374a52004-10-13 03:15:00 +0000405 from win32api import GetStdHandle, STD_INPUT_HANDLE, \
406 STD_OUTPUT_HANDLE, STD_ERROR_HANDLE
407 from win32api import GetCurrentProcess, DuplicateHandle, \
408 GetModuleFileName, GetVersion
Peter Astrandc1d65362004-11-07 14:30:34 +0000409 from win32con import DUPLICATE_SAME_ACCESS, SW_HIDE
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000410 from win32pipe import CreatePipe
Tim Peterse8374a52004-10-13 03:15:00 +0000411 from win32process import CreateProcess, STARTUPINFO, \
412 GetExitCodeProcess, STARTF_USESTDHANDLES, \
Peter Astrandc1d65362004-11-07 14:30:34 +0000413 STARTF_USESHOWWINDOW, CREATE_NEW_CONSOLE
Christian Heimese74c8f22008-04-19 02:23:57 +0000414 from win32process import TerminateProcess
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000415 from win32event import WaitForSingleObject, INFINITE, WAIT_OBJECT_0
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000416 else:
417 from _subprocess import *
418 class STARTUPINFO:
419 dwFlags = 0
420 hStdInput = None
421 hStdOutput = None
422 hStdError = None
Georg Brandlad624892006-06-04 22:15:37 +0000423 wShowWindow = 0
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000424 class pywintypes:
425 error = IOError
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000426else:
427 import select
428 import errno
429 import fcntl
430 import pickle
431
Peter Astrand454f7672005-01-01 09:36:35 +0000432__all__ = ["Popen", "PIPE", "STDOUT", "call", "check_call", "CalledProcessError"]
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000433
434try:
435 MAXFD = os.sysconf("SC_OPEN_MAX")
436except:
437 MAXFD = 256
438
439# True/False does not exist on 2.2.0
Brett Cannon03446c42008-08-08 04:19:32 +0000440#try:
441# False
442#except NameError:
443# False = 0
444# True = 1
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000445
446_active = []
447
448def _cleanup():
449 for inst in _active[:]:
Gregory P. Smitha36f8fe2008-08-04 00:13:29 +0000450 if inst._internal_poll(_deadstate=sys.maxint) >= 0:
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000451 try:
452 _active.remove(inst)
453 except ValueError:
454 # This can happen if two threads create a new Popen instance.
455 # It's harmless that it was already removed, so ignore.
456 pass
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000457
458PIPE = -1
459STDOUT = -2
460
461
Gregory P. Smithca574992010-03-01 00:35:34 +0000462def _eintr_retry_call(func, *args):
463 while True:
464 try:
465 return func(*args)
466 except OSError, e:
467 if e.errno == errno.EINTR:
468 continue
469 raise
470
471
Peter Astrand5f5e1412004-12-05 20:15:36 +0000472def call(*popenargs, **kwargs):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000473 """Run command with arguments. Wait for command to complete, then
474 return the returncode attribute.
475
476 The arguments are the same as for the Popen constructor. Example:
477
478 retcode = call(["ls", "-l"])
479 """
Peter Astrand5f5e1412004-12-05 20:15:36 +0000480 return Popen(*popenargs, **kwargs).wait()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000481
482
Peter Astrand454f7672005-01-01 09:36:35 +0000483def check_call(*popenargs, **kwargs):
484 """Run command with arguments. Wait for command to complete. If
485 the exit code was zero then return, otherwise raise
486 CalledProcessError. The CalledProcessError object will have the
Peter Astrand7d1d4362006-07-14 14:04:45 +0000487 return code in the returncode attribute.
Peter Astrand454f7672005-01-01 09:36:35 +0000488
489 The arguments are the same as for the Popen constructor. Example:
490
491 check_call(["ls", "-l"])
492 """
493 retcode = call(*popenargs, **kwargs)
494 cmd = kwargs.get("args")
495 if cmd is None:
496 cmd = popenargs[0]
497 if retcode:
Peter Astrand7d1d4362006-07-14 14:04:45 +0000498 raise CalledProcessError(retcode, cmd)
Peter Astrand454f7672005-01-01 09:36:35 +0000499 return retcode
500
501
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000502def list2cmdline(seq):
503 """
504 Translate a sequence of arguments into a command line
505 string, using the same rules as the MS C runtime:
506
507 1) Arguments are delimited by white space, which is either a
508 space or a tab.
509
510 2) A string surrounded by double quotation marks is
511 interpreted as a single argument, regardless of white space
Gregory P. Smith70eb2f92008-01-19 22:49:37 +0000512 or pipe characters contained within. A quoted string can be
513 embedded in an argument.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000514
515 3) A double quotation mark preceded by a backslash is
516 interpreted as a literal double quotation mark.
517
518 4) Backslashes are interpreted literally, unless they
519 immediately precede a double quotation mark.
520
521 5) If backslashes immediately precede a double quotation mark,
522 every pair of backslashes is interpreted as a literal
523 backslash. If the number of backslashes is odd, the last
524 backslash escapes the next double quotation mark as
525 described in rule 3.
526 """
527
528 # See
Eric Smithf4b581e2009-11-09 15:20:16 +0000529 # http://msdn.microsoft.com/en-us/library/17w5ykft.aspx
530 # or search http://msdn.microsoft.com for
531 # "Parsing C++ Command-Line Arguments"
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000532 result = []
533 needquote = False
534 for arg in seq:
535 bs_buf = []
536
537 # Add a space to separate this argument from the others
538 if result:
539 result.append(' ')
540
Gregory P. Smith70eb2f92008-01-19 22:49:37 +0000541 needquote = (" " in arg) or ("\t" in arg) or ("|" in arg) or not arg
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000542 if needquote:
543 result.append('"')
544
545 for c in arg:
546 if c == '\\':
Tim Peterse718f612004-10-12 21:51:32 +0000547 # Don't know if we need to double yet.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000548 bs_buf.append(c)
549 elif c == '"':
Gregory P. Smithe047e6d2008-01-19 20:49:02 +0000550 # Double backslashes.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000551 result.append('\\' * len(bs_buf)*2)
552 bs_buf = []
553 result.append('\\"')
554 else:
555 # Normal char
556 if bs_buf:
557 result.extend(bs_buf)
558 bs_buf = []
559 result.append(c)
560
Gregory P. Smithe047e6d2008-01-19 20:49:02 +0000561 # Add remaining backslashes, if any.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000562 if bs_buf:
563 result.extend(bs_buf)
564
565 if needquote:
Peter Astrand7e78ade2005-03-03 21:10:23 +0000566 result.extend(bs_buf)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000567 result.append('"')
568
569 return ''.join(result)
570
571
572class Popen(object):
573 def __init__(self, args, bufsize=0, executable=None,
574 stdin=None, stdout=None, stderr=None,
575 preexec_fn=None, close_fds=False, shell=False,
576 cwd=None, env=None, universal_newlines=False,
577 startupinfo=None, creationflags=0):
578 """Create new Popen instance."""
579 _cleanup()
580
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000581 self._child_created = False
Peter Astrand738131d2004-11-30 21:04:45 +0000582 if not isinstance(bufsize, (int, long)):
583 raise TypeError("bufsize must be an integer")
584
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000585 if mswindows:
Tim Peterse8374a52004-10-13 03:15:00 +0000586 if preexec_fn is not None:
587 raise ValueError("preexec_fn is not supported on Windows "
588 "platforms")
Peter Astrand81a191b2007-05-26 22:18:20 +0000589 if close_fds and (stdin is not None or stdout is not None or
590 stderr is not None):
Tim Peterse8374a52004-10-13 03:15:00 +0000591 raise ValueError("close_fds is not supported on Windows "
Peter Astrand81a191b2007-05-26 22:18:20 +0000592 "platforms if you redirect stdin/stdout/stderr")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000593 else:
594 # POSIX
Tim Peterse8374a52004-10-13 03:15:00 +0000595 if startupinfo is not None:
596 raise ValueError("startupinfo is only supported on Windows "
597 "platforms")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000598 if creationflags != 0:
Tim Peterse8374a52004-10-13 03:15:00 +0000599 raise ValueError("creationflags is only supported on Windows "
600 "platforms")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000601
Tim Peterse718f612004-10-12 21:51:32 +0000602 self.stdin = None
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000603 self.stdout = None
604 self.stderr = None
605 self.pid = None
606 self.returncode = None
607 self.universal_newlines = universal_newlines
608
609 # Input and output objects. The general principle is like
610 # this:
611 #
612 # Parent Child
613 # ------ -----
614 # p2cwrite ---stdin---> p2cread
615 # c2pread <--stdout--- c2pwrite
616 # errread <--stderr--- errwrite
617 #
618 # On POSIX, the child objects are file descriptors. On
619 # Windows, these are Windows file handles. The parent objects
620 # are file descriptors on both platforms. The parent objects
621 # are None when not using PIPEs. The child objects are None
622 # when not redirecting.
Tim Peterse718f612004-10-12 21:51:32 +0000623
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000624 (p2cread, p2cwrite,
625 c2pread, c2pwrite,
626 errread, errwrite) = self._get_handles(stdin, stdout, stderr)
627
628 self._execute_child(args, executable, preexec_fn, close_fds,
629 cwd, env, universal_newlines,
630 startupinfo, creationflags, shell,
631 p2cread, p2cwrite,
632 c2pread, c2pwrite,
633 errread, errwrite)
634
Peter Astrand5f9c6ae2007-02-06 15:37:50 +0000635 if mswindows:
Hirokazu Yamamoto857c00e2009-03-03 22:55:00 +0000636 if p2cwrite is not None:
637 p2cwrite = msvcrt.open_osfhandle(p2cwrite.Detach(), 0)
638 if c2pread is not None:
639 c2pread = msvcrt.open_osfhandle(c2pread.Detach(), 0)
640 if errread is not None:
641 errread = msvcrt.open_osfhandle(errread.Detach(), 0)
Peter Astrand5f9c6ae2007-02-06 15:37:50 +0000642
Peter Astrandf5400032007-02-02 19:06:36 +0000643 if p2cwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000644 self.stdin = os.fdopen(p2cwrite, 'wb', bufsize)
Peter Astrandf5400032007-02-02 19:06:36 +0000645 if c2pread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000646 if universal_newlines:
647 self.stdout = os.fdopen(c2pread, 'rU', bufsize)
648 else:
649 self.stdout = os.fdopen(c2pread, 'rb', bufsize)
Peter Astrandf5400032007-02-02 19:06:36 +0000650 if errread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000651 if universal_newlines:
652 self.stderr = os.fdopen(errread, 'rU', bufsize)
653 else:
654 self.stderr = os.fdopen(errread, 'rb', bufsize)
Tim Peterse718f612004-10-12 21:51:32 +0000655
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000656
657 def _translate_newlines(self, data):
658 data = data.replace("\r\n", "\n")
659 data = data.replace("\r", "\n")
660 return data
661
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000662
Georg Brandl24522982007-04-21 20:35:38 +0000663 def __del__(self, sys=sys):
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000664 if not self._child_created:
665 # We didn't get to successfully create a child process.
666 return
667 # In case the child hasn't been waited on, check if it's done.
Gregory P. Smitha36f8fe2008-08-04 00:13:29 +0000668 self._internal_poll(_deadstate=sys.maxint)
Georg Brandl13cf38c2006-07-20 16:28:39 +0000669 if self.returncode is None and _active is not None:
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000670 # Child is still running, keep us alive until we can wait on it.
671 _active.append(self)
672
673
Peter Astrand23109f02005-03-03 20:28:59 +0000674 def communicate(self, input=None):
675 """Interact with process: Send data to stdin. Read data from
676 stdout and stderr, until end-of-file is reached. Wait for
677 process to terminate. The optional input argument should be a
678 string to be sent to the child process, or None, if no data
679 should be sent to the child.
Tim Peterseba28be2005-03-28 01:08:02 +0000680
Peter Astrand23109f02005-03-03 20:28:59 +0000681 communicate() returns a tuple (stdout, stderr)."""
682
683 # Optimization: If we are only using one pipe, or no pipe at
684 # all, using select() or threads is unnecessary.
685 if [self.stdin, self.stdout, self.stderr].count(None) >= 2:
Tim Peterseba28be2005-03-28 01:08:02 +0000686 stdout = None
687 stderr = None
Peter Astrand23109f02005-03-03 20:28:59 +0000688 if self.stdin:
689 if input:
690 self.stdin.write(input)
691 self.stdin.close()
692 elif self.stdout:
693 stdout = self.stdout.read()
Gregory P. Smith4036fd42008-05-26 20:22:14 +0000694 self.stdout.close()
Peter Astrand23109f02005-03-03 20:28:59 +0000695 elif self.stderr:
696 stderr = self.stderr.read()
Gregory P. Smith4036fd42008-05-26 20:22:14 +0000697 self.stderr.close()
Peter Astrand23109f02005-03-03 20:28:59 +0000698 self.wait()
699 return (stdout, stderr)
Tim Peterseba28be2005-03-28 01:08:02 +0000700
Peter Astrand23109f02005-03-03 20:28:59 +0000701 return self._communicate(input)
702
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000703
Gregory P. Smitha36f8fe2008-08-04 00:13:29 +0000704 def poll(self):
705 return self._internal_poll()
706
707
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000708 if mswindows:
709 #
710 # Windows methods
711 #
712 def _get_handles(self, stdin, stdout, stderr):
713 """Construct and return tupel with IO objects:
714 p2cread, p2cwrite, c2pread, c2pwrite, errread, errwrite
715 """
Peter Astrandd38ddf42005-02-10 08:32:50 +0000716 if stdin is None and stdout is None and stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000717 return (None, None, None, None, None, None)
Tim Peterse718f612004-10-12 21:51:32 +0000718
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000719 p2cread, p2cwrite = None, None
720 c2pread, c2pwrite = None, None
721 errread, errwrite = None, None
722
Peter Astrandd38ddf42005-02-10 08:32:50 +0000723 if stdin is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000724 p2cread = GetStdHandle(STD_INPUT_HANDLE)
Hirokazu Yamamoto857c00e2009-03-03 22:55:00 +0000725 if p2cread is None:
726 p2cread, _ = CreatePipe(None, 0)
727 elif stdin == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000728 p2cread, p2cwrite = CreatePipe(None, 0)
Peter Astrandd38ddf42005-02-10 08:32:50 +0000729 elif isinstance(stdin, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000730 p2cread = msvcrt.get_osfhandle(stdin)
731 else:
732 # Assuming file-like object
733 p2cread = msvcrt.get_osfhandle(stdin.fileno())
734 p2cread = self._make_inheritable(p2cread)
735
Peter Astrandd38ddf42005-02-10 08:32:50 +0000736 if stdout is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000737 c2pwrite = GetStdHandle(STD_OUTPUT_HANDLE)
Hirokazu Yamamoto857c00e2009-03-03 22:55:00 +0000738 if c2pwrite is None:
739 _, c2pwrite = CreatePipe(None, 0)
740 elif stdout == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000741 c2pread, c2pwrite = CreatePipe(None, 0)
Peter Astrandd38ddf42005-02-10 08:32:50 +0000742 elif isinstance(stdout, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000743 c2pwrite = msvcrt.get_osfhandle(stdout)
744 else:
745 # Assuming file-like object
746 c2pwrite = msvcrt.get_osfhandle(stdout.fileno())
747 c2pwrite = self._make_inheritable(c2pwrite)
748
Peter Astrandd38ddf42005-02-10 08:32:50 +0000749 if stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000750 errwrite = GetStdHandle(STD_ERROR_HANDLE)
Hirokazu Yamamoto857c00e2009-03-03 22:55:00 +0000751 if errwrite is None:
752 _, errwrite = CreatePipe(None, 0)
753 elif stderr == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000754 errread, errwrite = CreatePipe(None, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000755 elif stderr == STDOUT:
756 errwrite = c2pwrite
Peter Astrandd38ddf42005-02-10 08:32:50 +0000757 elif isinstance(stderr, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000758 errwrite = msvcrt.get_osfhandle(stderr)
759 else:
760 # Assuming file-like object
761 errwrite = msvcrt.get_osfhandle(stderr.fileno())
762 errwrite = self._make_inheritable(errwrite)
763
764 return (p2cread, p2cwrite,
765 c2pread, c2pwrite,
766 errread, errwrite)
767
768
769 def _make_inheritable(self, handle):
770 """Return a duplicate of handle, which is inheritable"""
771 return DuplicateHandle(GetCurrentProcess(), handle,
772 GetCurrentProcess(), 0, 1,
773 DUPLICATE_SAME_ACCESS)
774
775
776 def _find_w9xpopen(self):
777 """Find and return absolut path to w9xpopen.exe"""
Tim Peterse8374a52004-10-13 03:15:00 +0000778 w9xpopen = os.path.join(os.path.dirname(GetModuleFileName(0)),
779 "w9xpopen.exe")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000780 if not os.path.exists(w9xpopen):
781 # Eeek - file-not-found - possibly an embedding
782 # situation - see if we can locate it in sys.exec_prefix
Tim Peterse8374a52004-10-13 03:15:00 +0000783 w9xpopen = os.path.join(os.path.dirname(sys.exec_prefix),
784 "w9xpopen.exe")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000785 if not os.path.exists(w9xpopen):
Tim Peterse8374a52004-10-13 03:15:00 +0000786 raise RuntimeError("Cannot locate w9xpopen.exe, which is "
787 "needed for Popen to work with your "
788 "shell or platform.")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000789 return w9xpopen
790
Tim Peterse718f612004-10-12 21:51:32 +0000791
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000792 def _execute_child(self, args, executable, preexec_fn, close_fds,
793 cwd, env, universal_newlines,
794 startupinfo, creationflags, shell,
795 p2cread, p2cwrite,
796 c2pread, c2pwrite,
797 errread, errwrite):
798 """Execute program (MS Windows version)"""
799
Peter Astrandc26516b2005-02-21 08:13:02 +0000800 if not isinstance(args, types.StringTypes):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000801 args = list2cmdline(args)
802
Peter Astrandc1d65362004-11-07 14:30:34 +0000803 # Process startup details
Peter Astrandd38ddf42005-02-10 08:32:50 +0000804 if startupinfo is None:
Georg Brandlad624892006-06-04 22:15:37 +0000805 startupinfo = STARTUPINFO()
806 if None not in (p2cread, c2pwrite, errwrite):
Peter Astrandc1d65362004-11-07 14:30:34 +0000807 startupinfo.dwFlags |= STARTF_USESTDHANDLES
808 startupinfo.hStdInput = p2cread
809 startupinfo.hStdOutput = c2pwrite
810 startupinfo.hStdError = errwrite
811
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000812 if shell:
Georg Brandlad624892006-06-04 22:15:37 +0000813 startupinfo.dwFlags |= STARTF_USESHOWWINDOW
814 startupinfo.wShowWindow = SW_HIDE
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000815 comspec = os.environ.get("COMSPEC", "cmd.exe")
816 args = comspec + " /c " + args
Tim Peterse8374a52004-10-13 03:15:00 +0000817 if (GetVersion() >= 0x80000000L or
818 os.path.basename(comspec).lower() == "command.com"):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000819 # Win9x, or using command.com on NT. We need to
820 # use the w9xpopen intermediate program. For more
821 # information, see KB Q150956
822 # (http://web.archive.org/web/20011105084002/http://support.microsoft.com/support/kb/articles/Q150/9/56.asp)
823 w9xpopen = self._find_w9xpopen()
824 args = '"%s" %s' % (w9xpopen, args)
825 # Not passing CREATE_NEW_CONSOLE has been known to
826 # cause random failures on win9x. Specifically a
827 # dialog: "Your program accessed mem currently in
828 # use at xxx" and a hopeful warning about the
829 # stability of your system. Cost is Ctrl+C wont
830 # kill children.
831 creationflags |= CREATE_NEW_CONSOLE
832
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000833 # Start the process
834 try:
835 hp, ht, pid, tid = CreateProcess(executable, args,
Tim Peterse8374a52004-10-13 03:15:00 +0000836 # no special security
837 None, None,
Peter Astrand81a191b2007-05-26 22:18:20 +0000838 int(not close_fds),
Tim Peterse8374a52004-10-13 03:15:00 +0000839 creationflags,
840 env,
841 cwd,
842 startupinfo)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000843 except pywintypes.error, e:
844 # Translate pywintypes.error to WindowsError, which is
845 # a subclass of OSError. FIXME: We should really
846 # translate errno using _sys_errlist (or simliar), but
847 # how can this be done from Python?
848 raise WindowsError(*e.args)
849
850 # Retain the process handle, but close the thread handle
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000851 self._child_created = True
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000852 self._handle = hp
853 self.pid = pid
854 ht.Close()
855
Andrew M. Kuchling51ee66e2004-10-12 16:38:42 +0000856 # Child is launched. Close the parent's copy of those pipe
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000857 # handles that only the child should have open. You need
858 # to make sure that no handles to the write end of the
859 # output pipe are maintained in this process or else the
860 # pipe will not close when the child process exits and the
861 # ReadFile will hang.
Peter Astrandd38ddf42005-02-10 08:32:50 +0000862 if p2cread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000863 p2cread.Close()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000864 if c2pwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000865 c2pwrite.Close()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000866 if errwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000867 errwrite.Close()
868
Tim Peterse718f612004-10-12 21:51:32 +0000869
Gregory P. Smitha36f8fe2008-08-04 00:13:29 +0000870 def _internal_poll(self, _deadstate=None):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000871 """Check if child process has terminated. Returns returncode
872 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +0000873 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000874 if WaitForSingleObject(self._handle, 0) == WAIT_OBJECT_0:
875 self.returncode = GetExitCodeProcess(self._handle)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000876 return self.returncode
877
878
879 def wait(self):
880 """Wait for child process to terminate. Returns returncode
881 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +0000882 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000883 obj = WaitForSingleObject(self._handle, INFINITE)
884 self.returncode = GetExitCodeProcess(self._handle)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000885 return self.returncode
886
887
888 def _readerthread(self, fh, buffer):
889 buffer.append(fh.read())
890
891
Peter Astrand23109f02005-03-03 20:28:59 +0000892 def _communicate(self, input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000893 stdout = None # Return
894 stderr = None # Return
895
896 if self.stdout:
897 stdout = []
Tim Peterse8374a52004-10-13 03:15:00 +0000898 stdout_thread = threading.Thread(target=self._readerthread,
899 args=(self.stdout, stdout))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000900 stdout_thread.setDaemon(True)
901 stdout_thread.start()
902 if self.stderr:
903 stderr = []
Tim Peterse8374a52004-10-13 03:15:00 +0000904 stderr_thread = threading.Thread(target=self._readerthread,
905 args=(self.stderr, stderr))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000906 stderr_thread.setDaemon(True)
907 stderr_thread.start()
908
909 if self.stdin:
Peter Astrandd38ddf42005-02-10 08:32:50 +0000910 if input is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000911 self.stdin.write(input)
912 self.stdin.close()
913
914 if self.stdout:
915 stdout_thread.join()
916 if self.stderr:
917 stderr_thread.join()
918
919 # All data exchanged. Translate lists into strings.
Peter Astrandd38ddf42005-02-10 08:32:50 +0000920 if stdout is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000921 stdout = stdout[0]
Peter Astrandd38ddf42005-02-10 08:32:50 +0000922 if stderr is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000923 stderr = stderr[0]
924
925 # Translate newlines, if requested. We cannot let the file
926 # object do the translation: It is based on stdio, which is
927 # impossible to combine with select (unless forcing no
928 # buffering).
Neal Norwitza6d01ce2006-05-02 06:23:22 +0000929 if self.universal_newlines and hasattr(file, 'newlines'):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000930 if stdout:
931 stdout = self._translate_newlines(stdout)
932 if stderr:
933 stderr = self._translate_newlines(stderr)
934
935 self.wait()
936 return (stdout, stderr)
937
Christian Heimese74c8f22008-04-19 02:23:57 +0000938 def send_signal(self, sig):
939 """Send a signal to the process
940 """
941 if sig == signal.SIGTERM:
942 self.terminate()
943 else:
944 raise ValueError("Only SIGTERM is supported on Windows")
945
946 def terminate(self):
947 """Terminates the process
948 """
949 TerminateProcess(self._handle, 1)
950
951 kill = terminate
952
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000953 else:
954 #
955 # POSIX methods
956 #
957 def _get_handles(self, stdin, stdout, stderr):
958 """Construct and return tupel with IO objects:
959 p2cread, p2cwrite, c2pread, c2pwrite, errread, errwrite
960 """
961 p2cread, p2cwrite = None, None
962 c2pread, c2pwrite = None, None
963 errread, errwrite = None, None
964
Peter Astrandd38ddf42005-02-10 08:32:50 +0000965 if stdin is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000966 pass
967 elif stdin == PIPE:
968 p2cread, p2cwrite = os.pipe()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000969 elif isinstance(stdin, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000970 p2cread = stdin
971 else:
972 # Assuming file-like object
973 p2cread = stdin.fileno()
974
Peter Astrandd38ddf42005-02-10 08:32:50 +0000975 if stdout is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000976 pass
977 elif stdout == PIPE:
978 c2pread, c2pwrite = os.pipe()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000979 elif isinstance(stdout, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000980 c2pwrite = stdout
981 else:
982 # Assuming file-like object
Tim Peterse718f612004-10-12 21:51:32 +0000983 c2pwrite = stdout.fileno()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000984
Peter Astrandd38ddf42005-02-10 08:32:50 +0000985 if stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000986 pass
987 elif stderr == PIPE:
988 errread, errwrite = os.pipe()
989 elif stderr == STDOUT:
990 errwrite = c2pwrite
Peter Astrandd38ddf42005-02-10 08:32:50 +0000991 elif isinstance(stderr, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000992 errwrite = stderr
993 else:
994 # Assuming file-like object
995 errwrite = stderr.fileno()
996
997 return (p2cread, p2cwrite,
998 c2pread, c2pwrite,
999 errread, errwrite)
1000
1001
1002 def _set_cloexec_flag(self, fd):
1003 try:
1004 cloexec_flag = fcntl.FD_CLOEXEC
1005 except AttributeError:
1006 cloexec_flag = 1
1007
1008 old = fcntl.fcntl(fd, fcntl.F_GETFD)
1009 fcntl.fcntl(fd, fcntl.F_SETFD, old | cloexec_flag)
1010
1011
1012 def _close_fds(self, but):
Georg Brandl309501a2008-01-19 20:22:13 +00001013 os.closerange(3, but)
1014 os.closerange(but + 1, MAXFD)
Tim Peterse718f612004-10-12 21:51:32 +00001015
1016
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001017 def _execute_child(self, args, executable, preexec_fn, close_fds,
1018 cwd, env, universal_newlines,
1019 startupinfo, creationflags, shell,
1020 p2cread, p2cwrite,
1021 c2pread, c2pwrite,
1022 errread, errwrite):
1023 """Execute program (POSIX version)"""
1024
Peter Astrandc26516b2005-02-21 08:13:02 +00001025 if isinstance(args, types.StringTypes):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001026 args = [args]
Georg Brandl6c0e1e82006-10-29 09:05:04 +00001027 else:
1028 args = list(args)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001029
1030 if shell:
1031 args = ["/bin/sh", "-c"] + args
1032
Peter Astrandd38ddf42005-02-10 08:32:50 +00001033 if executable is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001034 executable = args[0]
1035
1036 # For transferring possible exec failure from child to parent
1037 # The first char specifies the exception type: 0 means
1038 # OSError, 1 means some other error.
1039 errpipe_read, errpipe_write = os.pipe()
Gregory P. Smith87d49792008-01-19 20:57:59 +00001040 try:
Gregory P. Smith92ffc632008-01-19 22:23:56 +00001041 try:
Facundo Batista07638b92009-06-19 19:51:26 +00001042 self._set_cloexec_flag(errpipe_write)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001043
Facundo Batista07638b92009-06-19 19:51:26 +00001044 gc_was_enabled = gc.isenabled()
1045 # Disable gc to avoid bug where gc -> file_dealloc ->
1046 # write to stderr -> hang. http://bugs.python.org/issue1336
1047 gc.disable()
1048 try:
1049 self.pid = os.fork()
1050 except:
1051 if gc_was_enabled:
1052 gc.enable()
1053 raise
1054 self._child_created = True
1055 if self.pid == 0:
1056 # Child
1057 try:
1058 # Close parent's pipe ends
1059 if p2cwrite is not None:
1060 os.close(p2cwrite)
1061 if c2pread is not None:
1062 os.close(c2pread)
1063 if errread is not None:
1064 os.close(errread)
1065 os.close(errpipe_read)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001066
Facundo Batista07638b92009-06-19 19:51:26 +00001067 # Dup fds for child
1068 if p2cread is not None:
1069 os.dup2(p2cread, 0)
1070 if c2pwrite is not None:
1071 os.dup2(c2pwrite, 1)
1072 if errwrite is not None:
1073 os.dup2(errwrite, 2)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001074
Facundo Batista07638b92009-06-19 19:51:26 +00001075 # Close pipe fds. Make sure we don't close the same
1076 # fd more than once, or standard fds.
1077 if p2cread is not None and p2cread not in (0,):
1078 os.close(p2cread)
1079 if c2pwrite is not None and c2pwrite not in (p2cread, 1):
1080 os.close(c2pwrite)
1081 if errwrite is not None and errwrite not in (p2cread, c2pwrite, 2):
1082 os.close(errwrite)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001083
Facundo Batista07638b92009-06-19 19:51:26 +00001084 # Close all other fds, if asked for
1085 if close_fds:
1086 self._close_fds(but=errpipe_write)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001087
Facundo Batista07638b92009-06-19 19:51:26 +00001088 if cwd is not None:
1089 os.chdir(cwd)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001090
Facundo Batista07638b92009-06-19 19:51:26 +00001091 if preexec_fn:
1092 preexec_fn()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001093
Facundo Batista07638b92009-06-19 19:51:26 +00001094 if env is None:
1095 os.execvp(executable, args)
1096 else:
1097 os.execvpe(executable, args, env)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001098
Facundo Batista07638b92009-06-19 19:51:26 +00001099 except:
1100 exc_type, exc_value, tb = sys.exc_info()
1101 # Save the traceback and attach it to the exception object
1102 exc_lines = traceback.format_exception(exc_type,
1103 exc_value,
1104 tb)
1105 exc_value.child_traceback = ''.join(exc_lines)
1106 os.write(errpipe_write, pickle.dumps(exc_value))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001107
Facundo Batista07638b92009-06-19 19:51:26 +00001108 # This exitcode won't be reported to applications, so it
1109 # really doesn't matter what we return.
1110 os._exit(255)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001111
Facundo Batista07638b92009-06-19 19:51:26 +00001112 # Parent
1113 if gc_was_enabled:
1114 gc.enable()
1115 finally:
1116 # be sure the FD is closed no matter what
1117 os.close(errpipe_write)
1118
1119 if p2cread is not None and p2cwrite is not None:
1120 os.close(p2cread)
1121 if c2pwrite is not None and c2pread is not None:
1122 os.close(c2pwrite)
1123 if errwrite is not None and errread is not None:
1124 os.close(errwrite)
1125
1126 # Wait for exec to fail or succeed; possibly raising exception
Gregory P. Smithca574992010-03-01 00:35:34 +00001127 # Exception limited to 1M
1128 data = _eintr_retry_call(os.read, errpipe_read, 1048576)
Facundo Batista07638b92009-06-19 19:51:26 +00001129 finally:
1130 # be sure the FD is closed no matter what
1131 os.close(errpipe_read)
1132
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001133 if data != "":
Gregory P. Smithca574992010-03-01 00:35:34 +00001134 _eintr_retry_call(os.waitpid, self.pid, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001135 child_exception = pickle.loads(data)
Georg Brandl78162da2009-02-14 17:04:26 +00001136 for fd in (p2cwrite, c2pread, errread):
1137 if fd is not None:
1138 os.close(fd)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001139 raise child_exception
1140
1141
1142 def _handle_exitstatus(self, sts):
1143 if os.WIFSIGNALED(sts):
1144 self.returncode = -os.WTERMSIG(sts)
1145 elif os.WIFEXITED(sts):
1146 self.returncode = os.WEXITSTATUS(sts)
1147 else:
1148 # Should never happen
1149 raise RuntimeError("Unknown child exit status!")
1150
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001151
Gregory P. Smitha36f8fe2008-08-04 00:13:29 +00001152 def _internal_poll(self, _deadstate=None):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001153 """Check if child process has terminated. Returns returncode
1154 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +00001155 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001156 try:
1157 pid, sts = os.waitpid(self.pid, os.WNOHANG)
1158 if pid == self.pid:
1159 self._handle_exitstatus(sts)
1160 except os.error:
Martin v. Löwis17de8ff2006-04-10 15:55:37 +00001161 if _deadstate is not None:
1162 self.returncode = _deadstate
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001163 return self.returncode
1164
1165
1166 def wait(self):
1167 """Wait for child process to terminate. Returns returncode
1168 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +00001169 if self.returncode is None:
Gregory P. Smithca574992010-03-01 00:35:34 +00001170 pid, sts = _eintr_retry_call(os.waitpid, self.pid, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001171 self._handle_exitstatus(sts)
1172 return self.returncode
1173
1174
Peter Astrand23109f02005-03-03 20:28:59 +00001175 def _communicate(self, input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001176 read_set = []
1177 write_set = []
1178 stdout = None # Return
1179 stderr = None # Return
1180
1181 if self.stdin:
1182 # Flush stdio buffer. This might block, if the user has
1183 # been writing to .stdin in an uncontrolled fashion.
1184 self.stdin.flush()
1185 if input:
1186 write_set.append(self.stdin)
1187 else:
1188 self.stdin.close()
1189 if self.stdout:
1190 read_set.append(self.stdout)
Tim Peterse718f612004-10-12 21:51:32 +00001191 stdout = []
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001192 if self.stderr:
1193 read_set.append(self.stderr)
1194 stderr = []
1195
Peter Astrand1812f8c2007-01-07 14:34:16 +00001196 input_offset = 0
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001197 while read_set or write_set:
Gregory P. Smithf4140642008-07-06 07:16:40 +00001198 try:
1199 rlist, wlist, xlist = select.select(read_set, write_set, [])
1200 except select.error, e:
1201 if e.args[0] == errno.EINTR:
1202 continue
1203 raise
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001204
1205 if self.stdin in wlist:
1206 # When select has indicated that the file is writable,
1207 # we can write up to PIPE_BUF bytes without risk
1208 # blocking. POSIX defines PIPE_BUF >= 512
Brett Cannon03446c42008-08-08 04:19:32 +00001209 chunk = input[input_offset : input_offset + 512]
1210 bytes_written = os.write(self.stdin.fileno(), chunk)
Tim Petersf733abb2007-01-30 03:03:46 +00001211 input_offset += bytes_written
Peter Astrand1812f8c2007-01-07 14:34:16 +00001212 if input_offset >= len(input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001213 self.stdin.close()
1214 write_set.remove(self.stdin)
1215
1216 if self.stdout in rlist:
1217 data = os.read(self.stdout.fileno(), 1024)
1218 if data == "":
1219 self.stdout.close()
1220 read_set.remove(self.stdout)
1221 stdout.append(data)
1222
1223 if self.stderr in rlist:
1224 data = os.read(self.stderr.fileno(), 1024)
1225 if data == "":
1226 self.stderr.close()
1227 read_set.remove(self.stderr)
1228 stderr.append(data)
1229
1230 # All data exchanged. Translate lists into strings.
Peter Astrandd38ddf42005-02-10 08:32:50 +00001231 if stdout is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001232 stdout = ''.join(stdout)
Peter Astrandd38ddf42005-02-10 08:32:50 +00001233 if stderr is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001234 stderr = ''.join(stderr)
1235
1236 # Translate newlines, if requested. We cannot let the file
1237 # object do the translation: It is based on stdio, which is
1238 # impossible to combine with select (unless forcing no
1239 # buffering).
Neal Norwitza6d01ce2006-05-02 06:23:22 +00001240 if self.universal_newlines and hasattr(file, 'newlines'):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001241 if stdout:
1242 stdout = self._translate_newlines(stdout)
1243 if stderr:
1244 stderr = self._translate_newlines(stderr)
1245
1246 self.wait()
1247 return (stdout, stderr)
1248
Christian Heimese74c8f22008-04-19 02:23:57 +00001249 def send_signal(self, sig):
1250 """Send a signal to the process
1251 """
1252 os.kill(self.pid, sig)
1253
1254 def terminate(self):
1255 """Terminate the process with SIGTERM
1256 """
1257 self.send_signal(signal.SIGTERM)
1258
1259 def kill(self):
1260 """Kill the process with SIGKILL
1261 """
1262 self.send_signal(signal.SIGKILL)
1263
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001264
1265def _demo_posix():
1266 #
1267 # Example 1: Simple redirection: Get process list
1268 #
1269 plist = Popen(["ps"], stdout=PIPE).communicate()[0]
1270 print "Process list:"
1271 print plist
1272
1273 #
1274 # Example 2: Change uid before executing child
1275 #
1276 if os.getuid() == 0:
1277 p = Popen(["id"], preexec_fn=lambda: os.setuid(100))
1278 p.wait()
1279
1280 #
1281 # Example 3: Connecting several subprocesses
1282 #
1283 print "Looking for 'hda'..."
1284 p1 = Popen(["dmesg"], stdout=PIPE)
1285 p2 = Popen(["grep", "hda"], stdin=p1.stdout, stdout=PIPE)
1286 print repr(p2.communicate()[0])
1287
1288 #
1289 # Example 4: Catch execution error
1290 #
1291 print
1292 print "Trying a weird file..."
1293 try:
1294 print Popen(["/this/path/does/not/exist"]).communicate()
1295 except OSError, e:
1296 if e.errno == errno.ENOENT:
1297 print "The file didn't exist. I thought so..."
1298 print "Child traceback:"
1299 print e.child_traceback
1300 else:
1301 print "Error", e.errno
1302 else:
1303 print >>sys.stderr, "Gosh. No error."
1304
1305
1306def _demo_windows():
1307 #
1308 # Example 1: Connecting several subprocesses
1309 #
1310 print "Looking for 'PROMPT' in set output..."
1311 p1 = Popen("set", stdout=PIPE, shell=True)
1312 p2 = Popen('find "PROMPT"', stdin=p1.stdout, stdout=PIPE)
1313 print repr(p2.communicate()[0])
1314
1315 #
1316 # Example 2: Simple execution of program
1317 #
1318 print "Executing calc..."
1319 p = Popen("calc")
1320 p.wait()
1321
1322
1323if __name__ == "__main__":
1324 if mswindows:
1325 _demo_windows()
1326 else:
1327 _demo_posix()