14

我正在使用 Flask 并使用 before_request 装饰器将有关请求的信息发送到分析系统。我现在正在尝试创建一个装饰器，以防止在几条路线上发送这些事件。

我遇到的问题是让我的装饰器在 before_request 信号被触发之前被调用。

def exclude_from_analytics(func):

    @wraps(func)
    def wrapped(*args, **kwargs):
        print "Before decorated function"
        return func(*args, exclude_from_analytics=True, **kwargs)

    return wrapped

# ------------------------

@exclude_from_analytics
@app.route('/')
def index():
    return make_response('..')

# ------------------------

@app.before_request
def analytics_view(*args, **kwargs):
    if 'exclude_from_analytics' in kwargs and kwargs['exclude_from_analytics'] is True:
       return
4

2 回答 2

19

您可以使用装饰器在函数上简单地放置一个属性（在下面的示例中，我将_exclude_from_analytics其用作属性）。我使用request.endpoint和app.view_functions的组合找到了视图函数。

如果在端点上找不到该属性，您可以忽略分析。

from flask import Flask, request

app = Flask(__name__)

def exclude_from_analytics(func):
    func._exclude_from_analytics = True
    return func

@app.route('/a')
@exclude_from_analytics
def a():
    return 'a'

@app.route('/b')
def b():
    return 'b'

@app.before_request
def analytics_view(*args, **kwargs):
    # Default this to whatever you'd like.
    run_analytics = True

    # You can handle 404s differently here if you'd like.
    if request.endpoint in app.view_functions:
        view_func = app.view_functions[request.endpoint]
        run_analytics = not hasattr(view_func, '_exclude_from_analytics')

    print 'Should run analytics on {0}: {1}'.format(request.path, run_analytics)

app.run(debug=True)

输出（忽略静态文件...）

Should run analytics on /a: False
127.0.0.1 - - [24/Oct/2013 15:55:15] "GET /a HTTP/1.1" 200 -
Should run analytics on /b: True
127.0.0.1 - - [24/Oct/2013 15:55:18] "GET /b HTTP/1.1" 200 -

我还没有测试过这是否适用于蓝图。此外，包装并返回 NEW 函数的装饰器可能会导致它不起作用，因为该属性可能被隐藏。

于 2013-10-24T19:57:30.040 回答
3

这是@Mark Hildreth's answer的一个变体，它确实包装并返回了一个函数：

from functools import wraps
from flask import Flask, request, g

app = Flask(__name__)

def exclude_from_analytics(*args, **kw):
    def wrapper(endpoint_method):
        endpoint_method._skip_analytics = True

        @wraps(endpoint_method)
        def wrapped(*endpoint_args, **endpoint_kw):
            # This is what I want I want to do. Will not work.
            #g.skip_analytics = getattr(endpoint_method, '_skip_analytics', False)
            return endpoint_method(*endpoint_args, **endpoint_kw)
        return wrapped
    return wrapper

@app.route('/')
def no_skip():
    return 'Skip analytics? %s' % (g.skip_analytics)

@app.route('/skip')
@exclude_from_analytics()
def skip():
    return 'Skip analytics? %s' % (g.skip_analytics)

@app.before_request
def analytics_view(*args, **kwargs):
    if request.endpoint in app.view_functions:
        view_func = app.view_functions[request.endpoint]
        g.skip_analytics = hasattr(view_func, '_skip_analytics')
        print 'Should skip analytics on {0}: {1}'.format(request.path, g.skip_analytics)

app.run(debug=True)

它不能像我预期和希望的那样简单地工作的原因与 Flask 上下文堆栈和应用回调的顺序有关。这是方法调用的时间线（基于删除后的一些调试语句）：

$ python test-flask-app.py
# Application Launched
DECORATOR exclude_from_analytics
DECORATOR wrapper
 * Running on http://127.0.0.1:5000/ (Press CTRL+C to quit)

# REQUEST: /
DECORATOR app.before_request: analytics_view
> Should skip analytics on /: False
ENDPOINT no_skip
127.0.0.1 - - [14/May/2016 16:10:39] "GET / HTTP/1.1" 200 -

# REQUEST: /skip
DECORATOR app.before_request: analytics_view
> Should skip analytics on /skip: True
DECORATOR wrapped
ENDPOINT skip
127.0.0.1 - - [14/May/2016 16:12:46] "GET /skip HTTP/1.1" 200 -

我宁愿g.skip_analytics从wrapped函数内部进行设置。但是因为直到钩子之后才调用它analytics_view @app.before_request，所以我必须按照 Mark 的示例并_skip_analytics在端点方法上设置 attr，该方法加载在我调用的应用程序（而不是request）上下文中，该上下文仅在启动时被调用。

有关更多信息flask.g和应用程序上下文，请参阅此 StackOverflow 答案。

于 2016-05-14T23:39:43.940 回答