feat: 添加自定义404和403错误响应,并优化图像服务接口逻辑
This commit is contained in:
parent
101a819663
commit
12c66a9eb4
@ -1,17 +1,8 @@
|
||||
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"
|
||||
|
||||
@ -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)
|
Loading…
x
Reference in New Issue
Block a user