From 9f723cce1c8bc720ccce55b7aac4f7a115b01568 Mon Sep 17 00:00:00 2001 From: deflax Date: Thu, 18 Jan 2024 04:52:43 +0200 Subject: [PATCH] handle file not found --- src/scheduler/scheduler.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/scheduler/scheduler.py b/src/scheduler/scheduler.py index 7988c9a..91a1c50 100644 --- a/src/scheduler/scheduler.py +++ b/src/scheduler/scheduler.py @@ -5,7 +5,7 @@ import logging import json import requests from datetime import datetime -from flask import Flask, render_template, jsonify, request +from flask import Flask, render_template, jsonify, request, abort from flask.helpers import send_file from apscheduler.schedulers.background import BackgroundScheduler from core_client import Client @@ -311,14 +311,20 @@ def database_route(): @app.route("/video/", methods=['GET']) def video_route(file_name): + if not os.path.exists(file_name): + abort(404) return send_file(f"{rec_path}/vod/{file_name}",mimetype='video/mp4') @app.route("/thumb/", methods=['GET']) def thumb_route(file_name): + if not os.path.exists(file_name): + abort(404) return send_file(f"{rec_path}/thumb/{file_name}",mimetype='image/png') @app.route("/img/", methods=['GET']) def img_route(file_name): + if not os.path.exists(file_name): + abort(404) return send_file(f"./img/{file_name}",mimetype='image/png') def create_app():