python - 关闭 python 脚本之前的操作

Question

你好，我有一个 python 脚本，它在无限循环中运行一些 Action ，但有时我必须关闭脚本并用新版本更新它，在服务器上做一些工作，等等。

问题是，我怎样才能停止脚本并在关闭它之前，脚本在循环后执行一些操作，如关闭 sqlite 连接、关闭与代理的连接等。

我现在做的是 ctr+c 直到它停止(在它完成之前需要几次)然后手动关闭 sqlite 连接等。

Answer 1

您可以捕获信号并执行 sys.exit 以外的操作

 import signal
 import sys

 def signal_handler(signal, frame):
     print 'You pressed Ctrl+C - or killed me with -2'
     #.... Put your logic here .....
     sys.exit(0)

 signal.signal(signal.SIGINT, signal_handler)