Python装饰器参数如何获取？

百变鹏仔 5天前 #Python

文章标签参数

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

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

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

1. 在装饰器中存储变量

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

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

import inspectdef 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 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):    ...print(get_decorator_args(foo))  # {'_id': 'aaa', 'params': (1, 2, 3)}

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

文章推荐

Python装饰器参数如何获取？

如何使用Python实现克鲁斯卡尔算法？

如何用Python编写随机森林算法？

如何用Python编写插入排序算法？

如何使用Python实现决策树算法？

使用 Python 查找元组列表中的最小值和最大值