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

screenshots_flask.py

@@ -1,35 +1,35 @@
-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__)
+    file_path = SCREENSHOTS_DIR / filename
 
-    PROJECT_ROOT = Path(__file__).parent.resolve()
-    SCREENSHOTS_DIR = PROJECT_ROOT / "screenshots"
+    # 防止路径越界访问
+    try:
+        file_path.resolve().relative_to(SCREENSHOTS_DIR.resolve())
+    except ValueError:
+        abort(403, description=f"禁止访问目录外文件: {file_path.resolve()}")
 
-    @app.route('/image/<path:filename>')
-    def serve_image(filename):
-        file_path = SCREENSHOTS_DIR / filename
+    if not file_path.exists():
+        abort(404, description=f"文件不存在: {file_path.resolve()}")
 
-        # 防止路径越界访问
-        try:
-            file_path.resolve().relative_to(SCREENSHOTS_DIR.resolve())
-        except ValueError:
-            abort(403, description=f"禁止访问目录外文件: {file_path.resolve()}")
+    return send_file(file_path, as_attachment=False)
 
-        if not file_path.exists():
-            abort(404, description=f"文件不存在: {file_path.resolve()}")
-
-        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)