From 64386d03e682b39bb653e947a1b83178e987d637 Mon Sep 17 00:00:00 2001 From: deflax Date: Thu, 18 Jan 2024 04:56:52 +0200 Subject: [PATCH] fix the file exist check --- src/scheduler/scheduler.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/src/scheduler/scheduler.py b/src/scheduler/scheduler.py index 91a1c50..18847a4 100644 --- a/src/scheduler/scheduler.py +++ b/src/scheduler/scheduler.py @@ -311,21 +311,24 @@ def database_route(): @app.route("/video/", methods=['GET']) def video_route(file_name): - if not os.path.exists(file_name): + reqfile = f'{rec_path}/thumb/{file_name}' + if not os.path.exists(reqfile): abort(404) - return send_file(f"{rec_path}/vod/{file_name}",mimetype='video/mp4') + return send_file(reqfile, mimetype='video/mp4') @app.route("/thumb/", methods=['GET']) def thumb_route(file_name): - if not os.path.exists(file_name): + reqfile = f'{rec_path}/thumb/{file_name}' + if not os.path.exists(reqfile): abort(404) - return send_file(f"{rec_path}/thumb/{file_name}",mimetype='image/png') + return send_file(reqfile, mimetype='image/png') @app.route("/img/", methods=['GET']) def img_route(file_name): - if not os.path.exists(file_name): + reqfile = f'./img/{file_name}' + if not os.path.exists(reqfile): abort(404) - return send_file(f"./img/{file_name}",mimetype='image/png') + return send_file(reqfile, mimetype='image/png') def create_app(): return app