本站资源收集于互联网,不提供软件存储服务,每天免费更新优质的软件以及学习资源!

Python装饰器参数解析:如何用inspect或其他方法获取装饰器传入的参数?

网络教程 app 1℃

Python装饰器参数解析如何用inspect或其他方法获取装饰器传入的参数

解析装饰器传入参数

问题:

如何使用 inspect 获取装饰器中传入的参数?

背景:

代码如下,展示了一个自定义装饰器 task,它接受两个参数 _id 和 params。目标是解析 task 装饰器中传入的参数。

import inspectfrom functools import wrapsdef task(_id, params): def decorator(func): @wraps(func) def wrapper(*args, **kwargs):return func(*args, **kwargs) return wrapper return decorator@task(_id="aaa", params=(1,2,3))def foo(q=1): …if __name__ == ‘__main__’: print(inspect.getclosurevars(foo.__wrapped__).nonlocals) # 不生效

解答:

使用 inspect 无法直接获取装饰器中传入的参数。一种变通方法是在装饰器中手动存储变量:

import astdef get_decorator_args(func): source = inspect.getsource(func) tree = ast.parse(source) for node in ast.walk(tree): if isinstance(node, ast.FunctionDef):for decorator in node.decorator_list: if isinstance(decorator, ast.Call) and decorator.func.id == "task": args = {arg.arg: ast.literal_eval(arg.value)for arg in decorator.keywords } return args return {}print(get_decorator_args(foo)) # {‘_id’: ‘aaa’, ‘params’: (1, 2, 3)}

此方法仅适用于特定的代码,不具有通用性。

以上就是Python装饰器参数解析:如何用inspect或其他方法获取装饰器传入的参数?的详细内容,更多请关注范的资源库其它相关文章!

转载请注明:范的资源库 » Python装饰器参数解析:如何用inspect或其他方法获取装饰器传入的参数?

喜欢 (0)