创建一个可以访问参数和函数的装饰器

Question

我正在修改以下装饰器类。是否可以获取func 的参数以及访问该函数？

def arg(*args, **kwargs):
    def _decorator(func):
        # check for funcs args

        func.do_something(args, kwargs)
        return func
    return _decorator

似乎我尝试过的所有东西都意味着您只能访问其中一个。

Answer 1

我不确定访问参数和函数是什么意思。我猜你的意思是传递给函数的参数。

对于您的用例，对于接受参数的装饰器，您需要另一个包装函数，它会针对运行时传递给函数的 kwargs 进行 kwargs 测试。

例子-

>>> def arg(*args, **kwargs):
...     def _decorator(func):
...         def _wrapper(*args1, **kwargs1):
...             # check for funcs args
...             for k,v in kwargs.items():
...                 print("For key - ",k,v == kwargs1.get(k))
...             func(*args1, **kwargs1)
...         return _wrapper
...     return _decorator
...
>>>
>>> @arg(l = 1)
... def b(l):
...     print("In B", l)
...
>>> b(l=1)
For key -  l True
In B 1
>>> b(l=2)
For key -  l False
In B 2

---

编辑：这适用于没有参数的装饰器。

外部函数接收函数作为参数，内部函数接收函数的参数。

一个非常简单的例子-

>>> def a(func):
...     def x(*args, **kwargs):
...             print(args, kwargs)
...             func(*args, **kwargs)
...     return x
...
>>> @a
... def b(y):
...     print("In b", y)
...
>>> b("Hello")
('Hello',) {}
In b Hello

---