python 子进程自定义返回码

Question

这个问题已经有答案了:

Exit codes in Python (16 个回答)

已关闭 5 年前。

我想从 python 子进程调用中返回一个与返回的错误类型相对应的值。例如:

测试.py

RETCODE_OK = 0
RETCODE_THING_1_FAILED = 1
RETCODE_THING_2_FAILED = 2

def main():
    return RETCODE_THING_2_FAILED

if __name__ == '__main__':
    main()

然后我用子进程调用它，如下所示:

>>> import subprocess
>>> proc = subprocess.Popen('python test.py', shell=True)
>>> proc.communicate()
(None, None)
>>> proc.returncode
0

我希望它返回 main() 中返回的任何内容，在本例中为 2。我在 stdout 和 stderr 流中还有其他内容，所以我不能只是 print RETCODE_THING_2_FAILED并获取标准输出。

Answer 1

进程使用退出代码，而不是返回语句。您应该使用 sys.exit(STATUS) 而不是 return STATUS 语句:

test2.py:
---------
import sys

def main():
    sys.exit(RETCODE_THING_2_FAILED)

if __name__ == '__main__':
    main()

interpreter:
------------
>>> p = subprocess.Popen('python test2.py')
>>> p.communicate()
(None, None)
>>> p.returncode
2

这是因为该进程实际上正在关闭/退出，而不是向另一个函数返回值。