Python / Windows / ctypes:如何在调用WaitForMultipleObjects后获取进程返回状态?

时间:2012-04-25 08:19:47

标签: python windows process-management

我有与this guy相同的用例,我非常喜欢this answer,除了我有一个额外的要求:我需要从子进程获得返回状态。

这是我尝试修改他的程序。我是Windows-land和Python ctypes的游客,所以希望我没有做任何愚蠢的事......

import ctypes, subprocess
from random import randint
import os
SYNCHRONIZE=0x00100000
PROCESS_QUERY_INFORMATION=0x0400
INFINITE = -1
numprocs = 5
handles = {}

class Err(BaseException): pass

for i in xrange(numprocs):
    sleeptime = randint(5,10)
    p = subprocess.Popen([r"c:\MinGW\msys\1.0\bin\sleep.exe", str(sleeptime)], stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=False)
    h = ctypes.windll.kernel32.OpenProcess(SYNCHRONIZE | PROCESS_QUERY_INFORMATION, False, p.pid)
    handles[h] = p.pid
    print "Spawned Process %d" % p.pid

while len(handles) > 0:
    print "Waiting for %d children..." % len(handles)
    arrtype = ctypes.c_long * len(handles)
    handle_array = arrtype(*handles.keys())
    ret = ctypes.windll.kernel32.WaitForMultipleObjects(len(handle_array), handle_array, False, INFINITE)
    h = handle_array[ret]
    print "Process %d done" % handles[h]
    i = ctypes.c_int(0)
    pi = ctypes.pointer(i)
    if ctypes.windll.kernel32.GetExitCodeProcess(h, pi) == 0:
    err = ctypes.windll.kernel32.GetLastError()
        raise Err("GetExitCodeProcess: %d" % (err))
    print "Status code is: %d" % (i)
    if ctypes.windll.kernel32.CloseHandle(h) == 0:
    err = ctypes.windll.kernel32.GetLastError()
    raise Err("CloseHandle: %d" % (err))
    del handles[h]
print "All done!

但是当我跑这个时,我失败了:

Traceback (most recent call last):
  File "test.py", line 30, in <module>
    raise Err("GetExitCodeProcess: %d" % (err))
__main__.Err: GetExitCodeProcess: 6

似乎是error 6 is ERROR_INVALID_HANDLE。但我不确定为什么句柄无效;我要求PROCESS_QUERY_INFORMATION权限,如果我发表CloseHandle()来电,GetExitCodeProcess()来电可以正常工作。

有什么想法吗?如何在等待一堆进程后获取状态代码?

1 个答案:

答案 0 :(得分:1)

使用这个确切的Python代码(与你的相同,只修复了一些缩进和删除sleep.exe的路径):

import ctypes, subprocess
from random import randint
import os
SYNCHRONIZE=0x00100000
PROCESS_QUERY_INFORMATION=0x0400
INFINITE = -1
numprocs = 5
handles = {}

class Err(BaseException): pass

for i in xrange(numprocs):
    sleeptime = randint(5,10)
    p = subprocess.Popen([r"sleep.exe", str(sleeptime)], stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=False)
    h = ctypes.windll.kernel32.OpenProcess(SYNCHRONIZE | PROCESS_QUERY_INFORMATION, False, p.pid)
    handles[h] = p.pid
    print "Spawned Process %d" % p.pid

while len(handles) > 0:
    print "Waiting for %d children..." % len(handles)
    arrtype = ctypes.c_long * len(handles)
    handle_array = arrtype(*handles.keys())
    ret = ctypes.windll.kernel32.WaitForMultipleObjects(len(handle_array), handle_array, False, INFINITE)
    h = handle_array[ret]
    print "Process %d done" % handles[h]
    i = ctypes.c_int(0)
    pi = ctypes.pointer(i)
    if ctypes.windll.kernel32.GetExitCodeProcess(h, pi) == 0:
        err = ctypes.windll.kernel32.GetLastError()
        raise Err("GetExitCodeProcess: %d" % (err))
    print "Status code is: %d" % i.value
    if ctypes.windll.kernel32.CloseHandle(h) == 0:
        err = ctypes.windll.kernel32.GetLastError()
        raise Err("CloseHandle: %d" % (err))
    del handles[h]

print "All done!"

给了我这个输出:

C:\tmp\pyt>python -V
Python 2.7.3

C:\tmp\pyt>python a.py
Spawned Process 4368
Spawned Process 268
Spawned Process 5792
Spawned Process 4744
Spawned Process 4484
Waiting for 5 children...
Process 5792 done
Status code is: 0
Waiting for 4 children...
Process 4484 done
Status code is: 0
Waiting for 3 children...
Process 268 done
Status code is: 0
Waiting for 2 children...
Process 4368 done
Status code is: 0
Waiting for 1 children...
Process 4744 done
Status code is: 0
All done!

所以它似乎正在发挥作用。完全不是你的问题吗?您使用的操作系统(我的是Vista)?你有最新版本的Python(2.7.3)吗?你的“sleep.exe”不是以某种方式破坏了吗?你有没有尝试过另一个命令?

相关问题