Hello! 欢迎来到小浪资源网!



Python装饰器参数如何获取?


Python装饰器参数如何获取?

通过代码解析获取装饰器参数

python 中,inspect 模块被用于获取有关函数的信息。然而,对于装饰器,inspect 通常无法获取其传入的参数。

为了解决这个问题,可以通过以下方法之一:

1. 在装饰器中存储变量

装饰器可以自行存储传入的参数,例如:

立即学习Python免费学习笔记(深入)”;

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)}

需要注意的是,代码解析方法只适用于特定的代码结构,并且可能需要根据具体情况进行调整。

相关阅读