python - 将一些任意值绑定(bind)到 Flask 中的路由

Question

如何优雅地绑定(bind)一些任意值和 flask 路由？假设，我想在我的 session 接口(interface)实现中或在 before_request Hook 中访问这个值。

现在我是这样做的:

@app.route('/foo/<bar>', defaults={'_my_val': True}):
def foo(bar, _my_val):  # this is ugly hack
    pass

然后像这样通过请求对象访问这个值:

class MyRequest(flask.Request):
    def get_my_val(self):
        return (self.url_rule.defaults or {}).get('_my_val', False)

但它看起来像一个 hack。

UPD: 看起来可以通过扩展 werkzeug.routing.Rule 类并将 **kwargs 添加到其构造函数来完成。可以覆盖 Flask 中的 Rule 类吗？

Answer 1

最终我重写了 flask 的 Request 和 Rule 类:

# here app is a Flask current application object
from flask import Request as FlaskRequest
from werkzeug.routing import Rule as FlaskRule

class Request(FlaskRequest):
    def is_foo(self):
        return bool(self.url_rule._foo) if self.url_rule else False

    def get_bar(self):
        return getattr(self.url_rule, '_bar')

class Rule(FlaskRule):
    def __init__(self, *args, **kwargs):
        for param in ('_foo', '_bar'):
            setattr(self, param, kwargs.pop(param, None))
        super().__init__(*args, **kwargs)

# app initialization
app.request_class = Request
app.url_rule_class = Rule

# route example        
@app.route('/path', _foo=True, _bar='baz')
def route():
    pass