blob: 25248dc0d31d8c891ea463eeb274ac30d66ad27d [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()
Florent Xiclunacf741ce2010-03-08 10:58:12 +0000352if rc is not None and rc % 256:
Philip Jenvey8b902042009-09-29 19:10:15 +0000353 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
Ross Lagerwall104c3f12011-04-05 15:24:34 +0200399import errno
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000400
Peter Astrand454f7672005-01-01 09:36:35 +0000401# Exception classes used by this module.
Peter Astrand7d1d4362006-07-14 14:04:45 +0000402class CalledProcessError(Exception):
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000403 """This exception is raised when a process run by check_call() or
Gregory P. Smith26576802008-12-05 02:27:01 +0000404 check_output() returns a non-zero exit status.
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000405 The exit status will be stored in the returncode attribute;
Gregory P. Smith26576802008-12-05 02:27:01 +0000406 check_output() will also store the output in the output attribute.
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000407 """
408 def __init__(self, returncode, cmd, output=None):
Peter Astrand7d1d4362006-07-14 14:04:45 +0000409 self.returncode = returncode
410 self.cmd = cmd
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000411 self.output = output
Peter Astrand7d1d4362006-07-14 14:04:45 +0000412 def __str__(self):
413 return "Command '%s' returned non-zero exit status %d" % (self.cmd, self.returncode)
Tim Peters73a9ead2006-07-18 21:55:15 +0000414
Peter Astrand454f7672005-01-01 09:36:35 +0000415
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000416if mswindows:
417 import threading
418 import msvcrt
Brian Curtina2936cf2010-04-24 15:40:11 +0000419 import _subprocess
420 class STARTUPINFO:
421 dwFlags = 0
422 hStdInput = None
423 hStdOutput = None
424 hStdError = None
425 wShowWindow = 0
426 class pywintypes:
427 error = IOError
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000428else:
429 import select
Gregory P. Smithdd7ca242009-07-04 01:49:29 +0000430 _has_poll = hasattr(select, 'poll')
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000431 import fcntl
432 import pickle
433
Amaury Forgeot d'Arcce32eb72009-07-09 22:37:22 +0000434 # When select or poll has indicated that the file is writable,
435 # we can write up to _PIPE_BUF bytes without risk of blocking.
436 # POSIX defines PIPE_BUF as >= 512.
437 _PIPE_BUF = getattr(select, 'PIPE_BUF', 512)
438
439
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000440__all__ = ["Popen", "PIPE", "STDOUT", "call", "check_call",
Gregory P. Smith26576802008-12-05 02:27:01 +0000441 "check_output", "CalledProcessError"]
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000442
Brian Curtina2936cf2010-04-24 15:40:11 +0000443if mswindows:
Brett Cannon42a0ba72010-05-14 00:21:48 +0000444 from _subprocess import CREATE_NEW_CONSOLE, CREATE_NEW_PROCESS_GROUP
Brian Curtina2936cf2010-04-24 15:40:11 +0000445 __all__.extend(["CREATE_NEW_CONSOLE", "CREATE_NEW_PROCESS_GROUP"])
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000446try:
447 MAXFD = os.sysconf("SC_OPEN_MAX")
448except:
449 MAXFD = 256
450
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000451_active = []
452
453def _cleanup():
454 for inst in _active[:]:
Georg Brandl02e7dfd2009-12-28 08:09:32 +0000455 res = inst._internal_poll(_deadstate=sys.maxint)
456 if res is not None and res >= 0:
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000457 try:
458 _active.remove(inst)
459 except ValueError:
460 # This can happen if two threads create a new Popen instance.
461 # It's harmless that it was already removed, so ignore.
462 pass
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000463
464PIPE = -1
465STDOUT = -2
466
467
Gregory P. Smithcce211f2010-03-01 00:05:08 +0000468def _eintr_retry_call(func, *args):
469 while True:
470 try:
471 return func(*args)
472 except OSError, e:
473 if e.errno == errno.EINTR:
474 continue
475 raise
476
477
Peter Astrand5f5e1412004-12-05 20:15:36 +0000478def call(*popenargs, **kwargs):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000479 """Run command with arguments. Wait for command to complete, then
480 return the returncode attribute.
481
482 The arguments are the same as for the Popen constructor. Example:
483
484 retcode = call(["ls", "-l"])
485 """
Peter Astrand5f5e1412004-12-05 20:15:36 +0000486 return Popen(*popenargs, **kwargs).wait()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000487
488
Peter Astrand454f7672005-01-01 09:36:35 +0000489def check_call(*popenargs, **kwargs):
490 """Run command with arguments. Wait for command to complete. If
491 the exit code was zero then return, otherwise raise
492 CalledProcessError. The CalledProcessError object will have the
Peter Astrand7d1d4362006-07-14 14:04:45 +0000493 return code in the returncode attribute.
Peter Astrand454f7672005-01-01 09:36:35 +0000494
495 The arguments are the same as for the Popen constructor. Example:
496
497 check_call(["ls", "-l"])
498 """
499 retcode = call(*popenargs, **kwargs)
Peter Astrand454f7672005-01-01 09:36:35 +0000500 if retcode:
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000501 cmd = kwargs.get("args")
502 if cmd is None:
503 cmd = popenargs[0]
Peter Astrand7d1d4362006-07-14 14:04:45 +0000504 raise CalledProcessError(retcode, cmd)
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000505 return 0
506
507
Gregory P. Smith26576802008-12-05 02:27:01 +0000508def check_output(*popenargs, **kwargs):
Georg Brandl6ab5d082009-12-20 14:33:20 +0000509 r"""Run command with arguments and return its output as a byte string.
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000510
511 If the exit code was non-zero it raises a CalledProcessError. The
512 CalledProcessError object will have the return code in the returncode
513 attribute and output in the output attribute.
514
515 The arguments are the same as for the Popen constructor. Example:
516
Gregory P. Smith26576802008-12-05 02:27:01 +0000517 >>> check_output(["ls", "-l", "/dev/null"])
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000518 'crw-rw-rw- 1 root root 1, 3 Oct 18 2007 /dev/null\n'
519
520 The stdout argument is not allowed as it is used internally.
Georg Brandl02e7dfd2009-12-28 08:09:32 +0000521 To capture standard error in the result, use stderr=STDOUT.
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000522
Gregory P. Smith26576802008-12-05 02:27:01 +0000523 >>> check_output(["/bin/sh", "-c",
Georg Brandl6ab5d082009-12-20 14:33:20 +0000524 ... "ls -l non_existent_file ; exit 0"],
Georg Brandl02e7dfd2009-12-28 08:09:32 +0000525 ... stderr=STDOUT)
Mark Dickinson3e4caeb2009-02-21 20:27:01 +0000526 'ls: non_existent_file: No such file or directory\n'
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000527 """
528 if 'stdout' in kwargs:
529 raise ValueError('stdout argument not allowed, it will be overridden.')
Amaury Forgeot d'Arc5fe420e2009-06-18 22:32:50 +0000530 process = Popen(stdout=PIPE, *popenargs, **kwargs)
Gregory P. Smith97f49f42008-12-04 20:21:09 +0000531 output, unused_err = process.communicate()
532 retcode = process.poll()
533 if retcode:
534 cmd = kwargs.get("args")
535 if cmd is None:
536 cmd = popenargs[0]
537 raise CalledProcessError(retcode, cmd, output=output)
538 return output
Peter Astrand454f7672005-01-01 09:36:35 +0000539
540
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000541def list2cmdline(seq):
542 """
543 Translate a sequence of arguments into a command line
544 string, using the same rules as the MS C runtime:
545
546 1) Arguments are delimited by white space, which is either a
547 space or a tab.
548
549 2) A string surrounded by double quotation marks is
550 interpreted as a single argument, regardless of white space
Jean-Paul Calderoneb33f0c12010-06-18 20:00:17 +0000551 contained within. A quoted string can be embedded in an
552 argument.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000553
554 3) A double quotation mark preceded by a backslash is
555 interpreted as a literal double quotation mark.
556
557 4) Backslashes are interpreted literally, unless they
558 immediately precede a double quotation mark.
559
560 5) If backslashes immediately precede a double quotation mark,
561 every pair of backslashes is interpreted as a literal
562 backslash. If the number of backslashes is odd, the last
563 backslash escapes the next double quotation mark as
564 described in rule 3.
565 """
566
567 # See
Eric Smithd19915e2009-11-09 15:16:23 +0000568 # http://msdn.microsoft.com/en-us/library/17w5ykft.aspx
569 # or search http://msdn.microsoft.com for
570 # "Parsing C++ Command-Line Arguments"
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000571 result = []
572 needquote = False
573 for arg in seq:
574 bs_buf = []
575
576 # Add a space to separate this argument from the others
577 if result:
578 result.append(' ')
579
Jean-Paul Calderoneb33f0c12010-06-18 20:00:17 +0000580 needquote = (" " in arg) or ("\t" in arg) or not arg
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000581 if needquote:
582 result.append('"')
583
584 for c in arg:
585 if c == '\\':
Tim Peterse718f612004-10-12 21:51:32 +0000586 # Don't know if we need to double yet.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000587 bs_buf.append(c)
588 elif c == '"':
Gregory P. Smithe047e6d2008-01-19 20:49:02 +0000589 # Double backslashes.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000590 result.append('\\' * len(bs_buf)*2)
591 bs_buf = []
592 result.append('\\"')
593 else:
594 # Normal char
595 if bs_buf:
596 result.extend(bs_buf)
597 bs_buf = []
598 result.append(c)
599
Gregory P. Smithe047e6d2008-01-19 20:49:02 +0000600 # Add remaining backslashes, if any.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000601 if bs_buf:
602 result.extend(bs_buf)
603
604 if needquote:
Peter Astrand7e78ade2005-03-03 21:10:23 +0000605 result.extend(bs_buf)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000606 result.append('"')
607
608 return ''.join(result)
609
610
611class Popen(object):
612 def __init__(self, args, bufsize=0, executable=None,
613 stdin=None, stdout=None, stderr=None,
614 preexec_fn=None, close_fds=False, shell=False,
615 cwd=None, env=None, universal_newlines=False,
616 startupinfo=None, creationflags=0):
617 """Create new Popen instance."""
618 _cleanup()
619
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000620 self._child_created = False
Peter Astrand738131d2004-11-30 21:04:45 +0000621 if not isinstance(bufsize, (int, long)):
622 raise TypeError("bufsize must be an integer")
623
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000624 if mswindows:
Tim Peterse8374a52004-10-13 03:15:00 +0000625 if preexec_fn is not None:
626 raise ValueError("preexec_fn is not supported on Windows "
627 "platforms")
Peter Astrand81a191b2007-05-26 22:18:20 +0000628 if close_fds and (stdin is not None or stdout is not None or
629 stderr is not None):
Tim Peterse8374a52004-10-13 03:15:00 +0000630 raise ValueError("close_fds is not supported on Windows "
Peter Astrand81a191b2007-05-26 22:18:20 +0000631 "platforms if you redirect stdin/stdout/stderr")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000632 else:
633 # POSIX
Tim Peterse8374a52004-10-13 03:15:00 +0000634 if startupinfo is not None:
635 raise ValueError("startupinfo is only supported on Windows "
636 "platforms")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000637 if creationflags != 0:
Tim Peterse8374a52004-10-13 03:15:00 +0000638 raise ValueError("creationflags is only supported on Windows "
639 "platforms")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000640
Tim Peterse718f612004-10-12 21:51:32 +0000641 self.stdin = None
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000642 self.stdout = None
643 self.stderr = None
644 self.pid = None
645 self.returncode = None
646 self.universal_newlines = universal_newlines
647
648 # Input and output objects. The general principle is like
649 # this:
650 #
651 # Parent Child
652 # ------ -----
653 # p2cwrite ---stdin---> p2cread
654 # c2pread <--stdout--- c2pwrite
655 # errread <--stderr--- errwrite
656 #
657 # On POSIX, the child objects are file descriptors. On
658 # Windows, these are Windows file handles. The parent objects
659 # are file descriptors on both platforms. The parent objects
660 # are None when not using PIPEs. The child objects are None
661 # when not redirecting.
Tim Peterse718f612004-10-12 21:51:32 +0000662
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000663 (p2cread, p2cwrite,
664 c2pread, c2pwrite,
665 errread, errwrite) = self._get_handles(stdin, stdout, stderr)
666
667 self._execute_child(args, executable, preexec_fn, close_fds,
668 cwd, env, universal_newlines,
669 startupinfo, creationflags, shell,
670 p2cread, p2cwrite,
671 c2pread, c2pwrite,
672 errread, errwrite)
673
Peter Astrand5f9c6ae2007-02-06 15:37:50 +0000674 if mswindows:
Hirokazu Yamamotoeacbbdf2009-03-03 22:18:14 +0000675 if p2cwrite is not None:
676 p2cwrite = msvcrt.open_osfhandle(p2cwrite.Detach(), 0)
677 if c2pread is not None:
678 c2pread = msvcrt.open_osfhandle(c2pread.Detach(), 0)
679 if errread is not None:
680 errread = msvcrt.open_osfhandle(errread.Detach(), 0)
Peter Astrand5f9c6ae2007-02-06 15:37:50 +0000681
Peter Astrandf5400032007-02-02 19:06:36 +0000682 if p2cwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000683 self.stdin = os.fdopen(p2cwrite, 'wb', bufsize)
Peter Astrandf5400032007-02-02 19:06:36 +0000684 if c2pread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000685 if universal_newlines:
686 self.stdout = os.fdopen(c2pread, 'rU', bufsize)
687 else:
688 self.stdout = os.fdopen(c2pread, 'rb', bufsize)
Peter Astrandf5400032007-02-02 19:06:36 +0000689 if errread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000690 if universal_newlines:
691 self.stderr = os.fdopen(errread, 'rU', bufsize)
692 else:
693 self.stderr = os.fdopen(errread, 'rb', bufsize)
Tim Peterse718f612004-10-12 21:51:32 +0000694
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000695
696 def _translate_newlines(self, data):
697 data = data.replace("\r\n", "\n")
698 data = data.replace("\r", "\n")
699 return data
700
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000701
Brett Cannon42a0ba72010-05-14 00:21:48 +0000702 def __del__(self, _maxint=sys.maxint, _active=_active):
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000703 if not self._child_created:
704 # We didn't get to successfully create a child process.
705 return
706 # In case the child hasn't been waited on, check if it's done.
Brett Cannon42a0ba72010-05-14 00:21:48 +0000707 self._internal_poll(_deadstate=_maxint)
Georg Brandl13cf38c2006-07-20 16:28:39 +0000708 if self.returncode is None and _active is not None:
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000709 # Child is still running, keep us alive until we can wait on it.
710 _active.append(self)
711
712
Peter Astrand23109f02005-03-03 20:28:59 +0000713 def communicate(self, input=None):
714 """Interact with process: Send data to stdin. Read data from
715 stdout and stderr, until end-of-file is reached. Wait for
716 process to terminate. The optional input argument should be a
717 string to be sent to the child process, or None, if no data
718 should be sent to the child.
Tim Peterseba28be2005-03-28 01:08:02 +0000719
Peter Astrand23109f02005-03-03 20:28:59 +0000720 communicate() returns a tuple (stdout, stderr)."""
721
722 # Optimization: If we are only using one pipe, or no pipe at
723 # all, using select() or threads is unnecessary.
724 if [self.stdin, self.stdout, self.stderr].count(None) >= 2:
Tim Peterseba28be2005-03-28 01:08:02 +0000725 stdout = None
726 stderr = None
Peter Astrand23109f02005-03-03 20:28:59 +0000727 if self.stdin:
728 if input:
Ross Lagerwall104c3f12011-04-05 15:24:34 +0200729 try:
730 self.stdin.write(input)
731 except IOError as e:
732 if e.errno != errno.EPIPE and e.errno != errno.EINVAL:
733 raise
Peter Astrand23109f02005-03-03 20:28:59 +0000734 self.stdin.close()
735 elif self.stdout:
736 stdout = self.stdout.read()
Gregory P. Smith4036fd42008-05-26 20:22:14 +0000737 self.stdout.close()
Peter Astrand23109f02005-03-03 20:28:59 +0000738 elif self.stderr:
739 stderr = self.stderr.read()
Gregory P. Smith4036fd42008-05-26 20:22:14 +0000740 self.stderr.close()
Peter Astrand23109f02005-03-03 20:28:59 +0000741 self.wait()
742 return (stdout, stderr)
Tim Peterseba28be2005-03-28 01:08:02 +0000743
Peter Astrand23109f02005-03-03 20:28:59 +0000744 return self._communicate(input)
745
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000746
Gregory P. Smitha36f8fe2008-08-04 00:13:29 +0000747 def poll(self):
748 return self._internal_poll()
749
750
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000751 if mswindows:
752 #
753 # Windows methods
754 #
755 def _get_handles(self, stdin, stdout, stderr):
Amaury Forgeot d'Arc8318afa2009-07-10 16:47:42 +0000756 """Construct and return tuple with IO objects:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000757 p2cread, p2cwrite, c2pread, c2pwrite, errread, errwrite
758 """
Peter Astrandd38ddf42005-02-10 08:32:50 +0000759 if stdin is None and stdout is None and stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000760 return (None, None, None, None, None, None)
Tim Peterse718f612004-10-12 21:51:32 +0000761
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000762 p2cread, p2cwrite = None, None
763 c2pread, c2pwrite = None, None
764 errread, errwrite = None, None
765
Peter Astrandd38ddf42005-02-10 08:32:50 +0000766 if stdin is None:
Brian Curtina2936cf2010-04-24 15:40:11 +0000767 p2cread = _subprocess.GetStdHandle(_subprocess.STD_INPUT_HANDLE)
Hirokazu Yamamotoeacbbdf2009-03-03 22:18:14 +0000768 if p2cread is None:
Brian Curtina2936cf2010-04-24 15:40:11 +0000769 p2cread, _ = _subprocess.CreatePipe(None, 0)
Hirokazu Yamamotoeacbbdf2009-03-03 22:18:14 +0000770 elif stdin == PIPE:
Brian Curtina2936cf2010-04-24 15:40:11 +0000771 p2cread, p2cwrite = _subprocess.CreatePipe(None, 0)
Peter Astrandd38ddf42005-02-10 08:32:50 +0000772 elif isinstance(stdin, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000773 p2cread = msvcrt.get_osfhandle(stdin)
774 else:
775 # Assuming file-like object
776 p2cread = msvcrt.get_osfhandle(stdin.fileno())
777 p2cread = self._make_inheritable(p2cread)
778
Peter Astrandd38ddf42005-02-10 08:32:50 +0000779 if stdout is None:
Brian Curtina2936cf2010-04-24 15:40:11 +0000780 c2pwrite = _subprocess.GetStdHandle(_subprocess.STD_OUTPUT_HANDLE)
Hirokazu Yamamotoeacbbdf2009-03-03 22:18:14 +0000781 if c2pwrite is None:
Brian Curtina2936cf2010-04-24 15:40:11 +0000782 _, c2pwrite = _subprocess.CreatePipe(None, 0)
Hirokazu Yamamotoeacbbdf2009-03-03 22:18:14 +0000783 elif stdout == PIPE:
Brian Curtina2936cf2010-04-24 15:40:11 +0000784 c2pread, c2pwrite = _subprocess.CreatePipe(None, 0)
Peter Astrandd38ddf42005-02-10 08:32:50 +0000785 elif isinstance(stdout, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000786 c2pwrite = msvcrt.get_osfhandle(stdout)
787 else:
788 # Assuming file-like object
789 c2pwrite = msvcrt.get_osfhandle(stdout.fileno())
790 c2pwrite = self._make_inheritable(c2pwrite)
791
Peter Astrandd38ddf42005-02-10 08:32:50 +0000792 if stderr is None:
Brian Curtina2936cf2010-04-24 15:40:11 +0000793 errwrite = _subprocess.GetStdHandle(_subprocess.STD_ERROR_HANDLE)
Hirokazu Yamamotoeacbbdf2009-03-03 22:18:14 +0000794 if errwrite is None:
Brian Curtina2936cf2010-04-24 15:40:11 +0000795 _, errwrite = _subprocess.CreatePipe(None, 0)
Hirokazu Yamamotoeacbbdf2009-03-03 22:18:14 +0000796 elif stderr == PIPE:
Brian Curtina2936cf2010-04-24 15:40:11 +0000797 errread, errwrite = _subprocess.CreatePipe(None, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000798 elif stderr == STDOUT:
799 errwrite = c2pwrite
Peter Astrandd38ddf42005-02-10 08:32:50 +0000800 elif isinstance(stderr, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000801 errwrite = msvcrt.get_osfhandle(stderr)
802 else:
803 # Assuming file-like object
804 errwrite = msvcrt.get_osfhandle(stderr.fileno())
805 errwrite = self._make_inheritable(errwrite)
806
807 return (p2cread, p2cwrite,
808 c2pread, c2pwrite,
809 errread, errwrite)
810
811
812 def _make_inheritable(self, handle):
813 """Return a duplicate of handle, which is inheritable"""
Brian Curtina2936cf2010-04-24 15:40:11 +0000814 return _subprocess.DuplicateHandle(_subprocess.GetCurrentProcess(),
815 handle, _subprocess.GetCurrentProcess(), 0, 1,
816 _subprocess.DUPLICATE_SAME_ACCESS)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000817
818
819 def _find_w9xpopen(self):
820 """Find and return absolut path to w9xpopen.exe"""
Brian Curtina2936cf2010-04-24 15:40:11 +0000821 w9xpopen = os.path.join(
822 os.path.dirname(_subprocess.GetModuleFileName(0)),
Tim Peterse8374a52004-10-13 03:15:00 +0000823 "w9xpopen.exe")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000824 if not os.path.exists(w9xpopen):
825 # Eeek - file-not-found - possibly an embedding
826 # situation - see if we can locate it in sys.exec_prefix
Tim Peterse8374a52004-10-13 03:15:00 +0000827 w9xpopen = os.path.join(os.path.dirname(sys.exec_prefix),
828 "w9xpopen.exe")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000829 if not os.path.exists(w9xpopen):
Tim Peterse8374a52004-10-13 03:15:00 +0000830 raise RuntimeError("Cannot locate w9xpopen.exe, which is "
831 "needed for Popen to work with your "
832 "shell or platform.")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000833 return w9xpopen
834
Tim Peterse718f612004-10-12 21:51:32 +0000835
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000836 def _execute_child(self, args, executable, preexec_fn, close_fds,
837 cwd, env, universal_newlines,
838 startupinfo, creationflags, shell,
839 p2cread, p2cwrite,
840 c2pread, c2pwrite,
841 errread, errwrite):
842 """Execute program (MS Windows version)"""
843
Peter Astrandc26516b2005-02-21 08:13:02 +0000844 if not isinstance(args, types.StringTypes):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000845 args = list2cmdline(args)
846
Peter Astrandc1d65362004-11-07 14:30:34 +0000847 # Process startup details
Peter Astrandd38ddf42005-02-10 08:32:50 +0000848 if startupinfo is None:
Georg Brandlad624892006-06-04 22:15:37 +0000849 startupinfo = STARTUPINFO()
850 if None not in (p2cread, c2pwrite, errwrite):
Brian Curtina2936cf2010-04-24 15:40:11 +0000851 startupinfo.dwFlags |= _subprocess.STARTF_USESTDHANDLES
Peter Astrandc1d65362004-11-07 14:30:34 +0000852 startupinfo.hStdInput = p2cread
853 startupinfo.hStdOutput = c2pwrite
854 startupinfo.hStdError = errwrite
855
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000856 if shell:
Brian Curtina2936cf2010-04-24 15:40:11 +0000857 startupinfo.dwFlags |= _subprocess.STARTF_USESHOWWINDOW
858 startupinfo.wShowWindow = _subprocess.SW_HIDE
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000859 comspec = os.environ.get("COMSPEC", "cmd.exe")
Tim Golden8e4756c2010-08-12 11:00:35 +0000860 args = '{} /c "{}"'.format (comspec, args)
861 if (_subprocess.GetVersion() >= 0x80000000 or
Tim Peterse8374a52004-10-13 03:15:00 +0000862 os.path.basename(comspec).lower() == "command.com"):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000863 # Win9x, or using command.com on NT. We need to
864 # use the w9xpopen intermediate program. For more
865 # information, see KB Q150956
866 # (http://web.archive.org/web/20011105084002/http://support.microsoft.com/support/kb/articles/Q150/9/56.asp)
867 w9xpopen = self._find_w9xpopen()
868 args = '"%s" %s' % (w9xpopen, args)
869 # Not passing CREATE_NEW_CONSOLE has been known to
870 # cause random failures on win9x. Specifically a
871 # dialog: "Your program accessed mem currently in
872 # use at xxx" and a hopeful warning about the
873 # stability of your system. Cost is Ctrl+C wont
874 # kill children.
Brian Curtina2936cf2010-04-24 15:40:11 +0000875 creationflags |= _subprocess.CREATE_NEW_CONSOLE
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000876
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000877 # Start the process
878 try:
Brian Curtina2936cf2010-04-24 15:40:11 +0000879 hp, ht, pid, tid = _subprocess.CreateProcess(executable, args,
Tim Peterse8374a52004-10-13 03:15:00 +0000880 # no special security
881 None, None,
Peter Astrand81a191b2007-05-26 22:18:20 +0000882 int(not close_fds),
Tim Peterse8374a52004-10-13 03:15:00 +0000883 creationflags,
884 env,
885 cwd,
886 startupinfo)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000887 except pywintypes.error, e:
888 # Translate pywintypes.error to WindowsError, which is
889 # a subclass of OSError. FIXME: We should really
Ezio Melottic2077b02011-03-16 12:34:31 +0200890 # translate errno using _sys_errlist (or similar), but
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000891 # how can this be done from Python?
892 raise WindowsError(*e.args)
Tim Golden431774f2010-08-08 11:17:56 +0000893 finally:
894 # Child is launched. Close the parent's copy of those pipe
895 # handles that only the child should have open. You need
896 # to make sure that no handles to the write end of the
897 # output pipe are maintained in this process or else the
898 # pipe will not close when the child process exits and the
899 # ReadFile will hang.
900 if p2cread is not None:
901 p2cread.Close()
902 if c2pwrite is not None:
903 c2pwrite.Close()
904 if errwrite is not None:
905 errwrite.Close()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000906
907 # Retain the process handle, but close the thread handle
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000908 self._child_created = True
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000909 self._handle = hp
910 self.pid = pid
911 ht.Close()
912
Brett Cannon42a0ba72010-05-14 00:21:48 +0000913 def _internal_poll(self, _deadstate=None,
Victor Stinner2b271f72010-05-14 21:52:26 +0000914 _WaitForSingleObject=_subprocess.WaitForSingleObject,
915 _WAIT_OBJECT_0=_subprocess.WAIT_OBJECT_0,
916 _GetExitCodeProcess=_subprocess.GetExitCodeProcess):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000917 """Check if child process has terminated. Returns returncode
Brett Cannon42a0ba72010-05-14 00:21:48 +0000918 attribute.
919
920 This method is called by __del__, so it can only refer to objects
921 in its local scope.
922
923 """
Peter Astrandd38ddf42005-02-10 08:32:50 +0000924 if self.returncode is None:
Brett Cannon42a0ba72010-05-14 00:21:48 +0000925 if _WaitForSingleObject(self._handle, 0) == _WAIT_OBJECT_0:
926 self.returncode = _GetExitCodeProcess(self._handle)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000927 return self.returncode
928
929
930 def wait(self):
931 """Wait for child process to terminate. Returns returncode
932 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +0000933 if self.returncode is None:
Brian Curtina2936cf2010-04-24 15:40:11 +0000934 _subprocess.WaitForSingleObject(self._handle,
935 _subprocess.INFINITE)
936 self.returncode = _subprocess.GetExitCodeProcess(self._handle)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000937 return self.returncode
938
939
940 def _readerthread(self, fh, buffer):
941 buffer.append(fh.read())
942
943
Peter Astrand23109f02005-03-03 20:28:59 +0000944 def _communicate(self, input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000945 stdout = None # Return
946 stderr = None # Return
947
948 if self.stdout:
949 stdout = []
Tim Peterse8374a52004-10-13 03:15:00 +0000950 stdout_thread = threading.Thread(target=self._readerthread,
951 args=(self.stdout, stdout))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000952 stdout_thread.setDaemon(True)
953 stdout_thread.start()
954 if self.stderr:
955 stderr = []
Tim Peterse8374a52004-10-13 03:15:00 +0000956 stderr_thread = threading.Thread(target=self._readerthread,
957 args=(self.stderr, stderr))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000958 stderr_thread.setDaemon(True)
959 stderr_thread.start()
960
961 if self.stdin:
Peter Astrandd38ddf42005-02-10 08:32:50 +0000962 if input is not None:
Ross Lagerwall104c3f12011-04-05 15:24:34 +0200963 try:
964 self.stdin.write(input)
965 except IOError as e:
966 if e.errno != errno.EPIPE:
967 raise
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000968 self.stdin.close()
969
970 if self.stdout:
971 stdout_thread.join()
972 if self.stderr:
973 stderr_thread.join()
974
975 # All data exchanged. Translate lists into strings.
Peter Astrandd38ddf42005-02-10 08:32:50 +0000976 if stdout is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000977 stdout = stdout[0]
Peter Astrandd38ddf42005-02-10 08:32:50 +0000978 if stderr is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000979 stderr = stderr[0]
980
981 # Translate newlines, if requested. We cannot let the file
982 # object do the translation: It is based on stdio, which is
983 # impossible to combine with select (unless forcing no
984 # buffering).
Neal Norwitza6d01ce2006-05-02 06:23:22 +0000985 if self.universal_newlines and hasattr(file, 'newlines'):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000986 if stdout:
987 stdout = self._translate_newlines(stdout)
988 if stderr:
989 stderr = self._translate_newlines(stderr)
990
991 self.wait()
992 return (stdout, stderr)
993
Christian Heimese74c8f22008-04-19 02:23:57 +0000994 def send_signal(self, sig):
995 """Send a signal to the process
996 """
997 if sig == signal.SIGTERM:
998 self.terminate()
Brian Curtine5aa8862010-04-02 23:26:06 +0000999 elif sig == signal.CTRL_C_EVENT:
1000 os.kill(self.pid, signal.CTRL_C_EVENT)
1001 elif sig == signal.CTRL_BREAK_EVENT:
1002 os.kill(self.pid, signal.CTRL_BREAK_EVENT)
Christian Heimese74c8f22008-04-19 02:23:57 +00001003 else:
Brian Curtine80513c2010-09-07 13:27:20 +00001004 raise ValueError("Unsupported signal: {}".format(sig))
Christian Heimese74c8f22008-04-19 02:23:57 +00001005
1006 def terminate(self):
1007 """Terminates the process
1008 """
Brian Curtina2936cf2010-04-24 15:40:11 +00001009 _subprocess.TerminateProcess(self._handle, 1)
Christian Heimese74c8f22008-04-19 02:23:57 +00001010
1011 kill = terminate
1012
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001013 else:
1014 #
1015 # POSIX methods
1016 #
1017 def _get_handles(self, stdin, stdout, stderr):
Amaury Forgeot d'Arc8318afa2009-07-10 16:47:42 +00001018 """Construct and return tuple with IO objects:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001019 p2cread, p2cwrite, c2pread, c2pwrite, errread, errwrite
1020 """
1021 p2cread, p2cwrite = None, None
1022 c2pread, c2pwrite = None, None
1023 errread, errwrite = None, None
1024
Peter Astrandd38ddf42005-02-10 08:32:50 +00001025 if stdin is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001026 pass
1027 elif stdin == PIPE:
1028 p2cread, p2cwrite = os.pipe()
Peter Astrandd38ddf42005-02-10 08:32:50 +00001029 elif isinstance(stdin, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001030 p2cread = stdin
1031 else:
1032 # Assuming file-like object
1033 p2cread = stdin.fileno()
1034
Peter Astrandd38ddf42005-02-10 08:32:50 +00001035 if stdout is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001036 pass
1037 elif stdout == PIPE:
1038 c2pread, c2pwrite = os.pipe()
Peter Astrandd38ddf42005-02-10 08:32:50 +00001039 elif isinstance(stdout, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001040 c2pwrite = stdout
1041 else:
1042 # Assuming file-like object
Tim Peterse718f612004-10-12 21:51:32 +00001043 c2pwrite = stdout.fileno()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001044
Peter Astrandd38ddf42005-02-10 08:32:50 +00001045 if stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001046 pass
1047 elif stderr == PIPE:
1048 errread, errwrite = os.pipe()
1049 elif stderr == STDOUT:
1050 errwrite = c2pwrite
Peter Astrandd38ddf42005-02-10 08:32:50 +00001051 elif isinstance(stderr, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001052 errwrite = stderr
1053 else:
1054 # Assuming file-like object
1055 errwrite = stderr.fileno()
1056
1057 return (p2cread, p2cwrite,
1058 c2pread, c2pwrite,
1059 errread, errwrite)
1060
1061
Antoine Pitrou91ce0d92011-01-03 18:45:09 +00001062 def _set_cloexec_flag(self, fd, cloexec=True):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001063 try:
1064 cloexec_flag = fcntl.FD_CLOEXEC
1065 except AttributeError:
1066 cloexec_flag = 1
1067
1068 old = fcntl.fcntl(fd, fcntl.F_GETFD)
Antoine Pitrou91ce0d92011-01-03 18:45:09 +00001069 if cloexec:
1070 fcntl.fcntl(fd, fcntl.F_SETFD, old | cloexec_flag)
1071 else:
1072 fcntl.fcntl(fd, fcntl.F_SETFD, old & ~cloexec_flag)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001073
1074
1075 def _close_fds(self, but):
Amaury Forgeot d'Arc5fe420e2009-06-18 22:32:50 +00001076 if hasattr(os, 'closerange'):
1077 os.closerange(3, but)
1078 os.closerange(but + 1, MAXFD)
1079 else:
1080 for i in xrange(3, MAXFD):
1081 if i == but:
1082 continue
1083 try:
1084 os.close(i)
1085 except:
1086 pass
Tim Peterse718f612004-10-12 21:51:32 +00001087
1088
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001089 def _execute_child(self, args, executable, preexec_fn, close_fds,
1090 cwd, env, universal_newlines,
1091 startupinfo, creationflags, shell,
1092 p2cread, p2cwrite,
1093 c2pread, c2pwrite,
1094 errread, errwrite):
1095 """Execute program (POSIX version)"""
1096
Peter Astrandc26516b2005-02-21 08:13:02 +00001097 if isinstance(args, types.StringTypes):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001098 args = [args]
Georg Brandl6c0e1e82006-10-29 09:05:04 +00001099 else:
1100 args = list(args)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001101
1102 if shell:
1103 args = ["/bin/sh", "-c"] + args
Stefan Krahe9a6a7d2010-07-19 14:41:08 +00001104 if executable:
1105 args[0] = executable
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001106
Peter Astrandd38ddf42005-02-10 08:32:50 +00001107 if executable is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001108 executable = args[0]
1109
1110 # For transferring possible exec failure from child to parent
1111 # The first char specifies the exception type: 0 means
1112 # OSError, 1 means some other error.
1113 errpipe_read, errpipe_write = os.pipe()
Gregory P. Smith87d49792008-01-19 20:57:59 +00001114 try:
Gregory P. Smith92ffc632008-01-19 22:23:56 +00001115 try:
Facundo Batista8c826b72009-06-19 18:02:28 +00001116 self._set_cloexec_flag(errpipe_write)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001117
Facundo Batista8c826b72009-06-19 18:02:28 +00001118 gc_was_enabled = gc.isenabled()
1119 # Disable gc to avoid bug where gc -> file_dealloc ->
1120 # write to stderr -> hang. http://bugs.python.org/issue1336
1121 gc.disable()
1122 try:
1123 self.pid = os.fork()
Georg Brandl3e8b8692009-07-16 21:47:51 +00001124 except:
Facundo Batista8c826b72009-06-19 18:02:28 +00001125 if gc_was_enabled:
1126 gc.enable()
Georg Brandl3e8b8692009-07-16 21:47:51 +00001127 raise
Facundo Batista8c826b72009-06-19 18:02:28 +00001128 self._child_created = True
1129 if self.pid == 0:
1130 # Child
1131 try:
1132 # Close parent's pipe ends
1133 if p2cwrite is not None:
1134 os.close(p2cwrite)
1135 if c2pread is not None:
1136 os.close(c2pread)
1137 if errread is not None:
1138 os.close(errread)
1139 os.close(errpipe_read)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001140
Facundo Batista8c826b72009-06-19 18:02:28 +00001141 # Dup fds for child
Antoine Pitrou91ce0d92011-01-03 18:45:09 +00001142 def _dup2(a, b):
1143 # dup2() removes the CLOEXEC flag but
1144 # we must do it ourselves if dup2()
1145 # would be a no-op (issue #10806).
1146 if a == b:
1147 self._set_cloexec_flag(a, False)
1148 elif a is not None:
1149 os.dup2(a, b)
1150 _dup2(p2cread, 0)
1151 _dup2(c2pwrite, 1)
1152 _dup2(errwrite, 2)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001153
Antoine Pitrou91ce0d92011-01-03 18:45:09 +00001154 # Close pipe fds. Make sure we don't close the
1155 # same fd more than once, or standard fds.
1156 closed = { None }
1157 for fd in [p2cread, c2pwrite, errwrite]:
1158 if fd not in closed and fd > 2:
1159 os.close(fd)
1160 closed.add(fd)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001161
Facundo Batista8c826b72009-06-19 18:02:28 +00001162 # Close all other fds, if asked for
1163 if close_fds:
1164 self._close_fds(but=errpipe_write)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001165
Facundo Batista8c826b72009-06-19 18:02:28 +00001166 if cwd is not None:
1167 os.chdir(cwd)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001168
Facundo Batista8c826b72009-06-19 18:02:28 +00001169 if preexec_fn:
1170 preexec_fn()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001171
Facundo Batista8c826b72009-06-19 18:02:28 +00001172 if env is None:
1173 os.execvp(executable, args)
1174 else:
1175 os.execvpe(executable, args, env)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001176
Facundo Batista8c826b72009-06-19 18:02:28 +00001177 except:
1178 exc_type, exc_value, tb = sys.exc_info()
1179 # Save the traceback and attach it to the exception object
1180 exc_lines = traceback.format_exception(exc_type,
1181 exc_value,
1182 tb)
1183 exc_value.child_traceback = ''.join(exc_lines)
1184 os.write(errpipe_write, pickle.dumps(exc_value))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001185
Facundo Batista8c826b72009-06-19 18:02:28 +00001186 # This exitcode won't be reported to applications, so it
1187 # really doesn't matter what we return.
1188 os._exit(255)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001189
Facundo Batista8c826b72009-06-19 18:02:28 +00001190 # Parent
1191 if gc_was_enabled:
1192 gc.enable()
1193 finally:
1194 # be sure the FD is closed no matter what
1195 os.close(errpipe_write)
1196
1197 if p2cread is not None and p2cwrite is not None:
1198 os.close(p2cread)
1199 if c2pwrite is not None and c2pread is not None:
1200 os.close(c2pwrite)
1201 if errwrite is not None and errread is not None:
1202 os.close(errwrite)
1203
1204 # Wait for exec to fail or succeed; possibly raising exception
Gregory P. Smithcce211f2010-03-01 00:05:08 +00001205 # Exception limited to 1M
1206 data = _eintr_retry_call(os.read, errpipe_read, 1048576)
Facundo Batista8c826b72009-06-19 18:02:28 +00001207 finally:
1208 # be sure the FD is closed no matter what
1209 os.close(errpipe_read)
1210
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001211 if data != "":
Gregory P. Smith312efbc2010-12-14 15:02:53 +00001212 try:
1213 _eintr_retry_call(os.waitpid, self.pid, 0)
1214 except OSError as e:
1215 if e.errno != errno.ECHILD:
1216 raise
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001217 child_exception = pickle.loads(data)
Georg Brandlf3715d22009-02-14 17:01:36 +00001218 for fd in (p2cwrite, c2pread, errread):
1219 if fd is not None:
1220 os.close(fd)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001221 raise child_exception
1222
1223
Brett Cannon42a0ba72010-05-14 00:21:48 +00001224 def _handle_exitstatus(self, sts, _WIFSIGNALED=os.WIFSIGNALED,
1225 _WTERMSIG=os.WTERMSIG, _WIFEXITED=os.WIFEXITED,
1226 _WEXITSTATUS=os.WEXITSTATUS):
1227 # This method is called (indirectly) by __del__, so it cannot
1228 # refer to anything outside of its local scope."""
1229 if _WIFSIGNALED(sts):
1230 self.returncode = -_WTERMSIG(sts)
1231 elif _WIFEXITED(sts):
1232 self.returncode = _WEXITSTATUS(sts)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001233 else:
1234 # Should never happen
1235 raise RuntimeError("Unknown child exit status!")
1236
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001237
Brett Cannon42a0ba72010-05-14 00:21:48 +00001238 def _internal_poll(self, _deadstate=None, _waitpid=os.waitpid,
1239 _WNOHANG=os.WNOHANG, _os_error=os.error):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001240 """Check if child process has terminated. Returns returncode
Brett Cannon42a0ba72010-05-14 00:21:48 +00001241 attribute.
1242
1243 This method is called by __del__, so it cannot reference anything
1244 outside of the local scope (nor can any methods it calls).
1245
1246 """
Peter Astrandd38ddf42005-02-10 08:32:50 +00001247 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001248 try:
Brett Cannon42a0ba72010-05-14 00:21:48 +00001249 pid, sts = _waitpid(self.pid, _WNOHANG)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001250 if pid == self.pid:
1251 self._handle_exitstatus(sts)
Brett Cannon42a0ba72010-05-14 00:21:48 +00001252 except _os_error:
Martin v. Löwis17de8ff2006-04-10 15:55:37 +00001253 if _deadstate is not None:
1254 self.returncode = _deadstate
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001255 return self.returncode
1256
1257
1258 def wait(self):
1259 """Wait for child process to terminate. Returns returncode
1260 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +00001261 if self.returncode is None:
Gregory P. Smith312efbc2010-12-14 15:02:53 +00001262 try:
1263 pid, sts = _eintr_retry_call(os.waitpid, self.pid, 0)
1264 except OSError as e:
1265 if e.errno != errno.ECHILD:
1266 raise
1267 # This happens if SIGCLD is set to be ignored or waiting
1268 # for child processes has otherwise been disabled for our
1269 # process. This child is dead, we can't get the status.
1270 sts = 0
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001271 self._handle_exitstatus(sts)
1272 return self.returncode
1273
1274
Peter Astrand23109f02005-03-03 20:28:59 +00001275 def _communicate(self, input):
Gregory P. Smithdd7ca242009-07-04 01:49:29 +00001276 if self.stdin:
1277 # Flush stdio buffer. This might block, if the user has
1278 # been writing to .stdin in an uncontrolled fashion.
1279 self.stdin.flush()
1280 if not input:
1281 self.stdin.close()
1282
1283 if _has_poll:
1284 stdout, stderr = self._communicate_with_poll(input)
1285 else:
1286 stdout, stderr = self._communicate_with_select(input)
1287
1288 # All data exchanged. Translate lists into strings.
1289 if stdout is not None:
1290 stdout = ''.join(stdout)
1291 if stderr is not None:
1292 stderr = ''.join(stderr)
1293
1294 # Translate newlines, if requested. We cannot let the file
1295 # object do the translation: It is based on stdio, which is
1296 # impossible to combine with select (unless forcing no
1297 # buffering).
1298 if self.universal_newlines and hasattr(file, 'newlines'):
1299 if stdout:
1300 stdout = self._translate_newlines(stdout)
1301 if stderr:
1302 stderr = self._translate_newlines(stderr)
1303
1304 self.wait()
1305 return (stdout, stderr)
1306
1307
1308 def _communicate_with_poll(self, input):
1309 stdout = None # Return
1310 stderr = None # Return
1311 fd2file = {}
1312 fd2output = {}
1313
1314 poller = select.poll()
1315 def register_and_append(file_obj, eventmask):
1316 poller.register(file_obj.fileno(), eventmask)
1317 fd2file[file_obj.fileno()] = file_obj
1318
1319 def close_unregister_and_remove(fd):
1320 poller.unregister(fd)
1321 fd2file[fd].close()
1322 fd2file.pop(fd)
1323
1324 if self.stdin and input:
1325 register_and_append(self.stdin, select.POLLOUT)
1326
1327 select_POLLIN_POLLPRI = select.POLLIN | select.POLLPRI
1328 if self.stdout:
1329 register_and_append(self.stdout, select_POLLIN_POLLPRI)
1330 fd2output[self.stdout.fileno()] = stdout = []
1331 if self.stderr:
1332 register_and_append(self.stderr, select_POLLIN_POLLPRI)
1333 fd2output[self.stderr.fileno()] = stderr = []
1334
1335 input_offset = 0
1336 while fd2file:
1337 try:
1338 ready = poller.poll()
1339 except select.error, e:
1340 if e.args[0] == errno.EINTR:
1341 continue
1342 raise
1343
1344 for fd, mode in ready:
1345 if mode & select.POLLOUT:
1346 chunk = input[input_offset : input_offset + _PIPE_BUF]
Ross Lagerwall104c3f12011-04-05 15:24:34 +02001347 try:
1348 input_offset += os.write(fd, chunk)
1349 except OSError as e:
1350 if e.errno == errno.EPIPE:
1351 close_unregister_and_remove(fd)
1352 else:
1353 raise
1354 else:
1355 if input_offset >= len(input):
1356 close_unregister_and_remove(fd)
Gregory P. Smithdd7ca242009-07-04 01:49:29 +00001357 elif mode & select_POLLIN_POLLPRI:
1358 data = os.read(fd, 4096)
1359 if not data:
1360 close_unregister_and_remove(fd)
1361 fd2output[fd].append(data)
1362 else:
1363 # Ignore hang up or errors.
1364 close_unregister_and_remove(fd)
1365
1366 return (stdout, stderr)
1367
1368
1369 def _communicate_with_select(self, input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001370 read_set = []
1371 write_set = []
1372 stdout = None # Return
1373 stderr = None # Return
1374
Gregory P. Smithdd7ca242009-07-04 01:49:29 +00001375 if self.stdin and input:
1376 write_set.append(self.stdin)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001377 if self.stdout:
1378 read_set.append(self.stdout)
Tim Peterse718f612004-10-12 21:51:32 +00001379 stdout = []
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001380 if self.stderr:
1381 read_set.append(self.stderr)
1382 stderr = []
1383
Peter Astrand1812f8c2007-01-07 14:34:16 +00001384 input_offset = 0
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001385 while read_set or write_set:
Gregory P. Smithf4140642008-07-06 07:16:40 +00001386 try:
1387 rlist, wlist, xlist = select.select(read_set, write_set, [])
1388 except select.error, e:
1389 if e.args[0] == errno.EINTR:
1390 continue
1391 raise
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001392
1393 if self.stdin in wlist:
Gregory P. Smithdd7ca242009-07-04 01:49:29 +00001394 chunk = input[input_offset : input_offset + _PIPE_BUF]
Ross Lagerwall104c3f12011-04-05 15:24:34 +02001395 try:
1396 bytes_written = os.write(self.stdin.fileno(), chunk)
1397 except OSError as e:
1398 if e.errno == errno.EPIPE:
1399 self.stdin.close()
1400 write_set.remove(self.stdin)
1401 else:
1402 raise
1403 else:
1404 input_offset += bytes_written
1405 if input_offset >= len(input):
1406 self.stdin.close()
1407 write_set.remove(self.stdin)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001408
1409 if self.stdout in rlist:
1410 data = os.read(self.stdout.fileno(), 1024)
1411 if data == "":
1412 self.stdout.close()
1413 read_set.remove(self.stdout)
1414 stdout.append(data)
1415
1416 if self.stderr in rlist:
1417 data = os.read(self.stderr.fileno(), 1024)
1418 if data == "":
1419 self.stderr.close()
1420 read_set.remove(self.stderr)
1421 stderr.append(data)
1422
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001423 return (stdout, stderr)
1424
Gregory P. Smithdd7ca242009-07-04 01:49:29 +00001425
Christian Heimese74c8f22008-04-19 02:23:57 +00001426 def send_signal(self, sig):
1427 """Send a signal to the process
1428 """
1429 os.kill(self.pid, sig)
1430
1431 def terminate(self):
1432 """Terminate the process with SIGTERM
1433 """
1434 self.send_signal(signal.SIGTERM)
1435
1436 def kill(self):
1437 """Kill the process with SIGKILL
1438 """
1439 self.send_signal(signal.SIGKILL)
1440
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001441
1442def _demo_posix():
1443 #
1444 # Example 1: Simple redirection: Get process list
1445 #
1446 plist = Popen(["ps"], stdout=PIPE).communicate()[0]
1447 print "Process list:"
1448 print plist
1449
1450 #
1451 # Example 2: Change uid before executing child
1452 #
1453 if os.getuid() == 0:
1454 p = Popen(["id"], preexec_fn=lambda: os.setuid(100))
1455 p.wait()
1456
1457 #
1458 # Example 3: Connecting several subprocesses
1459 #
1460 print "Looking for 'hda'..."
1461 p1 = Popen(["dmesg"], stdout=PIPE)
1462 p2 = Popen(["grep", "hda"], stdin=p1.stdout, stdout=PIPE)
1463 print repr(p2.communicate()[0])
1464
1465 #
1466 # Example 4: Catch execution error
1467 #
1468 print
1469 print "Trying a weird file..."
1470 try:
1471 print Popen(["/this/path/does/not/exist"]).communicate()
1472 except OSError, e:
1473 if e.errno == errno.ENOENT:
1474 print "The file didn't exist. I thought so..."
1475 print "Child traceback:"
1476 print e.child_traceback
1477 else:
1478 print "Error", e.errno
1479 else:
1480 print >>sys.stderr, "Gosh. No error."
1481
1482
1483def _demo_windows():
1484 #
1485 # Example 1: Connecting several subprocesses
1486 #
1487 print "Looking for 'PROMPT' in set output..."
1488 p1 = Popen("set", stdout=PIPE, shell=True)
1489 p2 = Popen('find "PROMPT"', stdin=p1.stdout, stdout=PIPE)
1490 print repr(p2.communicate()[0])
1491
1492 #
1493 # Example 2: Simple execution of program
1494 #
1495 print "Executing calc..."
1496 p = Popen("calc")
1497 p.wait()
1498
1499
1500if __name__ == "__main__":
1501 if mswindows:
1502 _demo_windows()
1503 else:
1504 _demo_posix()