python flask如何将动态参数传递给装饰器

时间:2017-07-09 08:50:07

标签: python flask redis python-decorators

我正在使用python flask框架。我写了一个需要参数的装饰器,这个参数是动态的。

我的装饰者如下所示,将获得一个密钥,并使用redis中的密钥获取数据。

def redis_hash_shop_style(key):
    def fn_wrapper(f):
        @wraps(f)
        def decorated_function(*args, **kwargs):
            data = redis_hash(key)
            return data
        return decorated_function
return fn_wrapper

我有一个类使用这个decorater,像这样的代码

class ShopAreaAndStyleListAPI(Resource):
    @redis_hash_shop_style(key='shop_{}_style'.format(g.city.id))
    def get(self):
        # if not found from redis, query from mysql
        pass

如您所见,我的装饰者需要一个名为key的参数,我会像这样传递密钥

@redis_hash_shop_style(key='shop_{}_style'.format(g.city.id)) g.city.id将获得该城市的ID,如果一切正常,关键将是这样的

shop_100_style

但我得到了错误:

class ShopAreaAndStyleListAPI(Resource):
File "xx.py", line 659, in ShopAreaAndStyleListAPI

@redis_hash_shop_style(key='shop_{}_style'.format(g.city.id))

File "/Users/xx/.virtualenvs/yy/lib/python2.7/site-packages/werkzeug/local.py", line 347, in __getattr__
return getattr(self._get_current_object(), name)
File "/Users/xx/.virtualenvs/yy/lib/python2.7/site-packages/werkzeug/local.py", line 306, in _get_current_object
return self.__local()
File "/Users/xx/.virtualenvs/yy/lib/python2.7/site-packages/flask/globals.py", line 44, in _lookup_app_object
raise RuntimeError(_app_ctx_err_msg)
RuntimeError: Working outside of application context.

This typically means that you attempted to use functionality that 
needed to interface with the current application object in a way.  
To solve this set up an application context with app.app_context().  
See the documentation for more information.

我很困惑,在烧瓶中,如何将动态参数传递给装饰器?

感谢。

1 个答案:

答案 0 :(得分:1)

如果我们检查flask application globalflask.g的文档,则会说:

  

要共享仅对一个请求有效的数据从一个函数到另一个函数,全局变量不够好,因为它会在线程环境中中断。 Flask为您提供特殊对象,确保仅对活动请求有效,并为每个请求返回不同的值。

这是通过使用线程本地代理(在flask/globals.py中)来实现的:

g = LocalProxy(partial(_lookup_app_object, 'g'))

我们应该记住的另一件事是Python在"编译"期间执行装饰器的第一次传递。阶段,在任何请求之外,或flask申请。这意味着key参数在您的应用程序启动时(在解析/修饰您的类时)被赋予'shop_{}_style'.format(g.city.id)值,在flask请求上下文之外。

但我们可以通过使用延迟代理轻松延迟对flask.g的访问,该代理仅在使用时通过回调函数获取值。让我们使用与flask捆绑在一起的werkzeug.local.LocalProxy

from werkzeug.local import LocalProxy

class ShopAreaAndStyleListAPI(Resource):
    @redis_hash_shop_style(key=LocalProxy(lambda: 'shop_{}_style'.format(g.city.id)))
    def get(self):
        # if not found from redis, query from mysql
        pass

一般情况下(对于非flask或非 - werkzeug个应用),我们可以使用ProxyTypes包中的类似LazyProxy

与此无关,您还需要修复redis_hash_shop_style装饰器,不仅要从redis获取,还要更新(或创建)过期(或非过时)的值现有的),在适当时调用包裹的f()