| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1 | # subprocess - Subprocesses with accessible I/O streams | 
 | 2 | # | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 3 | # For more information about this module, see PEP 324. | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 4 | # | 
| Peter Astrand | c26516b | 2005-02-21 08:13:02 +0000 | [diff] [blame] | 5 | # This module should remain compatible with Python 2.2, see PEP 291. | 
 | 6 | # | 
| Peter Astrand | 3a708df | 2005-09-23 17:37:29 +0000 | [diff] [blame] | 7 | # Copyright (c) 2003-2005 by Peter Astrand <astrand@lysator.liu.se> | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 8 | # | 
| Peter Astrand | 69bf13f | 2005-02-14 08:56:32 +0000 | [diff] [blame] | 9 | # Licensed to PSF under a Contributor Agreement. | 
| Peter Astrand | 3a708df | 2005-09-23 17:37:29 +0000 | [diff] [blame] | 10 | # See http://www.python.org/2.4/license for licensing details. | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 11 |  | 
| Raymond Hettinger | 837dd93 | 2004-10-17 16:36:53 +0000 | [diff] [blame] | 12 | r"""subprocess - Subprocesses with accessible I/O streams | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 13 |  | 
| Fredrik Lundh | 15aaacc | 2004-10-17 14:47:05 +0000 | [diff] [blame] | 14 | This module allows you to spawn processes, connect to their | 
 | 15 | input/output/error pipes, and obtain their return codes.  This module | 
 | 16 | intends to replace several other, older modules and functions, like: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 17 |  | 
 | 18 | os.system | 
 | 19 | os.spawn* | 
 | 20 | os.popen* | 
 | 21 | popen2.* | 
 | 22 | commands.* | 
 | 23 |  | 
 | 24 | Information about how the subprocess module can be used to replace these | 
 | 25 | modules and functions can be found below. | 
 | 26 |  | 
 | 27 |  | 
 | 28 |  | 
 | 29 | Using the subprocess module | 
 | 30 | =========================== | 
 | 31 | This module defines one class called Popen: | 
 | 32 |  | 
 | 33 | class 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 |  | 
 | 40 | Arguments are: | 
 | 41 |  | 
 | 42 | args should be a string, or a sequence of program arguments.  The | 
 | 43 | program to execute is normally the first item in the args sequence or | 
 | 44 | string, but can be explicitly set by using the executable argument. | 
 | 45 |  | 
 | 46 | On UNIX, with shell=False (default): In this case, the Popen class | 
 | 47 | uses os.execvp() to execute the child program.  args should normally | 
 | 48 | be a sequence.  A string will be treated as a sequence with the string | 
 | 49 | as the only item (the program to execute). | 
 | 50 |  | 
 | 51 | On UNIX, with shell=True: If args is a string, it specifies the | 
 | 52 | command string to execute through the shell.  If args is a sequence, | 
 | 53 | the first item specifies the command string, and any additional items | 
 | 54 | will be treated as additional shell arguments. | 
 | 55 |  | 
 | 56 | On Windows: the Popen class uses CreateProcess() to execute the child | 
 | 57 | program, which operates on strings.  If args is a sequence, it will be | 
 | 58 | converted to a string using the list2cmdline method.  Please note that | 
 | 59 | not all MS Windows applications interpret the command line the same | 
 | 60 | way: The list2cmdline is designed for applications using the same | 
 | 61 | rules as the MS C runtime. | 
 | 62 |  | 
 | 63 | bufsize, if given, has the same meaning as the corresponding argument | 
 | 64 | to the built-in open() function: 0 means unbuffered, 1 means line | 
 | 65 | buffered, any other positive value means use a buffer of | 
 | 66 | (approximately) that size.  A negative bufsize means to use the system | 
 | 67 | default, which usually means fully buffered.  The default value for | 
 | 68 | bufsize is 0 (unbuffered). | 
 | 69 |  | 
 | 70 | stdin, stdout and stderr specify the executed programs' standard | 
 | 71 | input, standard output and standard error file handles, respectively. | 
 | 72 | Valid values are PIPE, an existing file descriptor (a positive | 
 | 73 | integer), an existing file object, and None.  PIPE indicates that a | 
 | 74 | new pipe to the child should be created.  With None, no redirection | 
 | 75 | will occur; the child's file handles will be inherited from the | 
 | 76 | parent.  Additionally, stderr can be STDOUT, which indicates that the | 
 | 77 | stderr data from the applications should be captured into the same | 
 | 78 | file handle as for stdout. | 
 | 79 |  | 
 | 80 | If preexec_fn is set to a callable object, this object will be called | 
 | 81 | in the child process just before the child is executed. | 
 | 82 |  | 
 | 83 | If close_fds is true, all file descriptors except 0, 1 and 2 will be | 
 | 84 | closed before the child process is executed. | 
 | 85 |  | 
 | 86 | if shell is true, the specified command will be executed through the | 
 | 87 | shell. | 
 | 88 |  | 
 | 89 | If cwd is not None, the current directory will be changed to cwd | 
 | 90 | before the child is executed. | 
 | 91 |  | 
 | 92 | If env is not None, it defines the environment variables for the new | 
 | 93 | process. | 
 | 94 |  | 
 | 95 | If universal_newlines is true, the file objects stdout and stderr are | 
 | 96 | opened as a text files, but lines may be terminated by any of '\n', | 
 | 97 | the Unix end-of-line convention, '\r', the Macintosh convention or | 
 | 98 | '\r\n', the Windows convention.  All of these external representations | 
 | 99 | are seen as '\n' by the Python program.  Note: This feature is only | 
 | 100 | available if Python is built with universal newline support (the | 
 | 101 | default).  Also, the newlines attribute of the file objects stdout, | 
 | 102 | stdin and stderr are not updated by the communicate() method. | 
 | 103 |  | 
 | 104 | The startupinfo and creationflags, if given, will be passed to the | 
 | 105 | underlying CreateProcess() function.  They can specify things such as | 
 | 106 | appearance of the main window and priority for the new process. | 
 | 107 | (Windows only) | 
 | 108 |  | 
 | 109 |  | 
| Gregory P. Smith | 97f49f4 | 2008-12-04 20:21:09 +0000 | [diff] [blame] | 110 | This module also defines some shortcut functions: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 111 |  | 
| Peter Astrand | 5f5e141 | 2004-12-05 20:15:36 +0000 | [diff] [blame] | 112 | call(*popenargs, **kwargs): | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 113 |     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 Astrand | 454f767 | 2005-01-01 09:36:35 +0000 | [diff] [blame] | 120 | check_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 Astrand | 7d1d436 | 2006-07-14 14:04:45 +0000 | [diff] [blame] | 124 |     return code in the returncode attribute. | 
| Peter Astrand | 454f767 | 2005-01-01 09:36:35 +0000 | [diff] [blame] | 125 |  | 
 | 126 |     The arguments are the same as for the Popen constructor.  Example: | 
 | 127 |  | 
 | 128 |     check_call(["ls", "-l"]) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 129 |  | 
| Gregory P. Smith | 2657680 | 2008-12-05 02:27:01 +0000 | [diff] [blame] | 130 | check_output(*popenargs, **kwargs): | 
| Gregory P. Smith | 97f49f4 | 2008-12-04 20:21:09 +0000 | [diff] [blame] | 131 |    Run command with arguments and return its output as a byte string. | 
 | 132 |  | 
 | 133 |    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. | 
 | 136 |  | 
 | 137 |    The arguments are the same as for the Popen constructor.  Example: | 
 | 138 |  | 
| Gregory P. Smith | 2657680 | 2008-12-05 02:27:01 +0000 | [diff] [blame] | 139 |       output = subprocess.check_output(["ls", "-l", "/dev/null"]) | 
| Gregory P. Smith | 97f49f4 | 2008-12-04 20:21:09 +0000 | [diff] [blame] | 140 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 141 | Exceptions | 
 | 142 | ---------- | 
 | 143 | Exceptions raised in the child process, before the new program has | 
 | 144 | started to execute, will be re-raised in the parent.  Additionally, | 
 | 145 | the exception object will have one extra attribute called | 
 | 146 | 'child_traceback', which is a string containing traceback information | 
 | 147 | from the childs point of view. | 
 | 148 |  | 
 | 149 | The most common exception raised is OSError.  This occurs, for | 
 | 150 | example, when trying to execute a non-existent file.  Applications | 
 | 151 | should prepare for OSErrors. | 
 | 152 |  | 
 | 153 | A ValueError will be raised if Popen is called with invalid arguments. | 
 | 154 |  | 
| Gregory P. Smith | 2657680 | 2008-12-05 02:27:01 +0000 | [diff] [blame] | 155 | check_call() and check_output() will raise CalledProcessError, if the | 
| Gregory P. Smith | 97f49f4 | 2008-12-04 20:21:09 +0000 | [diff] [blame] | 156 | called process returns a non-zero return code. | 
| Peter Astrand | 454f767 | 2005-01-01 09:36:35 +0000 | [diff] [blame] | 157 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 158 |  | 
 | 159 | Security | 
 | 160 | -------- | 
 | 161 | Unlike some other popen functions, this implementation will never call | 
 | 162 | /bin/sh implicitly.  This means that all characters, including shell | 
 | 163 | metacharacters, can safely be passed to child processes. | 
 | 164 |  | 
 | 165 |  | 
 | 166 | Popen objects | 
 | 167 | ============= | 
 | 168 | Instances of the Popen class have the following methods: | 
 | 169 |  | 
 | 170 | poll() | 
 | 171 |     Check if child process has terminated.  Returns returncode | 
 | 172 |     attribute. | 
 | 173 |  | 
 | 174 | wait() | 
 | 175 |     Wait for child process to terminate.  Returns returncode attribute. | 
 | 176 |  | 
 | 177 | communicate(input=None) | 
 | 178 |     Interact with process: Send data to stdin.  Read data from stdout | 
 | 179 |     and stderr, until end-of-file is reached.  Wait for process to | 
| Neal Norwitz | a186ee2 | 2006-12-29 03:01:53 +0000 | [diff] [blame] | 180 |     terminate.  The optional input argument should be a string to be | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 181 |     sent to the child process, or None, if no data should be sent to | 
 | 182 |     the child. | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 183 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 184 |     communicate() returns a tuple (stdout, stderr). | 
 | 185 |  | 
 | 186 |     Note: The data read is buffered in memory, so do not use this | 
 | 187 |     method if the data size is large or unlimited. | 
 | 188 |  | 
 | 189 | The following attributes are also available: | 
 | 190 |  | 
 | 191 | stdin | 
 | 192 |     If the stdin argument is PIPE, this attribute is a file object | 
 | 193 |     that provides input to the child process.  Otherwise, it is None. | 
 | 194 |  | 
 | 195 | stdout | 
 | 196 |     If the stdout argument is PIPE, this attribute is a file object | 
 | 197 |     that provides output from the child process.  Otherwise, it is | 
 | 198 |     None. | 
 | 199 |  | 
 | 200 | stderr | 
 | 201 |     If the stderr argument is PIPE, this attribute is file object that | 
 | 202 |     provides error output from the child process.  Otherwise, it is | 
 | 203 |     None. | 
 | 204 |  | 
 | 205 | pid | 
 | 206 |     The process ID of the child process. | 
 | 207 |  | 
 | 208 | returncode | 
 | 209 |     The child return code.  A None value indicates that the process | 
 | 210 |     hasn't terminated yet.  A negative value -N indicates that the | 
 | 211 |     child was terminated by signal N (UNIX only). | 
 | 212 |  | 
 | 213 |  | 
 | 214 | Replacing older functions with the subprocess module | 
 | 215 | ==================================================== | 
 | 216 | In this section, "a ==> b" means that b can be used as a replacement | 
 | 217 | for a. | 
 | 218 |  | 
 | 219 | Note: All functions in this section fail (more or less) silently if | 
 | 220 | the executed program cannot be found; this module raises an OSError | 
 | 221 | exception. | 
 | 222 |  | 
 | 223 | In the following examples, we assume that the subprocess module is | 
 | 224 | imported with "from subprocess import *". | 
 | 225 |  | 
 | 226 |  | 
 | 227 | Replacing /bin/sh shell backquote | 
 | 228 | --------------------------------- | 
 | 229 | output=`mycmd myarg` | 
 | 230 | ==> | 
 | 231 | output = Popen(["mycmd", "myarg"], stdout=PIPE).communicate()[0] | 
 | 232 |  | 
 | 233 |  | 
 | 234 | Replacing shell pipe line | 
 | 235 | ------------------------- | 
 | 236 | output=`dmesg | grep hda` | 
 | 237 | ==> | 
 | 238 | p1 = Popen(["dmesg"], stdout=PIPE) | 
| Peter Astrand | 6fdf3cb | 2004-11-30 18:06:42 +0000 | [diff] [blame] | 239 | p2 = Popen(["grep", "hda"], stdin=p1.stdout, stdout=PIPE) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 240 | output = p2.communicate()[0] | 
 | 241 |  | 
 | 242 |  | 
 | 243 | Replacing os.system() | 
 | 244 | --------------------- | 
 | 245 | sts = os.system("mycmd" + " myarg") | 
 | 246 | ==> | 
 | 247 | p = Popen("mycmd" + " myarg", shell=True) | 
| Neal Norwitz | 8440483 | 2006-07-10 00:05:34 +0000 | [diff] [blame] | 248 | pid, sts = os.waitpid(p.pid, 0) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 249 |  | 
 | 250 | Note: | 
 | 251 |  | 
 | 252 | * Calling the program through the shell is usually not required. | 
 | 253 |  | 
 | 254 | * It's easier to look at the returncode attribute than the | 
 | 255 |   exitstatus. | 
 | 256 |  | 
 | 257 | A more real-world example would look like this: | 
 | 258 |  | 
 | 259 | try: | 
 | 260 |     retcode = call("mycmd" + " myarg", shell=True) | 
 | 261 |     if retcode < 0: | 
 | 262 |         print >>sys.stderr, "Child was terminated by signal", -retcode | 
 | 263 |     else: | 
 | 264 |         print >>sys.stderr, "Child returned", retcode | 
 | 265 | except OSError, e: | 
 | 266 |     print >>sys.stderr, "Execution failed:", e | 
 | 267 |  | 
 | 268 |  | 
 | 269 | Replacing os.spawn* | 
 | 270 | ------------------- | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 271 | P_NOWAIT example: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 272 |  | 
 | 273 | pid = os.spawnlp(os.P_NOWAIT, "/bin/mycmd", "mycmd", "myarg") | 
 | 274 | ==> | 
 | 275 | pid = Popen(["/bin/mycmd", "myarg"]).pid | 
 | 276 |  | 
 | 277 |  | 
 | 278 | P_WAIT example: | 
 | 279 |  | 
 | 280 | retcode = os.spawnlp(os.P_WAIT, "/bin/mycmd", "mycmd", "myarg") | 
 | 281 | ==> | 
 | 282 | retcode = call(["/bin/mycmd", "myarg"]) | 
 | 283 |  | 
 | 284 |  | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 285 | Vector example: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 286 |  | 
 | 287 | os.spawnvp(os.P_NOWAIT, path, args) | 
 | 288 | ==> | 
 | 289 | Popen([path] + args[1:]) | 
 | 290 |  | 
 | 291 |  | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 292 | Environment example: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 293 |  | 
 | 294 | os.spawnlpe(os.P_NOWAIT, "/bin/mycmd", "mycmd", "myarg", env) | 
 | 295 | ==> | 
 | 296 | Popen(["/bin/mycmd", "myarg"], env={"PATH": "/usr/bin"}) | 
 | 297 |  | 
 | 298 |  | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 299 | Replacing os.popen* | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 300 | ------------------- | 
 | 301 | pipe = os.popen(cmd, mode='r', bufsize) | 
 | 302 | ==> | 
 | 303 | pipe = Popen(cmd, shell=True, bufsize=bufsize, stdout=PIPE).stdout | 
 | 304 |  | 
 | 305 | pipe = os.popen(cmd, mode='w', bufsize) | 
 | 306 | ==> | 
 | 307 | pipe = Popen(cmd, shell=True, bufsize=bufsize, stdin=PIPE).stdin | 
 | 308 |  | 
 | 309 |  | 
 | 310 | (child_stdin, child_stdout) = os.popen2(cmd, mode, bufsize) | 
 | 311 | ==> | 
 | 312 | p = Popen(cmd, shell=True, bufsize=bufsize, | 
 | 313 |           stdin=PIPE, stdout=PIPE, close_fds=True) | 
 | 314 | (child_stdin, child_stdout) = (p.stdin, p.stdout) | 
 | 315 |  | 
 | 316 |  | 
 | 317 | (child_stdin, | 
 | 318 |  child_stdout, | 
 | 319 |  child_stderr) = os.popen3(cmd, mode, bufsize) | 
 | 320 | ==> | 
 | 321 | p = Popen(cmd, shell=True, bufsize=bufsize, | 
 | 322 |           stdin=PIPE, stdout=PIPE, stderr=PIPE, close_fds=True) | 
 | 323 | (child_stdin, | 
 | 324 |  child_stdout, | 
 | 325 |  child_stderr) = (p.stdin, p.stdout, p.stderr) | 
 | 326 |  | 
 | 327 |  | 
 | 328 | (child_stdin, child_stdout_and_stderr) = os.popen4(cmd, mode, bufsize) | 
 | 329 | ==> | 
 | 330 | p = Popen(cmd, shell=True, bufsize=bufsize, | 
 | 331 |           stdin=PIPE, stdout=PIPE, stderr=STDOUT, close_fds=True) | 
 | 332 | (child_stdin, child_stdout_and_stderr) = (p.stdin, p.stdout) | 
 | 333 |  | 
 | 334 |  | 
 | 335 | Replacing popen2.* | 
 | 336 | ------------------ | 
 | 337 | Note: If the cmd argument to popen2 functions is a string, the command | 
 | 338 | is executed through /bin/sh.  If it is a list, the command is directly | 
 | 339 | executed. | 
 | 340 |  | 
 | 341 | (child_stdout, child_stdin) = popen2.popen2("somestring", bufsize, mode) | 
 | 342 | ==> | 
 | 343 | p = Popen(["somestring"], shell=True, bufsize=bufsize | 
 | 344 |           stdin=PIPE, stdout=PIPE, close_fds=True) | 
 | 345 | (child_stdout, child_stdin) = (p.stdout, p.stdin) | 
 | 346 |  | 
 | 347 |  | 
 | 348 | (child_stdout, child_stdin) = popen2.popen2(["mycmd", "myarg"], bufsize, mode) | 
 | 349 | ==> | 
 | 350 | p = Popen(["mycmd", "myarg"], bufsize=bufsize, | 
 | 351 |           stdin=PIPE, stdout=PIPE, close_fds=True) | 
 | 352 | (child_stdout, child_stdin) = (p.stdout, p.stdin) | 
 | 353 |  | 
| Neal Norwitz | aa87fb6 | 2007-05-11 06:23:01 +0000 | [diff] [blame] | 354 | The popen2.Popen3 and popen2.Popen4 basically works as subprocess.Popen, | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 355 | except that: | 
 | 356 |  | 
 | 357 | * subprocess.Popen raises an exception if the execution fails | 
 | 358 | * the capturestderr argument is replaced with the stderr argument. | 
 | 359 | * stdin=PIPE and stdout=PIPE must be specified. | 
 | 360 | * popen2 closes all filedescriptors by default, but you have to specify | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 361 |   close_fds=True with subprocess.Popen. | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 362 | """ | 
 | 363 |  | 
 | 364 | import sys | 
 | 365 | mswindows = (sys.platform == "win32") | 
 | 366 |  | 
 | 367 | import os | 
| Peter Astrand | c26516b | 2005-02-21 08:13:02 +0000 | [diff] [blame] | 368 | import types | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 369 | import traceback | 
| Gregory P. Smith | 87d4979 | 2008-01-19 20:57:59 +0000 | [diff] [blame] | 370 | import gc | 
| Christian Heimes | e74c8f2 | 2008-04-19 02:23:57 +0000 | [diff] [blame] | 371 | import signal | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 372 |  | 
| Peter Astrand | 454f767 | 2005-01-01 09:36:35 +0000 | [diff] [blame] | 373 | # Exception classes used by this module. | 
| Peter Astrand | 7d1d436 | 2006-07-14 14:04:45 +0000 | [diff] [blame] | 374 | class CalledProcessError(Exception): | 
| Gregory P. Smith | 97f49f4 | 2008-12-04 20:21:09 +0000 | [diff] [blame] | 375 |     """This exception is raised when a process run by check_call() or | 
| Gregory P. Smith | 2657680 | 2008-12-05 02:27:01 +0000 | [diff] [blame] | 376 |     check_output() returns a non-zero exit status. | 
| Gregory P. Smith | 97f49f4 | 2008-12-04 20:21:09 +0000 | [diff] [blame] | 377 |     The exit status will be stored in the returncode attribute; | 
| Gregory P. Smith | 2657680 | 2008-12-05 02:27:01 +0000 | [diff] [blame] | 378 |     check_output() will also store the output in the output attribute. | 
| Gregory P. Smith | 97f49f4 | 2008-12-04 20:21:09 +0000 | [diff] [blame] | 379 |     """ | 
 | 380 |     def __init__(self, returncode, cmd, output=None): | 
| Peter Astrand | 7d1d436 | 2006-07-14 14:04:45 +0000 | [diff] [blame] | 381 |         self.returncode = returncode | 
 | 382 |         self.cmd = cmd | 
| Gregory P. Smith | 97f49f4 | 2008-12-04 20:21:09 +0000 | [diff] [blame] | 383 |         self.output = output | 
| Peter Astrand | 7d1d436 | 2006-07-14 14:04:45 +0000 | [diff] [blame] | 384 |     def __str__(self): | 
 | 385 |         return "Command '%s' returned non-zero exit status %d" % (self.cmd, self.returncode) | 
| Tim Peters | 73a9ead | 2006-07-18 21:55:15 +0000 | [diff] [blame] | 386 |  | 
| Peter Astrand | 454f767 | 2005-01-01 09:36:35 +0000 | [diff] [blame] | 387 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 388 | if mswindows: | 
 | 389 |     import threading | 
 | 390 |     import msvcrt | 
| Fredrik Lundh | 3e73a01 | 2004-10-13 18:19:18 +0000 | [diff] [blame] | 391 |     if 0: # <-- change this to use pywin32 instead of the _subprocess driver | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 392 |         import pywintypes | 
| Tim Peters | e8374a5 | 2004-10-13 03:15:00 +0000 | [diff] [blame] | 393 |         from win32api import GetStdHandle, STD_INPUT_HANDLE, \ | 
 | 394 |                              STD_OUTPUT_HANDLE, STD_ERROR_HANDLE | 
 | 395 |         from win32api import GetCurrentProcess, DuplicateHandle, \ | 
 | 396 |                              GetModuleFileName, GetVersion | 
| Peter Astrand | c1d6536 | 2004-11-07 14:30:34 +0000 | [diff] [blame] | 397 |         from win32con import DUPLICATE_SAME_ACCESS, SW_HIDE | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 398 |         from win32pipe import CreatePipe | 
| Tim Peters | e8374a5 | 2004-10-13 03:15:00 +0000 | [diff] [blame] | 399 |         from win32process import CreateProcess, STARTUPINFO, \ | 
 | 400 |                                  GetExitCodeProcess, STARTF_USESTDHANDLES, \ | 
| Peter Astrand | c1d6536 | 2004-11-07 14:30:34 +0000 | [diff] [blame] | 401 |                                  STARTF_USESHOWWINDOW, CREATE_NEW_CONSOLE | 
| Christian Heimes | e74c8f2 | 2008-04-19 02:23:57 +0000 | [diff] [blame] | 402 |         from win32process import TerminateProcess | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 403 |         from win32event import WaitForSingleObject, INFINITE, WAIT_OBJECT_0 | 
| Fredrik Lundh | 3e73a01 | 2004-10-13 18:19:18 +0000 | [diff] [blame] | 404 |     else: | 
 | 405 |         from _subprocess import * | 
 | 406 |         class STARTUPINFO: | 
 | 407 |             dwFlags = 0 | 
 | 408 |             hStdInput = None | 
 | 409 |             hStdOutput = None | 
 | 410 |             hStdError = None | 
| Georg Brandl | ad62489 | 2006-06-04 22:15:37 +0000 | [diff] [blame] | 411 |             wShowWindow = 0 | 
| Fredrik Lundh | 3e73a01 | 2004-10-13 18:19:18 +0000 | [diff] [blame] | 412 |         class pywintypes: | 
 | 413 |             error = IOError | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 414 | else: | 
 | 415 |     import select | 
 | 416 |     import errno | 
 | 417 |     import fcntl | 
 | 418 |     import pickle | 
 | 419 |  | 
| Gregory P. Smith | 97f49f4 | 2008-12-04 20:21:09 +0000 | [diff] [blame] | 420 | __all__ = ["Popen", "PIPE", "STDOUT", "call", "check_call", | 
| Gregory P. Smith | 2657680 | 2008-12-05 02:27:01 +0000 | [diff] [blame] | 421 |            "check_output", "CalledProcessError"] | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 422 |  | 
 | 423 | try: | 
 | 424 |     MAXFD = os.sysconf("SC_OPEN_MAX") | 
 | 425 | except: | 
 | 426 |     MAXFD = 256 | 
 | 427 |  | 
 | 428 | # True/False does not exist on 2.2.0 | 
| Brett Cannon | 03446c4 | 2008-08-08 04:19:32 +0000 | [diff] [blame] | 429 | #try: | 
 | 430 | #    False | 
 | 431 | #except NameError: | 
 | 432 | #    False = 0 | 
 | 433 | #    True = 1 | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 434 |  | 
 | 435 | _active = [] | 
 | 436 |  | 
 | 437 | def _cleanup(): | 
 | 438 |     for inst in _active[:]: | 
| Gregory P. Smith | a36f8fe | 2008-08-04 00:13:29 +0000 | [diff] [blame] | 439 |         if inst._internal_poll(_deadstate=sys.maxint) >= 0: | 
| Martin v. Löwis | 17de8ff | 2006-04-10 15:55:37 +0000 | [diff] [blame] | 440 |             try: | 
 | 441 |                 _active.remove(inst) | 
 | 442 |             except ValueError: | 
 | 443 |                 # This can happen if two threads create a new Popen instance. | 
 | 444 |                 # It's harmless that it was already removed, so ignore. | 
 | 445 |                 pass | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 446 |  | 
 | 447 | PIPE = -1 | 
 | 448 | STDOUT = -2 | 
 | 449 |  | 
 | 450 |  | 
| Peter Astrand | 5f5e141 | 2004-12-05 20:15:36 +0000 | [diff] [blame] | 451 | def call(*popenargs, **kwargs): | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 452 |     """Run command with arguments.  Wait for command to complete, then | 
 | 453 |     return the returncode attribute. | 
 | 454 |  | 
 | 455 |     The arguments are the same as for the Popen constructor.  Example: | 
 | 456 |  | 
 | 457 |     retcode = call(["ls", "-l"]) | 
 | 458 |     """ | 
| Peter Astrand | 5f5e141 | 2004-12-05 20:15:36 +0000 | [diff] [blame] | 459 |     return Popen(*popenargs, **kwargs).wait() | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 460 |  | 
 | 461 |  | 
| Peter Astrand | 454f767 | 2005-01-01 09:36:35 +0000 | [diff] [blame] | 462 | def check_call(*popenargs, **kwargs): | 
 | 463 |     """Run command with arguments.  Wait for command to complete.  If | 
 | 464 |     the exit code was zero then return, otherwise raise | 
 | 465 |     CalledProcessError.  The CalledProcessError object will have the | 
| Peter Astrand | 7d1d436 | 2006-07-14 14:04:45 +0000 | [diff] [blame] | 466 |     return code in the returncode attribute. | 
| Peter Astrand | 454f767 | 2005-01-01 09:36:35 +0000 | [diff] [blame] | 467 |  | 
 | 468 |     The arguments are the same as for the Popen constructor.  Example: | 
 | 469 |  | 
 | 470 |     check_call(["ls", "-l"]) | 
 | 471 |     """ | 
 | 472 |     retcode = call(*popenargs, **kwargs) | 
| Peter Astrand | 454f767 | 2005-01-01 09:36:35 +0000 | [diff] [blame] | 473 |     if retcode: | 
| Gregory P. Smith | 97f49f4 | 2008-12-04 20:21:09 +0000 | [diff] [blame] | 474 |         cmd = kwargs.get("args") | 
 | 475 |         if cmd is None: | 
 | 476 |             cmd = popenargs[0] | 
| Peter Astrand | 7d1d436 | 2006-07-14 14:04:45 +0000 | [diff] [blame] | 477 |         raise CalledProcessError(retcode, cmd) | 
| Gregory P. Smith | 97f49f4 | 2008-12-04 20:21:09 +0000 | [diff] [blame] | 478 |     return 0 | 
 | 479 |  | 
 | 480 |  | 
| Gregory P. Smith | 2657680 | 2008-12-05 02:27:01 +0000 | [diff] [blame] | 481 | def check_output(*popenargs, **kwargs): | 
| Gregory P. Smith | 97f49f4 | 2008-12-04 20:21:09 +0000 | [diff] [blame] | 482 |     """Run command with arguments and return its output as a byte string. | 
 | 483 |  | 
 | 484 |     If the exit code was non-zero it raises a CalledProcessError.  The | 
 | 485 |     CalledProcessError object will have the return code in the returncode | 
 | 486 |     attribute and output in the output attribute. | 
 | 487 |  | 
 | 488 |     The arguments are the same as for the Popen constructor.  Example: | 
 | 489 |  | 
| Gregory P. Smith | 2657680 | 2008-12-05 02:27:01 +0000 | [diff] [blame] | 490 |     >>> check_output(["ls", "-l", "/dev/null"]) | 
| Gregory P. Smith | 97f49f4 | 2008-12-04 20:21:09 +0000 | [diff] [blame] | 491 |     'crw-rw-rw- 1 root root 1, 3 Oct 18  2007 /dev/null\n' | 
 | 492 |  | 
 | 493 |     The stdout argument is not allowed as it is used internally. | 
 | 494 |     To capture standard error in the result, use stderr=subprocess.STDOUT. | 
 | 495 |  | 
| Gregory P. Smith | 2657680 | 2008-12-05 02:27:01 +0000 | [diff] [blame] | 496 |     >>> check_output(["/bin/sh", "-c", | 
| Mark Dickinson | 3e4caeb | 2009-02-21 20:27:01 +0000 | [diff] [blame] | 497 |                       "ls -l non_existent_file ; exit 0"], | 
| Gregory P. Smith | 2657680 | 2008-12-05 02:27:01 +0000 | [diff] [blame] | 498 |                      stderr=subprocess.STDOUT) | 
| Mark Dickinson | 3e4caeb | 2009-02-21 20:27:01 +0000 | [diff] [blame] | 499 |     'ls: non_existent_file: No such file or directory\n' | 
| Gregory P. Smith | 97f49f4 | 2008-12-04 20:21:09 +0000 | [diff] [blame] | 500 |     """ | 
 | 501 |     if 'stdout' in kwargs: | 
 | 502 |         raise ValueError('stdout argument not allowed, it will be overridden.') | 
 | 503 |     process = Popen(*popenargs, stdout=PIPE, **kwargs) | 
 | 504 |     output, unused_err = process.communicate() | 
 | 505 |     retcode = process.poll() | 
 | 506 |     if retcode: | 
 | 507 |         cmd = kwargs.get("args") | 
 | 508 |         if cmd is None: | 
 | 509 |             cmd = popenargs[0] | 
 | 510 |         raise CalledProcessError(retcode, cmd, output=output) | 
 | 511 |     return output | 
| Peter Astrand | 454f767 | 2005-01-01 09:36:35 +0000 | [diff] [blame] | 512 |  | 
 | 513 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 514 | def list2cmdline(seq): | 
 | 515 |     """ | 
 | 516 |     Translate a sequence of arguments into a command line | 
 | 517 |     string, using the same rules as the MS C runtime: | 
 | 518 |  | 
 | 519 |     1) Arguments are delimited by white space, which is either a | 
 | 520 |        space or a tab. | 
 | 521 |  | 
 | 522 |     2) A string surrounded by double quotation marks is | 
 | 523 |        interpreted as a single argument, regardless of white space | 
| Gregory P. Smith | 70eb2f9 | 2008-01-19 22:49:37 +0000 | [diff] [blame] | 524 |        or pipe characters contained within.  A quoted string can be | 
 | 525 |        embedded in an argument. | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 526 |  | 
 | 527 |     3) A double quotation mark preceded by a backslash is | 
 | 528 |        interpreted as a literal double quotation mark. | 
 | 529 |  | 
 | 530 |     4) Backslashes are interpreted literally, unless they | 
 | 531 |        immediately precede a double quotation mark. | 
 | 532 |  | 
 | 533 |     5) If backslashes immediately precede a double quotation mark, | 
 | 534 |        every pair of backslashes is interpreted as a literal | 
 | 535 |        backslash.  If the number of backslashes is odd, the last | 
 | 536 |        backslash escapes the next double quotation mark as | 
 | 537 |        described in rule 3. | 
 | 538 |     """ | 
 | 539 |  | 
 | 540 |     # See | 
 | 541 |     # http://msdn.microsoft.com/library/en-us/vccelng/htm/progs_12.asp | 
 | 542 |     result = [] | 
 | 543 |     needquote = False | 
 | 544 |     for arg in seq: | 
 | 545 |         bs_buf = [] | 
 | 546 |  | 
 | 547 |         # Add a space to separate this argument from the others | 
 | 548 |         if result: | 
 | 549 |             result.append(' ') | 
 | 550 |  | 
| Gregory P. Smith | 70eb2f9 | 2008-01-19 22:49:37 +0000 | [diff] [blame] | 551 |         needquote = (" " in arg) or ("\t" in arg) or ("|" in arg) or not arg | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 552 |         if needquote: | 
 | 553 |             result.append('"') | 
 | 554 |  | 
 | 555 |         for c in arg: | 
 | 556 |             if c == '\\': | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 557 |                 # Don't know if we need to double yet. | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 558 |                 bs_buf.append(c) | 
 | 559 |             elif c == '"': | 
| Gregory P. Smith | e047e6d | 2008-01-19 20:49:02 +0000 | [diff] [blame] | 560 |                 # Double backslashes. | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 561 |                 result.append('\\' * len(bs_buf)*2) | 
 | 562 |                 bs_buf = [] | 
 | 563 |                 result.append('\\"') | 
 | 564 |             else: | 
 | 565 |                 # Normal char | 
 | 566 |                 if bs_buf: | 
 | 567 |                     result.extend(bs_buf) | 
 | 568 |                     bs_buf = [] | 
 | 569 |                 result.append(c) | 
 | 570 |  | 
| Gregory P. Smith | e047e6d | 2008-01-19 20:49:02 +0000 | [diff] [blame] | 571 |         # Add remaining backslashes, if any. | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 572 |         if bs_buf: | 
 | 573 |             result.extend(bs_buf) | 
 | 574 |  | 
 | 575 |         if needquote: | 
| Peter Astrand | 7e78ade | 2005-03-03 21:10:23 +0000 | [diff] [blame] | 576 |             result.extend(bs_buf) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 577 |             result.append('"') | 
 | 578 |  | 
 | 579 |     return ''.join(result) | 
 | 580 |  | 
 | 581 |  | 
 | 582 | class Popen(object): | 
 | 583 |     def __init__(self, args, bufsize=0, executable=None, | 
 | 584 |                  stdin=None, stdout=None, stderr=None, | 
 | 585 |                  preexec_fn=None, close_fds=False, shell=False, | 
 | 586 |                  cwd=None, env=None, universal_newlines=False, | 
 | 587 |                  startupinfo=None, creationflags=0): | 
 | 588 |         """Create new Popen instance.""" | 
 | 589 |         _cleanup() | 
 | 590 |  | 
| Martin v. Löwis | 17de8ff | 2006-04-10 15:55:37 +0000 | [diff] [blame] | 591 |         self._child_created = False | 
| Peter Astrand | 738131d | 2004-11-30 21:04:45 +0000 | [diff] [blame] | 592 |         if not isinstance(bufsize, (int, long)): | 
 | 593 |             raise TypeError("bufsize must be an integer") | 
 | 594 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 595 |         if mswindows: | 
| Tim Peters | e8374a5 | 2004-10-13 03:15:00 +0000 | [diff] [blame] | 596 |             if preexec_fn is not None: | 
 | 597 |                 raise ValueError("preexec_fn is not supported on Windows " | 
 | 598 |                                  "platforms") | 
| Peter Astrand | 81a191b | 2007-05-26 22:18:20 +0000 | [diff] [blame] | 599 |             if close_fds and (stdin is not None or stdout is not None or | 
 | 600 |                               stderr is not None): | 
| Tim Peters | e8374a5 | 2004-10-13 03:15:00 +0000 | [diff] [blame] | 601 |                 raise ValueError("close_fds is not supported on Windows " | 
| Peter Astrand | 81a191b | 2007-05-26 22:18:20 +0000 | [diff] [blame] | 602 |                                  "platforms if you redirect stdin/stdout/stderr") | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 603 |         else: | 
 | 604 |             # POSIX | 
| Tim Peters | e8374a5 | 2004-10-13 03:15:00 +0000 | [diff] [blame] | 605 |             if startupinfo is not None: | 
 | 606 |                 raise ValueError("startupinfo is only supported on Windows " | 
 | 607 |                                  "platforms") | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 608 |             if creationflags != 0: | 
| Tim Peters | e8374a5 | 2004-10-13 03:15:00 +0000 | [diff] [blame] | 609 |                 raise ValueError("creationflags is only supported on Windows " | 
 | 610 |                                  "platforms") | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 611 |  | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 612 |         self.stdin = None | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 613 |         self.stdout = None | 
 | 614 |         self.stderr = None | 
 | 615 |         self.pid = None | 
 | 616 |         self.returncode = None | 
 | 617 |         self.universal_newlines = universal_newlines | 
 | 618 |  | 
 | 619 |         # Input and output objects. The general principle is like | 
 | 620 |         # this: | 
 | 621 |         # | 
 | 622 |         # Parent                   Child | 
 | 623 |         # ------                   ----- | 
 | 624 |         # p2cwrite   ---stdin--->  p2cread | 
 | 625 |         # c2pread    <--stdout---  c2pwrite | 
 | 626 |         # errread    <--stderr---  errwrite | 
 | 627 |         # | 
 | 628 |         # On POSIX, the child objects are file descriptors.  On | 
 | 629 |         # Windows, these are Windows file handles.  The parent objects | 
 | 630 |         # are file descriptors on both platforms.  The parent objects | 
 | 631 |         # are None when not using PIPEs. The child objects are None | 
 | 632 |         # when not redirecting. | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 633 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 634 |         (p2cread, p2cwrite, | 
 | 635 |          c2pread, c2pwrite, | 
 | 636 |          errread, errwrite) = self._get_handles(stdin, stdout, stderr) | 
 | 637 |  | 
 | 638 |         self._execute_child(args, executable, preexec_fn, close_fds, | 
 | 639 |                             cwd, env, universal_newlines, | 
 | 640 |                             startupinfo, creationflags, shell, | 
 | 641 |                             p2cread, p2cwrite, | 
 | 642 |                             c2pread, c2pwrite, | 
 | 643 |                             errread, errwrite) | 
 | 644 |  | 
| Peter Astrand | 5f9c6ae | 2007-02-06 15:37:50 +0000 | [diff] [blame] | 645 |         if mswindows: | 
| Hirokazu Yamamoto | eacbbdf | 2009-03-03 22:18:14 +0000 | [diff] [blame] | 646 |             if p2cwrite is not None: | 
 | 647 |                 p2cwrite = msvcrt.open_osfhandle(p2cwrite.Detach(), 0) | 
 | 648 |             if c2pread is not None: | 
 | 649 |                 c2pread = msvcrt.open_osfhandle(c2pread.Detach(), 0) | 
 | 650 |             if errread is not None: | 
 | 651 |                 errread = msvcrt.open_osfhandle(errread.Detach(), 0) | 
| Peter Astrand | 5f9c6ae | 2007-02-06 15:37:50 +0000 | [diff] [blame] | 652 |  | 
| Peter Astrand | f540003 | 2007-02-02 19:06:36 +0000 | [diff] [blame] | 653 |         if p2cwrite is not None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 654 |             self.stdin = os.fdopen(p2cwrite, 'wb', bufsize) | 
| Peter Astrand | f540003 | 2007-02-02 19:06:36 +0000 | [diff] [blame] | 655 |         if c2pread is not None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 656 |             if universal_newlines: | 
 | 657 |                 self.stdout = os.fdopen(c2pread, 'rU', bufsize) | 
 | 658 |             else: | 
 | 659 |                 self.stdout = os.fdopen(c2pread, 'rb', bufsize) | 
| Peter Astrand | f540003 | 2007-02-02 19:06:36 +0000 | [diff] [blame] | 660 |         if errread is not None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 661 |             if universal_newlines: | 
 | 662 |                 self.stderr = os.fdopen(errread, 'rU', bufsize) | 
 | 663 |             else: | 
 | 664 |                 self.stderr = os.fdopen(errread, 'rb', bufsize) | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 665 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 666 |  | 
 | 667 |     def _translate_newlines(self, data): | 
 | 668 |         data = data.replace("\r\n", "\n") | 
 | 669 |         data = data.replace("\r", "\n") | 
 | 670 |         return data | 
 | 671 |  | 
| Martin v. Löwis | 17de8ff | 2006-04-10 15:55:37 +0000 | [diff] [blame] | 672 |  | 
| Georg Brandl | 2452298 | 2007-04-21 20:35:38 +0000 | [diff] [blame] | 673 |     def __del__(self, sys=sys): | 
| Martin v. Löwis | 17de8ff | 2006-04-10 15:55:37 +0000 | [diff] [blame] | 674 |         if not self._child_created: | 
 | 675 |             # We didn't get to successfully create a child process. | 
 | 676 |             return | 
 | 677 |         # In case the child hasn't been waited on, check if it's done. | 
| Gregory P. Smith | a36f8fe | 2008-08-04 00:13:29 +0000 | [diff] [blame] | 678 |         self._internal_poll(_deadstate=sys.maxint) | 
| Georg Brandl | 13cf38c | 2006-07-20 16:28:39 +0000 | [diff] [blame] | 679 |         if self.returncode is None and _active is not None: | 
| Martin v. Löwis | 17de8ff | 2006-04-10 15:55:37 +0000 | [diff] [blame] | 680 |             # Child is still running, keep us alive until we can wait on it. | 
 | 681 |             _active.append(self) | 
 | 682 |  | 
 | 683 |  | 
| Peter Astrand | 23109f0 | 2005-03-03 20:28:59 +0000 | [diff] [blame] | 684 |     def communicate(self, input=None): | 
 | 685 |         """Interact with process: Send data to stdin.  Read data from | 
 | 686 |         stdout and stderr, until end-of-file is reached.  Wait for | 
 | 687 |         process to terminate.  The optional input argument should be a | 
 | 688 |         string to be sent to the child process, or None, if no data | 
 | 689 |         should be sent to the child. | 
| Tim Peters | eba28be | 2005-03-28 01:08:02 +0000 | [diff] [blame] | 690 |  | 
| Peter Astrand | 23109f0 | 2005-03-03 20:28:59 +0000 | [diff] [blame] | 691 |         communicate() returns a tuple (stdout, stderr).""" | 
 | 692 |  | 
 | 693 |         # Optimization: If we are only using one pipe, or no pipe at | 
 | 694 |         # all, using select() or threads is unnecessary. | 
 | 695 |         if [self.stdin, self.stdout, self.stderr].count(None) >= 2: | 
| Tim Peters | eba28be | 2005-03-28 01:08:02 +0000 | [diff] [blame] | 696 |             stdout = None | 
 | 697 |             stderr = None | 
| Peter Astrand | 23109f0 | 2005-03-03 20:28:59 +0000 | [diff] [blame] | 698 |             if self.stdin: | 
 | 699 |                 if input: | 
 | 700 |                     self.stdin.write(input) | 
 | 701 |                 self.stdin.close() | 
 | 702 |             elif self.stdout: | 
 | 703 |                 stdout = self.stdout.read() | 
| Gregory P. Smith | 4036fd4 | 2008-05-26 20:22:14 +0000 | [diff] [blame] | 704 |                 self.stdout.close() | 
| Peter Astrand | 23109f0 | 2005-03-03 20:28:59 +0000 | [diff] [blame] | 705 |             elif self.stderr: | 
 | 706 |                 stderr = self.stderr.read() | 
| Gregory P. Smith | 4036fd4 | 2008-05-26 20:22:14 +0000 | [diff] [blame] | 707 |                 self.stderr.close() | 
| Peter Astrand | 23109f0 | 2005-03-03 20:28:59 +0000 | [diff] [blame] | 708 |             self.wait() | 
 | 709 |             return (stdout, stderr) | 
| Tim Peters | eba28be | 2005-03-28 01:08:02 +0000 | [diff] [blame] | 710 |  | 
| Peter Astrand | 23109f0 | 2005-03-03 20:28:59 +0000 | [diff] [blame] | 711 |         return self._communicate(input) | 
 | 712 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 713 |  | 
| Gregory P. Smith | a36f8fe | 2008-08-04 00:13:29 +0000 | [diff] [blame] | 714 |     def poll(self): | 
 | 715 |         return self._internal_poll() | 
 | 716 |  | 
 | 717 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 718 |     if mswindows: | 
 | 719 |         # | 
 | 720 |         # Windows methods | 
 | 721 |         # | 
 | 722 |         def _get_handles(self, stdin, stdout, stderr): | 
 | 723 |             """Construct and return tupel with IO objects: | 
 | 724 |             p2cread, p2cwrite, c2pread, c2pwrite, errread, errwrite | 
 | 725 |             """ | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 726 |             if stdin is None and stdout is None and stderr is None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 727 |                 return (None, None, None, None, None, None) | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 728 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 729 |             p2cread, p2cwrite = None, None | 
 | 730 |             c2pread, c2pwrite = None, None | 
 | 731 |             errread, errwrite = None, None | 
 | 732 |  | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 733 |             if stdin is None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 734 |                 p2cread = GetStdHandle(STD_INPUT_HANDLE) | 
| Hirokazu Yamamoto | eacbbdf | 2009-03-03 22:18:14 +0000 | [diff] [blame] | 735 |                 if p2cread is None: | 
 | 736 |                     p2cread, _ = CreatePipe(None, 0) | 
 | 737 |             elif stdin == PIPE: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 738 |                 p2cread, p2cwrite = CreatePipe(None, 0) | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 739 |             elif isinstance(stdin, int): | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 740 |                 p2cread = msvcrt.get_osfhandle(stdin) | 
 | 741 |             else: | 
 | 742 |                 # Assuming file-like object | 
 | 743 |                 p2cread = msvcrt.get_osfhandle(stdin.fileno()) | 
 | 744 |             p2cread = self._make_inheritable(p2cread) | 
 | 745 |  | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 746 |             if stdout is None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 747 |                 c2pwrite = GetStdHandle(STD_OUTPUT_HANDLE) | 
| Hirokazu Yamamoto | eacbbdf | 2009-03-03 22:18:14 +0000 | [diff] [blame] | 748 |                 if c2pwrite is None: | 
 | 749 |                     _, c2pwrite = CreatePipe(None, 0) | 
 | 750 |             elif stdout == PIPE: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 751 |                 c2pread, c2pwrite = CreatePipe(None, 0) | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 752 |             elif isinstance(stdout, int): | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 753 |                 c2pwrite = msvcrt.get_osfhandle(stdout) | 
 | 754 |             else: | 
 | 755 |                 # Assuming file-like object | 
 | 756 |                 c2pwrite = msvcrt.get_osfhandle(stdout.fileno()) | 
 | 757 |             c2pwrite = self._make_inheritable(c2pwrite) | 
 | 758 |  | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 759 |             if stderr is None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 760 |                 errwrite = GetStdHandle(STD_ERROR_HANDLE) | 
| Hirokazu Yamamoto | eacbbdf | 2009-03-03 22:18:14 +0000 | [diff] [blame] | 761 |                 if errwrite is None: | 
 | 762 |                     _, errwrite = CreatePipe(None, 0) | 
 | 763 |             elif stderr == PIPE: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 764 |                 errread, errwrite = CreatePipe(None, 0) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 765 |             elif stderr == STDOUT: | 
 | 766 |                 errwrite = c2pwrite | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 767 |             elif isinstance(stderr, int): | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 768 |                 errwrite = msvcrt.get_osfhandle(stderr) | 
 | 769 |             else: | 
 | 770 |                 # Assuming file-like object | 
 | 771 |                 errwrite = msvcrt.get_osfhandle(stderr.fileno()) | 
 | 772 |             errwrite = self._make_inheritable(errwrite) | 
 | 773 |  | 
 | 774 |             return (p2cread, p2cwrite, | 
 | 775 |                     c2pread, c2pwrite, | 
 | 776 |                     errread, errwrite) | 
 | 777 |  | 
 | 778 |  | 
 | 779 |         def _make_inheritable(self, handle): | 
 | 780 |             """Return a duplicate of handle, which is inheritable""" | 
 | 781 |             return DuplicateHandle(GetCurrentProcess(), handle, | 
 | 782 |                                    GetCurrentProcess(), 0, 1, | 
 | 783 |                                    DUPLICATE_SAME_ACCESS) | 
 | 784 |  | 
 | 785 |  | 
 | 786 |         def _find_w9xpopen(self): | 
 | 787 |             """Find and return absolut path to w9xpopen.exe""" | 
| Tim Peters | e8374a5 | 2004-10-13 03:15:00 +0000 | [diff] [blame] | 788 |             w9xpopen = os.path.join(os.path.dirname(GetModuleFileName(0)), | 
 | 789 |                                     "w9xpopen.exe") | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 790 |             if not os.path.exists(w9xpopen): | 
 | 791 |                 # Eeek - file-not-found - possibly an embedding | 
 | 792 |                 # situation - see if we can locate it in sys.exec_prefix | 
| Tim Peters | e8374a5 | 2004-10-13 03:15:00 +0000 | [diff] [blame] | 793 |                 w9xpopen = os.path.join(os.path.dirname(sys.exec_prefix), | 
 | 794 |                                         "w9xpopen.exe") | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 795 |                 if not os.path.exists(w9xpopen): | 
| Tim Peters | e8374a5 | 2004-10-13 03:15:00 +0000 | [diff] [blame] | 796 |                     raise RuntimeError("Cannot locate w9xpopen.exe, which is " | 
 | 797 |                                        "needed for Popen to work with your " | 
 | 798 |                                        "shell or platform.") | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 799 |             return w9xpopen | 
 | 800 |  | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 801 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 802 |         def _execute_child(self, args, executable, preexec_fn, close_fds, | 
 | 803 |                            cwd, env, universal_newlines, | 
 | 804 |                            startupinfo, creationflags, shell, | 
 | 805 |                            p2cread, p2cwrite, | 
 | 806 |                            c2pread, c2pwrite, | 
 | 807 |                            errread, errwrite): | 
 | 808 |             """Execute program (MS Windows version)""" | 
 | 809 |  | 
| Peter Astrand | c26516b | 2005-02-21 08:13:02 +0000 | [diff] [blame] | 810 |             if not isinstance(args, types.StringTypes): | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 811 |                 args = list2cmdline(args) | 
 | 812 |  | 
| Peter Astrand | c1d6536 | 2004-11-07 14:30:34 +0000 | [diff] [blame] | 813 |             # Process startup details | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 814 |             if startupinfo is None: | 
| Georg Brandl | ad62489 | 2006-06-04 22:15:37 +0000 | [diff] [blame] | 815 |                 startupinfo = STARTUPINFO() | 
 | 816 |             if None not in (p2cread, c2pwrite, errwrite): | 
| Peter Astrand | c1d6536 | 2004-11-07 14:30:34 +0000 | [diff] [blame] | 817 |                 startupinfo.dwFlags |= STARTF_USESTDHANDLES | 
 | 818 |                 startupinfo.hStdInput = p2cread | 
 | 819 |                 startupinfo.hStdOutput = c2pwrite | 
 | 820 |                 startupinfo.hStdError = errwrite | 
 | 821 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 822 |             if shell: | 
| Georg Brandl | ad62489 | 2006-06-04 22:15:37 +0000 | [diff] [blame] | 823 |                 startupinfo.dwFlags |= STARTF_USESHOWWINDOW | 
 | 824 |                 startupinfo.wShowWindow = SW_HIDE | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 825 |                 comspec = os.environ.get("COMSPEC", "cmd.exe") | 
 | 826 |                 args = comspec + " /c " + args | 
| Tim Peters | e8374a5 | 2004-10-13 03:15:00 +0000 | [diff] [blame] | 827 |                 if (GetVersion() >= 0x80000000L or | 
 | 828 |                         os.path.basename(comspec).lower() == "command.com"): | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 829 |                     # Win9x, or using command.com on NT. We need to | 
 | 830 |                     # use the w9xpopen intermediate program. For more | 
 | 831 |                     # information, see KB Q150956 | 
 | 832 |                     # (http://web.archive.org/web/20011105084002/http://support.microsoft.com/support/kb/articles/Q150/9/56.asp) | 
 | 833 |                     w9xpopen = self._find_w9xpopen() | 
 | 834 |                     args = '"%s" %s' % (w9xpopen, args) | 
 | 835 |                     # Not passing CREATE_NEW_CONSOLE has been known to | 
 | 836 |                     # cause random failures on win9x.  Specifically a | 
 | 837 |                     # dialog: "Your program accessed mem currently in | 
 | 838 |                     # use at xxx" and a hopeful warning about the | 
 | 839 |                     # stability of your system.  Cost is Ctrl+C wont | 
 | 840 |                     # kill children. | 
 | 841 |                     creationflags |= CREATE_NEW_CONSOLE | 
 | 842 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 843 |             # Start the process | 
 | 844 |             try: | 
 | 845 |                 hp, ht, pid, tid = CreateProcess(executable, args, | 
| Tim Peters | e8374a5 | 2004-10-13 03:15:00 +0000 | [diff] [blame] | 846 |                                          # no special security | 
 | 847 |                                          None, None, | 
| Peter Astrand | 81a191b | 2007-05-26 22:18:20 +0000 | [diff] [blame] | 848 |                                          int(not close_fds), | 
| Tim Peters | e8374a5 | 2004-10-13 03:15:00 +0000 | [diff] [blame] | 849 |                                          creationflags, | 
 | 850 |                                          env, | 
 | 851 |                                          cwd, | 
 | 852 |                                          startupinfo) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 853 |             except pywintypes.error, e: | 
 | 854 |                 # Translate pywintypes.error to WindowsError, which is | 
 | 855 |                 # a subclass of OSError.  FIXME: We should really | 
 | 856 |                 # translate errno using _sys_errlist (or simliar), but | 
 | 857 |                 # how can this be done from Python? | 
 | 858 |                 raise WindowsError(*e.args) | 
 | 859 |  | 
 | 860 |             # Retain the process handle, but close the thread handle | 
| Martin v. Löwis | 17de8ff | 2006-04-10 15:55:37 +0000 | [diff] [blame] | 861 |             self._child_created = True | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 862 |             self._handle = hp | 
 | 863 |             self.pid = pid | 
 | 864 |             ht.Close() | 
 | 865 |  | 
| Andrew M. Kuchling | 51ee66e | 2004-10-12 16:38:42 +0000 | [diff] [blame] | 866 |             # Child is launched. Close the parent's copy of those pipe | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 867 |             # handles that only the child should have open.  You need | 
 | 868 |             # to make sure that no handles to the write end of the | 
 | 869 |             # output pipe are maintained in this process or else the | 
 | 870 |             # pipe will not close when the child process exits and the | 
 | 871 |             # ReadFile will hang. | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 872 |             if p2cread is not None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 873 |                 p2cread.Close() | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 874 |             if c2pwrite is not None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 875 |                 c2pwrite.Close() | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 876 |             if errwrite is not None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 877 |                 errwrite.Close() | 
 | 878 |  | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 879 |  | 
| Gregory P. Smith | a36f8fe | 2008-08-04 00:13:29 +0000 | [diff] [blame] | 880 |         def _internal_poll(self, _deadstate=None): | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 881 |             """Check if child process has terminated.  Returns returncode | 
 | 882 |             attribute.""" | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 883 |             if self.returncode is None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 884 |                 if WaitForSingleObject(self._handle, 0) == WAIT_OBJECT_0: | 
 | 885 |                     self.returncode = GetExitCodeProcess(self._handle) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 886 |             return self.returncode | 
 | 887 |  | 
 | 888 |  | 
 | 889 |         def wait(self): | 
 | 890 |             """Wait for child process to terminate.  Returns returncode | 
 | 891 |             attribute.""" | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 892 |             if self.returncode is None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 893 |                 obj = WaitForSingleObject(self._handle, INFINITE) | 
 | 894 |                 self.returncode = GetExitCodeProcess(self._handle) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 895 |             return self.returncode | 
 | 896 |  | 
 | 897 |  | 
 | 898 |         def _readerthread(self, fh, buffer): | 
 | 899 |             buffer.append(fh.read()) | 
 | 900 |  | 
 | 901 |  | 
| Peter Astrand | 23109f0 | 2005-03-03 20:28:59 +0000 | [diff] [blame] | 902 |         def _communicate(self, input): | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 903 |             stdout = None # Return | 
 | 904 |             stderr = None # Return | 
 | 905 |  | 
 | 906 |             if self.stdout: | 
 | 907 |                 stdout = [] | 
| Tim Peters | e8374a5 | 2004-10-13 03:15:00 +0000 | [diff] [blame] | 908 |                 stdout_thread = threading.Thread(target=self._readerthread, | 
 | 909 |                                                  args=(self.stdout, stdout)) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 910 |                 stdout_thread.setDaemon(True) | 
 | 911 |                 stdout_thread.start() | 
 | 912 |             if self.stderr: | 
 | 913 |                 stderr = [] | 
| Tim Peters | e8374a5 | 2004-10-13 03:15:00 +0000 | [diff] [blame] | 914 |                 stderr_thread = threading.Thread(target=self._readerthread, | 
 | 915 |                                                  args=(self.stderr, stderr)) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 916 |                 stderr_thread.setDaemon(True) | 
 | 917 |                 stderr_thread.start() | 
 | 918 |  | 
 | 919 |             if self.stdin: | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 920 |                 if input is not None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 921 |                     self.stdin.write(input) | 
 | 922 |                 self.stdin.close() | 
 | 923 |  | 
 | 924 |             if self.stdout: | 
 | 925 |                 stdout_thread.join() | 
 | 926 |             if self.stderr: | 
 | 927 |                 stderr_thread.join() | 
 | 928 |  | 
 | 929 |             # All data exchanged.  Translate lists into strings. | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 930 |             if stdout is not None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 931 |                 stdout = stdout[0] | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 932 |             if stderr is not None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 933 |                 stderr = stderr[0] | 
 | 934 |  | 
 | 935 |             # Translate newlines, if requested.  We cannot let the file | 
 | 936 |             # object do the translation: It is based on stdio, which is | 
 | 937 |             # impossible to combine with select (unless forcing no | 
 | 938 |             # buffering). | 
| Neal Norwitz | a6d01ce | 2006-05-02 06:23:22 +0000 | [diff] [blame] | 939 |             if self.universal_newlines and hasattr(file, 'newlines'): | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 940 |                 if stdout: | 
 | 941 |                     stdout = self._translate_newlines(stdout) | 
 | 942 |                 if stderr: | 
 | 943 |                     stderr = self._translate_newlines(stderr) | 
 | 944 |  | 
 | 945 |             self.wait() | 
 | 946 |             return (stdout, stderr) | 
 | 947 |  | 
| Christian Heimes | e74c8f2 | 2008-04-19 02:23:57 +0000 | [diff] [blame] | 948 |         def send_signal(self, sig): | 
 | 949 |             """Send a signal to the process | 
 | 950 |             """ | 
 | 951 |             if sig == signal.SIGTERM: | 
 | 952 |                 self.terminate() | 
 | 953 |             else: | 
 | 954 |                 raise ValueError("Only SIGTERM is supported on Windows") | 
 | 955 |  | 
 | 956 |         def terminate(self): | 
 | 957 |             """Terminates the process | 
 | 958 |             """ | 
 | 959 |             TerminateProcess(self._handle, 1) | 
 | 960 |  | 
 | 961 |         kill = terminate | 
 | 962 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 963 |     else: | 
 | 964 |         # | 
 | 965 |         # POSIX methods | 
 | 966 |         # | 
 | 967 |         def _get_handles(self, stdin, stdout, stderr): | 
 | 968 |             """Construct and return tupel with IO objects: | 
 | 969 |             p2cread, p2cwrite, c2pread, c2pwrite, errread, errwrite | 
 | 970 |             """ | 
 | 971 |             p2cread, p2cwrite = None, None | 
 | 972 |             c2pread, c2pwrite = None, None | 
 | 973 |             errread, errwrite = None, None | 
 | 974 |  | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 975 |             if stdin is None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 976 |                 pass | 
 | 977 |             elif stdin == PIPE: | 
 | 978 |                 p2cread, p2cwrite = os.pipe() | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 979 |             elif isinstance(stdin, int): | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 980 |                 p2cread = stdin | 
 | 981 |             else: | 
 | 982 |                 # Assuming file-like object | 
 | 983 |                 p2cread = stdin.fileno() | 
 | 984 |  | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 985 |             if stdout is None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 986 |                 pass | 
 | 987 |             elif stdout == PIPE: | 
 | 988 |                 c2pread, c2pwrite = os.pipe() | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 989 |             elif isinstance(stdout, int): | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 990 |                 c2pwrite = stdout | 
 | 991 |             else: | 
 | 992 |                 # Assuming file-like object | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 993 |                 c2pwrite = stdout.fileno() | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 994 |  | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 995 |             if stderr is None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 996 |                 pass | 
 | 997 |             elif stderr == PIPE: | 
 | 998 |                 errread, errwrite = os.pipe() | 
 | 999 |             elif stderr == STDOUT: | 
 | 1000 |                 errwrite = c2pwrite | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 1001 |             elif isinstance(stderr, int): | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1002 |                 errwrite = stderr | 
 | 1003 |             else: | 
 | 1004 |                 # Assuming file-like object | 
 | 1005 |                 errwrite = stderr.fileno() | 
 | 1006 |  | 
 | 1007 |             return (p2cread, p2cwrite, | 
 | 1008 |                     c2pread, c2pwrite, | 
 | 1009 |                     errread, errwrite) | 
 | 1010 |  | 
 | 1011 |  | 
 | 1012 |         def _set_cloexec_flag(self, fd): | 
 | 1013 |             try: | 
 | 1014 |                 cloexec_flag = fcntl.FD_CLOEXEC | 
 | 1015 |             except AttributeError: | 
 | 1016 |                 cloexec_flag = 1 | 
 | 1017 |  | 
 | 1018 |             old = fcntl.fcntl(fd, fcntl.F_GETFD) | 
 | 1019 |             fcntl.fcntl(fd, fcntl.F_SETFD, old | cloexec_flag) | 
 | 1020 |  | 
 | 1021 |  | 
 | 1022 |         def _close_fds(self, but): | 
| Georg Brandl | 309501a | 2008-01-19 20:22:13 +0000 | [diff] [blame] | 1023 |             os.closerange(3, but) | 
 | 1024 |             os.closerange(but + 1, MAXFD) | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 1025 |  | 
 | 1026 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1027 |         def _execute_child(self, args, executable, preexec_fn, close_fds, | 
 | 1028 |                            cwd, env, universal_newlines, | 
 | 1029 |                            startupinfo, creationflags, shell, | 
 | 1030 |                            p2cread, p2cwrite, | 
 | 1031 |                            c2pread, c2pwrite, | 
 | 1032 |                            errread, errwrite): | 
 | 1033 |             """Execute program (POSIX version)""" | 
 | 1034 |  | 
| Peter Astrand | c26516b | 2005-02-21 08:13:02 +0000 | [diff] [blame] | 1035 |             if isinstance(args, types.StringTypes): | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1036 |                 args = [args] | 
| Georg Brandl | 6c0e1e8 | 2006-10-29 09:05:04 +0000 | [diff] [blame] | 1037 |             else: | 
 | 1038 |                 args = list(args) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1039 |  | 
 | 1040 |             if shell: | 
 | 1041 |                 args = ["/bin/sh", "-c"] + args | 
 | 1042 |  | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 1043 |             if executable is None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1044 |                 executable = args[0] | 
 | 1045 |  | 
 | 1046 |             # For transferring possible exec failure from child to parent | 
 | 1047 |             # The first char specifies the exception type: 0 means | 
 | 1048 |             # OSError, 1 means some other error. | 
 | 1049 |             errpipe_read, errpipe_write = os.pipe() | 
 | 1050 |             self._set_cloexec_flag(errpipe_write) | 
 | 1051 |  | 
| Gregory P. Smith | 87d4979 | 2008-01-19 20:57:59 +0000 | [diff] [blame] | 1052 |             gc_was_enabled = gc.isenabled() | 
 | 1053 |             # Disable gc to avoid bug where gc -> file_dealloc -> | 
 | 1054 |             # write to stderr -> hang.  http://bugs.python.org/issue1336 | 
 | 1055 |             gc.disable() | 
 | 1056 |             try: | 
 | 1057 |                 self.pid = os.fork() | 
 | 1058 |             except: | 
 | 1059 |                 if gc_was_enabled: | 
 | 1060 |                     gc.enable() | 
 | 1061 |                 raise | 
| Gregory P. Smith | 92ffc63 | 2008-01-19 22:23:56 +0000 | [diff] [blame] | 1062 |             self._child_created = True | 
 | 1063 |             if self.pid == 0: | 
 | 1064 |                 # Child | 
 | 1065 |                 try: | 
 | 1066 |                     # Close parent's pipe ends | 
 | 1067 |                     if p2cwrite is not None: | 
 | 1068 |                         os.close(p2cwrite) | 
 | 1069 |                     if c2pread is not None: | 
 | 1070 |                         os.close(c2pread) | 
 | 1071 |                     if errread is not None: | 
 | 1072 |                         os.close(errread) | 
 | 1073 |                     os.close(errpipe_read) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1074 |  | 
| Gregory P. Smith | 92ffc63 | 2008-01-19 22:23:56 +0000 | [diff] [blame] | 1075 |                     # Dup fds for child | 
 | 1076 |                     if p2cread is not None: | 
 | 1077 |                         os.dup2(p2cread, 0) | 
 | 1078 |                     if c2pwrite is not None: | 
 | 1079 |                         os.dup2(c2pwrite, 1) | 
 | 1080 |                     if errwrite is not None: | 
 | 1081 |                         os.dup2(errwrite, 2) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1082 |  | 
| Gregory P. Smith | 92ffc63 | 2008-01-19 22:23:56 +0000 | [diff] [blame] | 1083 |                     # Close pipe fds.  Make sure we don't close the same | 
 | 1084 |                     # fd more than once, or standard fds. | 
 | 1085 |                     if p2cread is not None and p2cread not in (0,): | 
 | 1086 |                         os.close(p2cread) | 
 | 1087 |                     if c2pwrite is not None and c2pwrite not in (p2cread, 1): | 
 | 1088 |                         os.close(c2pwrite) | 
 | 1089 |                     if errwrite is not None and errwrite not in (p2cread, c2pwrite, 2): | 
 | 1090 |                         os.close(errwrite) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1091 |  | 
| Gregory P. Smith | 92ffc63 | 2008-01-19 22:23:56 +0000 | [diff] [blame] | 1092 |                     # Close all other fds, if asked for | 
 | 1093 |                     if close_fds: | 
 | 1094 |                         self._close_fds(but=errpipe_write) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1095 |  | 
| Gregory P. Smith | 92ffc63 | 2008-01-19 22:23:56 +0000 | [diff] [blame] | 1096 |                     if cwd is not None: | 
 | 1097 |                         os.chdir(cwd) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1098 |  | 
| Gregory P. Smith | 92ffc63 | 2008-01-19 22:23:56 +0000 | [diff] [blame] | 1099 |                     if preexec_fn: | 
| Brett Cannon | 03446c4 | 2008-08-08 04:19:32 +0000 | [diff] [blame] | 1100 |                         preexec_fn() | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1101 |  | 
| Gregory P. Smith | 92ffc63 | 2008-01-19 22:23:56 +0000 | [diff] [blame] | 1102 |                     if env is None: | 
 | 1103 |                         os.execvp(executable, args) | 
 | 1104 |                     else: | 
 | 1105 |                         os.execvpe(executable, args, env) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1106 |  | 
| Gregory P. Smith | 92ffc63 | 2008-01-19 22:23:56 +0000 | [diff] [blame] | 1107 |                 except: | 
 | 1108 |                     exc_type, exc_value, tb = sys.exc_info() | 
 | 1109 |                     # Save the traceback and attach it to the exception object | 
 | 1110 |                     exc_lines = traceback.format_exception(exc_type, | 
 | 1111 |                                                            exc_value, | 
 | 1112 |                                                            tb) | 
 | 1113 |                     exc_value.child_traceback = ''.join(exc_lines) | 
 | 1114 |                     os.write(errpipe_write, pickle.dumps(exc_value)) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1115 |  | 
| Gregory P. Smith | 92ffc63 | 2008-01-19 22:23:56 +0000 | [diff] [blame] | 1116 |                 # This exitcode won't be reported to applications, so it | 
 | 1117 |                 # really doesn't matter what we return. | 
 | 1118 |                 os._exit(255) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1119 |  | 
 | 1120 |             # Parent | 
| Gregory P. Smith | 87d4979 | 2008-01-19 20:57:59 +0000 | [diff] [blame] | 1121 |             if gc_was_enabled: | 
 | 1122 |                 gc.enable() | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1123 |             os.close(errpipe_write) | 
| Peter Astrand | f540003 | 2007-02-02 19:06:36 +0000 | [diff] [blame] | 1124 |             if p2cread is not None and p2cwrite is not None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1125 |                 os.close(p2cread) | 
| Peter Astrand | f540003 | 2007-02-02 19:06:36 +0000 | [diff] [blame] | 1126 |             if c2pwrite is not None and c2pread is not None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1127 |                 os.close(c2pwrite) | 
| Peter Astrand | f540003 | 2007-02-02 19:06:36 +0000 | [diff] [blame] | 1128 |             if errwrite is not None and errread is not None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1129 |                 os.close(errwrite) | 
 | 1130 |  | 
 | 1131 |             # Wait for exec to fail or succeed; possibly raising exception | 
 | 1132 |             data = os.read(errpipe_read, 1048576) # Exceptions limited to 1 MB | 
 | 1133 |             os.close(errpipe_read) | 
 | 1134 |             if data != "": | 
| Peter Astrand | f791d7a | 2005-01-01 09:38:57 +0000 | [diff] [blame] | 1135 |                 os.waitpid(self.pid, 0) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1136 |                 child_exception = pickle.loads(data) | 
| Georg Brandl | f3715d2 | 2009-02-14 17:01:36 +0000 | [diff] [blame] | 1137 |                 for fd in (p2cwrite, c2pread, errread): | 
 | 1138 |                     if fd is not None: | 
 | 1139 |                         os.close(fd) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1140 |                 raise child_exception | 
 | 1141 |  | 
 | 1142 |  | 
 | 1143 |         def _handle_exitstatus(self, sts): | 
 | 1144 |             if os.WIFSIGNALED(sts): | 
 | 1145 |                 self.returncode = -os.WTERMSIG(sts) | 
 | 1146 |             elif os.WIFEXITED(sts): | 
 | 1147 |                 self.returncode = os.WEXITSTATUS(sts) | 
 | 1148 |             else: | 
 | 1149 |                 # Should never happen | 
 | 1150 |                 raise RuntimeError("Unknown child exit status!") | 
 | 1151 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1152 |  | 
| Gregory P. Smith | a36f8fe | 2008-08-04 00:13:29 +0000 | [diff] [blame] | 1153 |         def _internal_poll(self, _deadstate=None): | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1154 |             """Check if child process has terminated.  Returns returncode | 
 | 1155 |             attribute.""" | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 1156 |             if self.returncode is None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1157 |                 try: | 
 | 1158 |                     pid, sts = os.waitpid(self.pid, os.WNOHANG) | 
 | 1159 |                     if pid == self.pid: | 
 | 1160 |                         self._handle_exitstatus(sts) | 
 | 1161 |                 except os.error: | 
| Martin v. Löwis | 17de8ff | 2006-04-10 15:55:37 +0000 | [diff] [blame] | 1162 |                     if _deadstate is not None: | 
 | 1163 |                         self.returncode = _deadstate | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1164 |             return self.returncode | 
 | 1165 |  | 
 | 1166 |  | 
 | 1167 |         def wait(self): | 
 | 1168 |             """Wait for child process to terminate.  Returns returncode | 
 | 1169 |             attribute.""" | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 1170 |             if self.returncode is None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1171 |                 pid, sts = os.waitpid(self.pid, 0) | 
 | 1172 |                 self._handle_exitstatus(sts) | 
 | 1173 |             return self.returncode | 
 | 1174 |  | 
 | 1175 |  | 
| Peter Astrand | 23109f0 | 2005-03-03 20:28:59 +0000 | [diff] [blame] | 1176 |         def _communicate(self, input): | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1177 |             read_set = [] | 
 | 1178 |             write_set = [] | 
 | 1179 |             stdout = None # Return | 
 | 1180 |             stderr = None # Return | 
 | 1181 |  | 
 | 1182 |             if self.stdin: | 
 | 1183 |                 # Flush stdio buffer.  This might block, if the user has | 
 | 1184 |                 # been writing to .stdin in an uncontrolled fashion. | 
 | 1185 |                 self.stdin.flush() | 
 | 1186 |                 if input: | 
 | 1187 |                     write_set.append(self.stdin) | 
 | 1188 |                 else: | 
 | 1189 |                     self.stdin.close() | 
 | 1190 |             if self.stdout: | 
 | 1191 |                 read_set.append(self.stdout) | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 1192 |                 stdout = [] | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1193 |             if self.stderr: | 
 | 1194 |                 read_set.append(self.stderr) | 
 | 1195 |                 stderr = [] | 
 | 1196 |  | 
| Peter Astrand | 1812f8c | 2007-01-07 14:34:16 +0000 | [diff] [blame] | 1197 |             input_offset = 0 | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1198 |             while read_set or write_set: | 
| Gregory P. Smith | f414064 | 2008-07-06 07:16:40 +0000 | [diff] [blame] | 1199 |                 try: | 
 | 1200 |                     rlist, wlist, xlist = select.select(read_set, write_set, []) | 
 | 1201 |                 except select.error, e: | 
 | 1202 |                     if e.args[0] == errno.EINTR: | 
 | 1203 |                         continue | 
 | 1204 |                     raise | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1205 |  | 
 | 1206 |                 if self.stdin in wlist: | 
 | 1207 |                     # When select has indicated that the file is writable, | 
 | 1208 |                     # we can write up to PIPE_BUF bytes without risk | 
 | 1209 |                     # blocking.  POSIX defines PIPE_BUF >= 512 | 
| Brett Cannon | 03446c4 | 2008-08-08 04:19:32 +0000 | [diff] [blame] | 1210 |                     chunk = input[input_offset : input_offset + 512] | 
 | 1211 |                     bytes_written = os.write(self.stdin.fileno(), chunk) | 
| Tim Peters | f733abb | 2007-01-30 03:03:46 +0000 | [diff] [blame] | 1212 |                     input_offset += bytes_written | 
| Peter Astrand | 1812f8c | 2007-01-07 14:34:16 +0000 | [diff] [blame] | 1213 |                     if input_offset >= len(input): | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1214 |                         self.stdin.close() | 
 | 1215 |                         write_set.remove(self.stdin) | 
 | 1216 |  | 
 | 1217 |                 if self.stdout in rlist: | 
 | 1218 |                     data = os.read(self.stdout.fileno(), 1024) | 
 | 1219 |                     if data == "": | 
 | 1220 |                         self.stdout.close() | 
 | 1221 |                         read_set.remove(self.stdout) | 
 | 1222 |                     stdout.append(data) | 
 | 1223 |  | 
 | 1224 |                 if self.stderr in rlist: | 
 | 1225 |                     data = os.read(self.stderr.fileno(), 1024) | 
 | 1226 |                     if data == "": | 
 | 1227 |                         self.stderr.close() | 
 | 1228 |                         read_set.remove(self.stderr) | 
 | 1229 |                     stderr.append(data) | 
 | 1230 |  | 
 | 1231 |             # All data exchanged.  Translate lists into strings. | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 1232 |             if stdout is not None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1233 |                 stdout = ''.join(stdout) | 
| Peter Astrand | d38ddf4 | 2005-02-10 08:32:50 +0000 | [diff] [blame] | 1234 |             if stderr is not None: | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1235 |                 stderr = ''.join(stderr) | 
 | 1236 |  | 
 | 1237 |             # Translate newlines, if requested.  We cannot let the file | 
 | 1238 |             # object do the translation: It is based on stdio, which is | 
 | 1239 |             # impossible to combine with select (unless forcing no | 
 | 1240 |             # buffering). | 
| Neal Norwitz | a6d01ce | 2006-05-02 06:23:22 +0000 | [diff] [blame] | 1241 |             if self.universal_newlines and hasattr(file, 'newlines'): | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1242 |                 if stdout: | 
 | 1243 |                     stdout = self._translate_newlines(stdout) | 
 | 1244 |                 if stderr: | 
 | 1245 |                     stderr = self._translate_newlines(stderr) | 
 | 1246 |  | 
 | 1247 |             self.wait() | 
 | 1248 |             return (stdout, stderr) | 
 | 1249 |  | 
| Christian Heimes | e74c8f2 | 2008-04-19 02:23:57 +0000 | [diff] [blame] | 1250 |         def send_signal(self, sig): | 
 | 1251 |             """Send a signal to the process | 
 | 1252 |             """ | 
 | 1253 |             os.kill(self.pid, sig) | 
 | 1254 |  | 
 | 1255 |         def terminate(self): | 
 | 1256 |             """Terminate the process with SIGTERM | 
 | 1257 |             """ | 
 | 1258 |             self.send_signal(signal.SIGTERM) | 
 | 1259 |  | 
 | 1260 |         def kill(self): | 
 | 1261 |             """Kill the process with SIGKILL | 
 | 1262 |             """ | 
 | 1263 |             self.send_signal(signal.SIGKILL) | 
 | 1264 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1265 |  | 
 | 1266 | def _demo_posix(): | 
 | 1267 |     # | 
 | 1268 |     # Example 1: Simple redirection: Get process list | 
 | 1269 |     # | 
 | 1270 |     plist = Popen(["ps"], stdout=PIPE).communicate()[0] | 
 | 1271 |     print "Process list:" | 
 | 1272 |     print plist | 
 | 1273 |  | 
 | 1274 |     # | 
 | 1275 |     # Example 2: Change uid before executing child | 
 | 1276 |     # | 
 | 1277 |     if os.getuid() == 0: | 
 | 1278 |         p = Popen(["id"], preexec_fn=lambda: os.setuid(100)) | 
 | 1279 |         p.wait() | 
 | 1280 |  | 
 | 1281 |     # | 
 | 1282 |     # Example 3: Connecting several subprocesses | 
 | 1283 |     # | 
 | 1284 |     print "Looking for 'hda'..." | 
 | 1285 |     p1 = Popen(["dmesg"], stdout=PIPE) | 
 | 1286 |     p2 = Popen(["grep", "hda"], stdin=p1.stdout, stdout=PIPE) | 
 | 1287 |     print repr(p2.communicate()[0]) | 
 | 1288 |  | 
 | 1289 |     # | 
 | 1290 |     # Example 4: Catch execution error | 
 | 1291 |     # | 
 | 1292 |     print | 
 | 1293 |     print "Trying a weird file..." | 
 | 1294 |     try: | 
 | 1295 |         print Popen(["/this/path/does/not/exist"]).communicate() | 
 | 1296 |     except OSError, e: | 
 | 1297 |         if e.errno == errno.ENOENT: | 
 | 1298 |             print "The file didn't exist.  I thought so..." | 
 | 1299 |             print "Child traceback:" | 
 | 1300 |             print e.child_traceback | 
 | 1301 |         else: | 
 | 1302 |             print "Error", e.errno | 
 | 1303 |     else: | 
 | 1304 |         print >>sys.stderr, "Gosh.  No error." | 
 | 1305 |  | 
 | 1306 |  | 
 | 1307 | def _demo_windows(): | 
 | 1308 |     # | 
 | 1309 |     # Example 1: Connecting several subprocesses | 
 | 1310 |     # | 
 | 1311 |     print "Looking for 'PROMPT' in set output..." | 
 | 1312 |     p1 = Popen("set", stdout=PIPE, shell=True) | 
 | 1313 |     p2 = Popen('find "PROMPT"', stdin=p1.stdout, stdout=PIPE) | 
 | 1314 |     print repr(p2.communicate()[0]) | 
 | 1315 |  | 
 | 1316 |     # | 
 | 1317 |     # Example 2: Simple execution of program | 
 | 1318 |     # | 
 | 1319 |     print "Executing calc..." | 
 | 1320 |     p = Popen("calc") | 
 | 1321 |     p.wait() | 
 | 1322 |  | 
 | 1323 |  | 
 | 1324 | if __name__ == "__main__": | 
 | 1325 |     if mswindows: | 
 | 1326 |         _demo_windows() | 
 | 1327 |     else: | 
 | 1328 |         _demo_posix() |