通过inspect模块获取装饰器的传入参数
为了解析装饰器传入的参数，在使用inspect模块时可能无法直接获取。对此，有两种解决方案：
方案一：修改装饰器内部
在装饰器函数中，可以手动存储传入的参数。这样，在需要时就可以通过直接访问这些存储的变量来获取参数。
立即学习“”；
方案二：针对性解析代码
对于特定的代码，可以利用python的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): ...
登录后复制
可以通过调用get_decorator_args(foo)函数获取装饰器的参数：
print(get_decorator_args(foo)) # {'_id': 'aaa', 'params': (1, 2, 3)}
登录后复制
需要注意的是，这种方法只适用于解析特定的代码，并不具有通用性。
以上就是如何通过inspect模块获取Python装饰器的传入参数？的详细内容，更多请关注php中文网其它相关文章！
微信扫一扫打赏
支付宝扫一扫打赏
