From bce3c2a1cacb5393c51773f3ddc45f1b206c2b6b Mon Sep 17 00:00:00 2001 From: zy123 <646228430@qq.com> Date: Sun, 16 Feb 2025 18:14:11 +0800 Subject: [PATCH] =?UTF-8?q?2.16=20=E5=B0=9D=E8=AF=95=E8=A7=A3=E5=86=B3?= =?UTF-8?q?=E5=86=85=E5=AD=98=E6=B3=84=E6=BC=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- flask_app/start_up.py | 29 +++++++++++++---------------- 1 file changed, 13 insertions(+), 16 deletions(-) diff --git a/flask_app/start_up.py b/flask_app/start_up.py index 304b75d..15593a4 100644 --- a/flask_app/start_up.py +++ b/flask_app/start_up.py @@ -24,28 +24,25 @@ def create_app(): app.register_blueprint(test_zbparse_bp) app.register_blueprint(judge_zbfile_bp) app.register_blueprint(test_process_bp) - @app.teardown_request - def close_logger(exception=None): + def teardown_request(exception): + # 接口请求之后都会执行该代码，做一些清理工作 logger = getattr(g, 'logger', None) if logger: for handler in logger.handlers[:]: handler.close() logger.removeHandler(handler) - # @app.teardown_request - # def teardown_request(exception): - # # 接口请求之后都会执行该代码，做一些清理工作 - # output_folder = getattr(g, 'output_folder', None) - # if output_folder: - # # 执行与output_folder相关的清理操作（例如删除临时文件） - # logger = g.logger # 使用 app 的 logger - # logger.info(f"正在清理输出文件夹: {output_folder}") - # file_ids = read_file_ids(output_folder) - # failed_file_ids = delete_file_by_ids(file_ids) - # if failed_file_ids: - # logger.error(f"以下文件删除失败: {failed_file_ids}") - # else: - # logger.info("清理完毕！") + output_folder = getattr(g, 'output_folder', None) + if output_folder: + # 执行与output_folder相关的清理操作（例如删除临时文件） + logger = g.logger # 使用 app 的 logger + logger.info(f"正在清理输出文件夹: {output_folder}") + file_ids = read_file_ids(output_folder) + failed_file_ids = delete_file_by_ids(file_ids) + if failed_file_ids: + logger.error(f"以下文件删除失败: {failed_file_ids}") + else: + logger.info("清理完毕！") return app