如何安全地终止 python 脚本，使其最终块执行

Question

我已经在 nohup 中启动了一个 python 脚本。 该脚本包含一个 finally 块，如果发生任何异常或键盘中断，它将打印队列中的剩余项目以打印到日志文件中。

只要我在执行脚本的同一 shell 中，它就可以正常工作。 例如：以下步骤按我的预期工作。

# Run the script in the background with nohup
nohup script_name.py > script_name.log 2>&1 &

# List the processes currently running in the background
jobs

# Bring back the process to foreground
fg 1

# Send Keyboard interrupt to stop the process by allowing to execute finally code block
Press CTRL+C

The script prints all the pending items in the queue to the "script_name.log" file.

但问题是，一旦我退出当前的 shell 从我启动“nohup”进程的位置，我就不能再将该进程带回前台发送键盘中断。

同时，如果我使用kill PROCESS_ID命令，它不允许finally块执行，所以我也会丢失队列中的待处理项目。

有什么方法可以终止进程或向 nohup 进程发送键盘中断，仍然允许finally块执行代码？

谢谢你。

更新：高级 python 代码：它实际上是一个 Python 项目，其中包含很多目录和其他脚本，但让我在这里只给出代码的 finally 块部分。

try:
    cc = crawler.MultiThreadedWebCrawler(max_workers)
    cc.run_web_crawler()
finally:
    cc.info(script_start_time)
    
    
    
# Inside another script, the definition of "info" method is,

def info(self, script_start_time):
    print('\n', self.crawl_queue.qsize(), ' URLs in crawl_queue are:\n')
    while self.crawl_queue.qsize() > 0:
        print(self.crawl_queue.qsize(), ' ', self.crawl_queue.get(), '\n')
    print("Script execution started at ", script_start_time)
    print("Script execution ended at ", datetime.now().strftime('%Y-%m-%d__%H_%M_%S'))

Answer 1

使用 txt 文件中的值作为终止开关。

让你的程序在每次完成它的“工作”时打开一个 txt 文件。 检查提取的值是1还是0 。 如果为0 ，则继续。 如果它是1则退出脚本和 output 结果。

KillOrStay = open("/root/KillOrStay.txt").readlines()[0]

if int(KillOrStay) == 1:
    # Output your results here
    exit()
else:
    pass