handle file not found

This commit is contained in:
deflax 2024-01-18 04:52:43 +02:00
parent f75a4417ef
commit 9f723cce1c

View file

@ -5,7 +5,7 @@ import logging
import json import json
import requests import requests
from datetime import datetime 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 flask.helpers import send_file
from apscheduler.schedulers.background import BackgroundScheduler from apscheduler.schedulers.background import BackgroundScheduler
from core_client import Client from core_client import Client
@ -311,14 +311,20 @@ 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):
abort(404)
return send_file(f"{rec_path}/vod/{file_name}",mimetype='video/mp4') return send_file(f"{rec_path}/vod/{file_name}",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):
abort(404)
return send_file(f"{rec_path}/thumb/{file_name}",mimetype='image/png') return send_file(f"{rec_path}/thumb/{file_name}",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):
abort(404)
return send_file(f"./img/{file_name}",mimetype='image/png') return send_file(f"./img/{file_name}",mimetype='image/png')
def create_app(): def create_app():