关闭 Python 守护程序时运行代码

Question

我正在使用著名的code referenced here 或here 在 Python 中做一个守护进程，如下所示：

import sys, daemon

class test(daemon.Daemon):
    def run(self):
        self.db = somedb.connect()            # connect to a DB
        self.blah = 127
        with open('blah0.txt', 'w') as f:
            f.write(self.blah)
        # doing lots of things here, modifying self.blah

    def before_stop(self):
        self.db.close()                      # properly close the DB (sync to disk, etc.)
        with open('blah1.txt', 'w') as f:
            f.write(self.blah)

daemon = test(pidfile='_.pid')

if 'start' == sys.argv[1]: 
    daemon.start()
elif 'stop' == sys.argv[1]: 
    daemon.before_stop()               # AttributeError: test instance has no attribute 'blah'
    daemon.stop()

问题是当调用./myscript.py stop 和daemon.before_stop() 时，不再有对self.blah 的引用！

AttributeError: 测试实例没有属性'blah'

因此，使用这种守护程序方法，在停止守护程序之前无法访问守护程序的变量...

问题：如何访问之前的守护类变量：

• 停在./myscript.py stop

• 使用 SIGTERM 停止

• （被杀？）

---

编辑：已解决，here is a working daemon code with a quit() method.

Answer 1

守护进程代码向守护进程发送一个 SIGTERM 信号，要求它停止。如果您希望某些东西由守护进程本身运行，它必须从信号处理程序或从atexit.register 调用的方法运行。

daemonize 方法已经安装了这样的方法，只需从那里调用beforestop：

# this one could be either in a subclass or in a modified base daemeon class
def delpid(self):
        if hasattr(self, 'before_stop'):
                self.before_stop()
        os.remove(self.pidfile)

# this one should be in subclass
def before_stop(self):
    self.db.close()                      # properly close the DB (sync to disk, etc.)
    with open('blah1.txt', 'w') as f:
        f.write(self.blah)

但这还不够！ Python 标准库文档中提到了atexit：

当程序被 Python 未处理的信号杀死时，不会调用通过此模块注册的函数

由于进程预计会收到 SIGTERM 信号，因此您必须安装一个处理程序。如一个active state recipe所示，非常简单：只要程序收到信号就停止：

...
from signal import signal, SIGTERM
...

atexit.register(self.delpid)
signal(SIGTERM, lambda signum, stack_frame: exit(1))

Answer 2

当您调用before_stop() 时，Daemon 类不会检索以前的对象，它只是通过 PID 找到进程并使用SIGTERM 杀死。如果你需要一些实例属性，要么在 __init__() 中初始化它们，要么将它们写在一些临时文件中，以便稍后在你的 before_stop() 实现中读取它们。

作为一般规则，您应该始终初始化 __init__() 内的所有实例绑定属性。