Python 多处理 atexit 错误“Error in atexit._run_exitfuncs”

Question

我正在尝试用 Python 运行一个简单的多进程应用程序。 主线程生成 1 到 N 个进程并等待它们全部完成处理。 每个进程都运行一个无限循环，因此它们可能会永远运行而无需用户中断，因此我放入了一些代码来处理键盘中断：

#!/usr/bin/env python
import sys
import time
from multiprocessing import Process

def main():
    # Set up inputs..

    # Spawn processes
    Proc( 1).start()
    Proc( 2).start()

class Proc ( Process ):
    def __init__ ( self, procNum):
        self.id = procNum
        Process.__init__(self)

    def run ( self ):
        doneWork = False

        while True:

            try:
                # Do work...
                time.sleep(1)
                sys.stdout.write('.')

                if doneWork:
                    print "PROC#" + str(self.id) + " Done."
                    break

            except KeyboardInterrupt:
                print "User aborted."
                sys.exit()

# Main Entry
if __name__=="__main__":
    main()

问题是，当使用 CTRL-C 退出时，我会收到一个额外的错误，即使进程似乎立即退出：

......User aborted.
Error in atexit._run_exitfuncs:
Traceback (most recent call last):
  File "C:\Python26\lib\atexit.py", line 24, in _run_exitfuncs
    func(*targs, **kargs)
  File "C:\Python26\lib\multiprocessing\util.py", line 281, in _exit_function
    p.join()
  File "C:\Python26\lib\multiprocessing\process.py", line 119, in join
    res = self._popen.wait(timeout)
  File "C:\Python26\lib\multiprocessing\forking.py", line 259, in wait
    res = _subprocess.WaitForSingleObject(int(self._handle), msecs)
KeyboardInterrupt
Error in sys.exitfunc:
Traceback (most recent call last):
  File "C:\Python26\lib\atexit.py", line 24, in _run_exitfuncs
    func(*targs, **kargs)
  File "C:\Python26\lib\multiprocessing\util.py", line 281, in _exit_function
    p.join()
  File "C:\Python26\lib\multiprocessing\process.py", line 119, in join
    res = self._popen.wait(timeout)
  File "C:\Python26\lib\multiprocessing\forking.py", line 259, in wait
    res = _subprocess.WaitForSingleObject(int(self._handle), msecs)
KeyboardInterrupt

我正在 Windows 上运行 Python 2.6。 如果有更好的方法在 Python 中处理多处理，请告诉我。

Answer 1

这是一个非常古老的问题，但似乎接受的答案并没有真正解决问题。

主要问题是您还需要在父进程中处理键盘中断。 除此之外，在 while 循环中，您只需要退出循环，无需调用 sys.exit()

我尝试尽可能匹配原问题中的示例。 doneWork 代码在示例中不执行任何操作，因此为了清楚起见，已将其删除。

import sys
import time
from multiprocessing import Process


def main():
    # Set up inputs..

    # Spawn processes
    try:
        processes = [Proc(1), Proc(2)]
        [p.start() for p in processes]
        [p.join() for p in processes]
    except KeyboardInterrupt:
        pass


class Proc(Process):
    def __init__(self, procNum):
        self.id = procNum
        Process.__init__(self)

    def run(self):
        while True:
            try:
                # Do work...
                time.sleep(1)
                sys.stdout.write('.')

            except KeyboardInterrupt:
                print("User aborted.")
                break


# Main Entry
if __name__ == "__main__":
    main()

Answer 2

您想要向线程发送信号以告诉它们停止，而不是仅仅强制 sys.exit()。 研究在 Python 中使用 信号处理程序 和线程。

您可以通过将 while True: 循环更改为 while keep_processing: 来实现此目的，其中 keep_processing 是设置为的某种全局变量键盘中断异常。 但我认为这不是一个好的做法。