我基本上是在尝试跟踪 API 中的事件。有几条具有不同功能的路线。我带来了这样的东西:
# defining a decorator like this:
class tracker():
def __init__(self, data):
self.data = data
def __call__(self, function):
@functools.wraps(function)
async def decorated(*args, **kwargs):
self.save_event(*args, **kwargs)
return await function(*args, **kwargs)
return decorated
def save_event(self, *args, **kwargs):
this_would_be_function_arg = self.data
# do something with the data
pass
# to be used like this:
@tracker(data=function_arg):
def my_func(function_arg):
# something...
pass
是否可以?我知道我可以从 访问kwargs
,但我不会总是知道传递的是什么,而且我不需要每个参数,所以我不想得到一切。
来自inspect
模块的函数签名可以提供帮助。
使用包装函数的保存签名,您可以在每次调用时将参数绑定到它,并按名称查找它们,而不管类型(arg 或 kwarg)。
文档:内省可调用对象
import inspect
import functools
class tracker():
def __init__(self, data):
self.data = data
self.signature = None
def __call__(self, function):
self.signature = inspect.signature(function)
@functools.wraps(function)
def decorated(*args, **kwargs):
bound = self.signature.bind(*args, **kwargs)
bound.apply_defaults()
self.save_event(bound.arguments[self.data])
return function(*args, **kwargs)
return decorated
def save_event(self, value):
print(f"called with {self.data}={value}")
@tracker(data='function_arg')
def my_func(arg1, function_arg=None):
pass
my_func(0,10)
my_func(10, function_arg=20)
my_func(-1)
# output:
# called with function_arg=10
# called with function_arg=20
# called with function_arg=None
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句