如何捕获 ctrl-c 以杀死 Flask python 脚本

Question

我有以下脚本，它只是启动一个为动态创建的网站提供服务的网络服务器。为了获取动态数据，脚本会打开一个文件来读取数据。

我关心的是如何捕获 CTRL-C 命令来杀死 python 脚本，这样我就可以在脚本线程被杀死之前关闭文件。

我尝试了以下几件事，但都没有成功：

from flask import Flask, render_template
import time

# Initialize the Flask application
app = Flask(__name__)

fileNames = {}
fileDesc = {}
for idx in range(1,4):
    fileNames["name{}".format(idx)] = "./name" + str(idx) + ".txt"
    fileDesc["name{}".format(idx)] = open(fileNames["name{}".format(idx)],'r')

try:
    @app.route('/')
    def index():
        # code for reading data from files
        return render_template('index.html', var1 = var1)

    @app.errorhandler(Exception)
    def all_exception_handler(error):
        print("Closing")
        for key, value in fileDesc:
            val.close()
        print("Files closed")


    if __name__ == '__main__':
        app.run(
            host="192.168.0.166",
            port=int("8080"),
            debug=True
        )

except KeyboardInterrupt:
    print("Closing")
    for key, value in fileDesc:
        val.close()
    print("Files closed")

提前致谢。

Answer 1

我在我的项目中遇到同样的事情。对我有用的是使用signal 来捕获 CTRL-C。

import sys
import signal
def handler(signal, frame):
  print('CTRL-C pressed!')
  sys.exit(0)
signal.signal(signal.SIGINT, handler)
signal.pause()

将这段代码放入运行 Flask 应用程序的脚本中时，可以捕获 CTRL-C。到目前为止，您必须使用 CTRL-C 两次，然后执行处理程序。如果我发现新的东西，我会进一步调查并编辑答案。

编辑 1

好的，我做了更多的研究并想出了一些其他的方法，因为上面的方法很简单。

在生产环境中，诸如关闭数据库或文件之类的清理代码是通过 @app.teardown_appcontext 装饰器完成的。见this part of the tutorial。

在使用简单服务器时，您可以通过暴露 werkzeug 关闭功能将其关闭。见this post。

编辑 2

我已经测试了 Werkzeug 关闭功能，它还可以与 teardown_appcontext 功能一起使用。所以我建议使用装饰器编写你的拆卸函数，并编写一个简单的函数来关闭 werkzeug 服务器。这样生产和开发代码是相同的。