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

百变鹏仔 5天前 #Python

文章标签模块

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

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

方案一：修改装饰器内部

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

立即学习“Python免费学习笔记（深入）”；

方案二：针对性解析代码

对于特定的代码，可以利用python的ast模块来解析代码结构，并从中提取装饰器的参数信息。以下是一个示例代码片段：

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

例如，对于以下代码中的装饰器：

@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装饰器的传入参数？

解析Python编程领域中的最佳就业趋势

揭秘Python在推荐系统开发中的重要角色

为何Python编程是当前最佳的就业方向？

Python程序员的就业前景及发展路径

探索Python编程领域中最热门的就业选项