在关闭信号上在python脚本中运行代码

Question

我有一个在启动时在后台运行的python脚本。启动方法是run.sh文件中的一个条目，该文件由/etc/rc.local调用。确切的输入将是“ sudo
python /home/pi/run/main.py＆”。该系统是带有微调的树莓派。

该脚本正在运行，到目前为止没有问题。如果将关闭命令发送到系统（通过控制台“ sudo shutdown -h
now”），我还需要该脚本不要立即终止，而是先执行一些代码。那就是我到目前为止所得到的：

#!/usr/bin/env python

import atexit

@atexit.register
def byebye():
    c = "End"
    datei = open("/home/pi/logfile",'a+b')
    datei.write(c + "\n")
    datei.close()

def main():

   while True:
     ...do anything...

main()

现在，它似乎仅在关闭时退出主循环。我是否需要使用其他方法关闭系统，以便将信号传输到我的脚本，或者我可能无法使用“ @atexit”方法？有任何想法吗？

谢谢

Answer 1

shutdown发送不处理的SIGTERM信号atexit。上下文管理器，finally块等也不会。

import signal

signal.getsignal(signal.SIGTERM)
Out[64]: 0 #i.e. nothing

与此相反，例如ctrl-C：

signal.getsignal(signal.SIGINT)
Out[65]: <function signal.default_int_handler> #i.e. something

您可以向其注册byebye函数signal以运行而不是不执行任何操作（这将导致解释器最终被shell杀死）

signal.signal(signal.SIGTERM,byebye)

如果执行上述操作，则需要做两件事：

• 更改的签名byebye以接受signal将传递给它的两个参数。
• 您应该sys.exit()在byebye函数末尾执行类似call的操作，以允许python正常关闭商店。

您也可以将signal 和 进行某种组合atexit：

import sys
signal.signal(signal.SIGTERM, lambda num, frame: sys.exit(0))

它将直接插入您当前的代码。这样可以确保清理操作的原子性（即byebye保证是最后的I / O操作），但会有些笨拙。