Python 子进程 - 检查 fork 程序的退出代码

Question

假设我有一个这样的脚本:

p = subprocess.Popen(['python forked_job.py'], shell=True)
status = p.wait()
# Do something with status

然后 forked_job.py 看起来像这样:

import os
import sys

print 'hi'
pid = os.fork()
if pid == 0:
    sys.exit(do_some_work())
else:
    sys.exit(do_other_work())

如何确保两个进程都返回 0 状态代码？

Answer 1

当你 fork 时，你有一个父进程和一个非父进程。当pid == 0时，你处于子进程中；你的 else 语句是当你在父进程中时的。

与调用 Popen.wait 类似，正如您在第一个脚本中所做的那样，您希望在第二个脚本中调用 os.wait。

来自the docs :

os.wait()

Wait for completion of a child process, and return a tuple containing its pid and exit status indication: a 16-bit number, whose low byte is the signal number that killed the process, and whose high byte is the exit status (if the signal number is zero); the high bit of the low byte is set if a core file was produced.

Availability: Unix

正如您所看到的，这当然假设您正在运行 UNIX。由于 os.fork 也仅适用于 Unix，因此这似乎是有可能的。

因此，让父级调用 os.wait 并在父级返回的内容中反射(reflect)状态。

需要注意一件事，尽管这可能并不重要，而且您可能已经意识到了。从技术上讲，您没有这样做:

    main_script
     /       \
forked_job   forked_job

但是:

        main_script
            |
      forked_job_parent
            |
       forked_job_child

(我试图显示“所有权”，以及第二次等待的用法。)