fix the file exist check

This commit is contained in:
deflax 2024-01-18 04:56:52 +02:00
parent 9f723cce1c
commit 64386d03e6

View file

@ -311,21 +311,24 @@ def database_route():
@app.route("/video/<file_name>", methods=['GET']) @app.route("/video/<file_name>", methods=['GET'])
def video_route(file_name): 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) abort(404)
return send_file(f"{rec_path}/vod/{file_name}",mimetype='video/mp4') return send_file(reqfile, mimetype='video/mp4')
@app.route("/thumb/<file_name>", methods=['GET']) @app.route("/thumb/<file_name>", methods=['GET'])
def thumb_route(file_name): 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) abort(404)
return send_file(f"{rec_path}/thumb/{file_name}",mimetype='image/png') return send_file(reqfile, mimetype='image/png')
@app.route("/img/<file_name>", methods=['GET']) @app.route("/img/<file_name>", methods=['GET'])
def img_route(file_name): 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) abort(404)
return send_file(f"./img/{file_name}",mimetype='image/png') return send_file(reqfile, mimetype='image/png')
def create_app(): def create_app():
return app return app