如何通过inspect模块获取Python装饰器的传入参数?

ID:21587 / 打印

如何通过inspect模块获取python装饰器的传入参数?

通过inspect模块获取python装饰器的传入参数

为了解析装饰器传入的参数,在使用inspect模块时可能无法直接获取。对此,有两种解决方案:

方案一:修改装饰器内部

在装饰器函数中,可以手动存储传入的参数。这样,在需要时就可以通过直接访问这些存储的变量来获取参数。

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

方案二:针对性解析代码

对于特定的代码,可以利用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)}

需要注意的是,这种方法只适用于解析特定的代码,并不具有通用性。

上一篇: asyncio.Queue() 如何实现超时处理?
下一篇: Python中如何动态修改嵌套JSON请求负载的值?

作者:admin @ 24资源网   2025-01-14

本站所有软件、源码、文章均有网友提供,如有侵权联系308410122@qq.com

与本文相关文章

发表评论:

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。