kali/DailyMotion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 添加自定义404和403错误响应，并优化图像服务接口逻辑

Browse Source
This commit is contained in:
晓丰 2025-06-06 23:00:23 +08:00
parent 101a819663
commit 12c66a9eb4
1 changed files with 19 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
screenshots_flask.py
View File

@ -1,22 +1,13 @@
from flask import Flask, send_file, abort
from flask import Flask, send_file, abort, request, jsonify
from pathlib import Path
app = Flask(__name__)
# 固定 screenshots 目录
PROJECT_ROOT = Path(__file__).parent.resolve()
SCREENSHOTS_DIR = PROJECT_ROOT / "screenshots"
@app.route('/image/<path:filename>')
def serve_image(filename):
app = Flask(__name__)
PROJECT_ROOT = Path(__file__).parent.resolve()
SCREENSHOTS_DIR = PROJECT_ROOT / "screenshots"
@app.route('/image/<path:filename>')
def serve_image(filename):
file_path = SCREENSHOTS_DIR / filename
# 防止路径越界访问
@ -30,6 +21,15 @@ def serve_image(filename):
return send_file(file_path, as_attachment=False)
# 自定义 404 错误响应
@app.errorhandler(404)
def handle_404(e):
return f"404 错误：{e.description}", 404
# 自定义 403 错误响应
@app.errorhandler(403)
def handle_403(e):
return f"403 错误：{e.description}", 403
if __name__ == '__main__':
app.run(host='0.0.0.0', debug=True, port=5000)
app.run(host='0.0.0.0', debug=False, port=5000)