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

发布日期： 2024-11-19 阅读量： 作者： ADMIN

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

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

为了解析装饰器传入的参数，在使用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 {}

上一篇: wordpress怎么把所有文章分类单独在一个页面显示

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

随便看看