在atexit中引用其他模块

时间:2010-04-03 18:11:48

标签: python atexit

我有一个函数负责在程序结束时终止子进程:

class MySingleton:
    def __init__(self):
        import atexit
        atexit.register(self.stop)

    def stop(self):
        os.kill(self.sel_server_pid, signal.SIGTERM)

但是,调用此函数时会收到错误消息:

Traceback (most recent call last):
File "/usr/lib/python2.5/atexit.py", line 24, in _run_exitfuncs
   func(*targs, **kargs)
File "/home/commando/Development/Diploma/streaminatr/stream/selenium_tests.py", line 66, in stop
   os.kill(self.sel_server_pid, signal.SIGTERM)
AttributeError: 'NoneType' object has no attribute 'kill'

看起来ossignal模块在​​调用atexit之前卸载。重新导入它们解决了这个问题,但这种行为对我来说似乎很奇怪 - 这些模块是在我注册处理程序之前导入的,所以为什么在我自己的退出处理程序运行之前它们被卸载了?

1 个答案:

答案 0 :(得分:6)

对于在程序终止时销毁内容的顺序没有强有力的保证,因此最好确保atexit - 已注册的函数是自包含的。例如,在您的情况下:

class MySingleton:
    def __init__(self):
        import atexit
        atexit.register(self.stop)
        self._dokill = os.kill
        self._thesig = signal.SIGTERM

    def stop(self):
        self._dokill(self.sel_server_pid, self._thesig)

这比重新导入模块更好(这可能会导致程序终止速度减慢甚至无休止的循环,尽管“系统提供的”模块(例如os)的风险较小。