通过代码解析获取装饰器参数
在 中，inspect 模块被用于获取有关函数的信息。然而，对于装饰器，inspect 通常无法获取其传入的参数。
为了解决这个问题，可以通过以下方法之一：
1. 在装饰器中存储变量
装饰器可以自行存储传入的参数，例如：
立即学习“”；
import inspect def task(_id, params): def decorator(func): @wraps(func) def wrapper(*args, **kwargs): return func(*args, **kwargs) wrapper._id = _id wrapper.params = params return wrapper return decorator
登录后复制
2. 代码解析
使用 ast 模块解析代码，可以提取装饰器的参数：
import ast def 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 {}
登录后复制
示例：
@task(_id="aaa", params=(1, 2, 3)) def foo(q=1): ... print(get_decorator_args(foo)) # {'_id': 'aaa', 'params': (1, 2, 3)}
登录后复制
需要注意的是，代码解析方法只适用于特定的代码结构，并且可能需要根据具体情况进行调整。
以上就是Python装饰器参数如何获取？的详细内容，更多请关注php中文网其它相关文章！
微信扫一扫打赏
支付宝扫一扫打赏
