python - 使用 flask 使路由只能在 Debug模式下访问

Question

我有一堆我只想在 Debug模式下访问的路由。是否有装饰器或其他东西允许我这样做，或者我是否必须完全注释/删除代码？示例:

@debug_only
@app.route("/send_data/<data>", methods=["GET", "POST"])
def send_data(data):
    return jsonfy("{'data': data}")

Answer 1

Flask 没有为此提供任何内置装饰器。编写一个装饰器来检查 current_app.debug 并在未处于 Debug模式时返回 404。

from functools import wraps
from flask import current_app, abort

def debug_only(f):
    @wraps(f)
    def wrapped(**kwargs):
        if not current_app.debug:
            abort(404)

        return f(**kwargs)

    return wrapped

@app.route("/debug")
@debug_only
def debug_info():
    ...