比较包装的函数是否是装饰器的实例

Question

我有decorator1和decorator2函数。我正在使用这些来装饰功能。

@decorator1("some", "args")
@decorator2(1,2)
@decorator1()
def my_func(): print("my func")

当我一遍又一遍地迭代调用包装的函数时，输出变成这样：

对于my_func.__wrapped__：

decorator1
decorator2
decorator1
my func

对于my_func.__wrapped__.__wrapped__()：

decorator2
decorator1
my func

问题是每个包装函数的名称都是my_func。我想检查这个链中的一个函数是否是假设decorator1的实例。我想知道这一点，因为我将使用装饰器的参数。(我已经通过使用__closure__单元格了解它们。

澄清

我决定举个例子来表明我的目的。

@route("/index")
def index(): pass
@route("/settings")
@need_permission("admin")
def settings: pass
@route("/blog")
@need_permission("admin", "user")
def blog(): pass

我可以在其他地方获取所有这些路由函数，并且我想提取哪个需要哪些权限。

这是我的发现：

>>> blog()
route blog
permissions admin user
>>> blog.__closure__[0].cell_contents
('/blog',)
>>> blog.__closure__[1].cell_contents()
permissions admin user
>>> blog.__closure__[0].cell_contents.__closure__[0].cell_contents
('admin', 'user')
>>> blog.__closure__[0].cell_contents.__closure__[1].cell_contents()
>>> 

我只想提取拥有权限的元组。我可以按某些特定顺序应用我的装饰器并轻松提取，或者我需要实现我的DecoratorApplier功能，如@Poolka指出的那样。如果没有办法像第一个选项那样知道，我会遵循第二个选项。

Answer 1

instance of a decorator- 使用函数和常规函数实现的装饰器都只是类型function的函数。

不确定你到底想得到什么。下面的代码是我处理问题的方法。基本上，我将属性primal_type添加到装饰嘉年华中涉及的所有功能中，以存储功能/装饰器的名称。我用另一个名叫DecoratorApplier的装饰器来做这件事。代码似乎执行了与问题中的问题相关的操作。

编辑

添加的澄清并没有使一切都清楚。我想以这种方式混合函数和装饰器逻辑是不好的做法。也许还有另一种选择可以在函数内获取所需信息？无论如何，下面我的原始方法的两个修改版本(oda代表可选的装饰器参数)。

(1) - 带DecoratorApplier

import functools

def decorator_1(*d1_args, **d1_kwargs):
def decorator(func):
@functools.wraps(func)
def wrapped(*args, **kwargs):
print('inside decorator_1', d1_args, d1_kwargs)
return func(*args, **kwargs)
return wrapped
return decorator

def decorator_2(*d2_args, **d2_kwargs):
def decorator(func):
@functools.wraps(func)
def wrapped(*args, **kwargs):
print('inside decorator_2', d2_args, d2_kwargs)
return func(*args, **kwargs)
return wrapped
return decorator

class DecoratorApplier:
def __init__(self, *decorators):
self.decorators = decorators
def __call__(self, func):
func.oda = dict()
for decorator in self.decorators:
func = decorator[0](*decorator[1], **decorator[2])(func)
(
func
.oda
.setdefault(decorator[0].__name__, list())
.extend([decorator[1], decorator[2]])
)
return func

@DecoratorApplier(
(decorator_1, (1, 2), {'x': 10, 'y': 20}),
(decorator_2, tuple(), dict()))
def f_1():
print('inside f_1')
print(f_1.oda)
return

if __name__ == '__main__':
f_1()

(2) - 修改原始装饰器

import functools

def decorator_1(*d1_args, **d1_kwargs):
def decorator(func):
@functools.wraps(func)
def wrapped(*args, **kwargs):
print('inside decorator_1', d1_args, d1_kwargs)
(
kwargs
.setdefault('oda', dict())
.setdefault('decorator_1', list())
.extend([d1_args, d1_kwargs])
)
return func(*args, **kwargs)
return wrapped
return decorator

def decorator_2(*d2_args, **d2_kwargs):
def decorator(func):
@functools.wraps(func)
def wrapped(*args, **kwargs):
print('inside decorator_2', d2_args, d2_kwargs)
(
kwargs
.setdefault('oda', dict())
.setdefault('decorator_2', list())
.extend([d2_args, d2_kwargs])
)
return func(*args, **kwargs)
return wrapped
return decorator

@decorator_1(1, 2, x=10, y=20)
@decorator_2()
def f_1(oda=None):
print('inside f_1')
print('    oda', oda)
return

if __name__ == '__main__':
f_1()