Python装饰器参数如何获取?

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装饰器参数如何获取?的详细内容,更多请关注【创想鸟】其它相关文章!

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至253000106@qq.com举报,一经查实,本站将立刻删除。

发布者:PHP中文网,转转请注明出处:https://www.chuangxiangniao.com/p/2178036.html

(0)
上一篇 2025年2月25日 13:31:24
下一篇 2025年2月19日 04:40:55

AD推荐 黄金广告位招租... 更多推荐

相关推荐

发表回复

登录后才能评论