python装饰器解析请求参数_如何用参数创建一个装饰器来解析请求？

每次调用函数作用域时，您似乎都想向它注入一些变量。一种方法是将数据临时插入函数的globals作用域(find_videos_in_polygon.__globals__)，然后再进行清理。在def inject_variables(func, _new_values=None):

for k, v in _new_values.items():

func.__globals__[k] = v

def cleanup_variables(func, _default=None, _new_values=None, _old_values=None):

"""

Reset function's global scope with data in `_old_value`.

If a particular key's value is sentinel then it means the

key didn't exist and we can remove it.

"""

for k, v in _old_values.items():

old_value = _old_values[k]

if old_value is _default:

del func.__globals__[k]

else:

func.__globals__[k] = old_value

def parse_request(*names):

def decorator(func):

async def wrapper(*args, **kwargs):

request = args[0]

new_values = {}

current_global_values = {}

sentinel = object()

for var in names:

name = var

if isinstance(var, tuple):

name, value = var

new_values[name] = request.query.get(name, value)

else:

try:

new_values[name] = request.query[name]

except KeyError:

raise UnboundLocalError("local variable '{name}' referenced before assignment".format(

name=name

))

current_global_values[name] = func.__globals__.get(name, sentinel)

inject_variables(func, _new_values=new_values)

val = await func(*args, **kwargs)

cleanup_variables(func, _default=sentinel, _new_values=new_values, _old_values=current_global_values)

return val

return wrapper

return decorator

测试代码：

^{pr2}$

输出：1 2 3 4 10

10000

1 2 3 4 15

10000

...

UnboundLocalError: local variable 'eggs' referenced before assignment

但当然，这样做会降低代码的可读性和调试难度。因此，最好是明确，做一些更容易理解和测试的事情。我建议添加一个helper函数，它可以解析并返回一个dict，其中包含search_in_polygon所需的所有关键字参数。在def parse_request(request, *names):

data = {}

for var in names:

name = var

if isinstance(var, tuple):

name, value = var

data[name] = request.query.get(name, value)

else:

data[name] = request.query[name]

return data

async def func(request):

request_data = parse_request(request, 'foo', 'bar', 'spam', 'eggs', ('missing', '10'))

print(request_data)

# data = await search_in_polygon(**request_data)

演示：>>> Request.query = {'foo': 1, 'bar': 2, 'spam': 3, 'eggs': 4}

>>> loop.run_until_complete(func(Request))

{'foo': 1, 'bar': 2, 'spam': 3, 'eggs': 4, 'missing': '10'}

>>> Request.query = {'foo': 1, 'bar': 2, 'spam': 3, 'eggs': 4, 'missing': 15}

>>> loop.run_until_complete(func(Request))

{'foo': 1, 'bar': 2, 'spam': 3, 'eggs': 4, 'missing': 15}

>>> Request.query = {'foo': 1, 'bar': 2, 'spam': 3}

>>> loop.run_until_complete(func(Request))

-

KeyError: 'eggs'