登录
首页 >  文章 >  python教程

带状态的Python装饰器实现方法

时间:2026-01-24 13:16:30 284浏览 收藏

欢迎各位小伙伴来到golang学习网,相聚于此都是缘哈哈哈!今天我给大家带来《Python 实现带状态的装饰器方法》,这篇文章主要讲到等等知识,如果你对文章相关的知识非常感兴趣或者正在自学,都可以关注我,我会持续更新相关文章!当然,有什么建议也欢迎在评论留言提出!一起学习!

带状态的装饰器是能保存和访问内部变量的装饰器,常用类或闭包实现:类方式通过__call__和实例属性管理状态,支持多实例隔离与扩展;闭包方式用nonlocal修改外层变量,适合轻量单状态场景;参数化装饰器推荐类实现,如限流器;需用functools.wraps保留原函数元信息,避免全局变量共享状态。

Python 如何实现一个带状态的装饰器?

带状态的装饰器是指在装饰过程中能保存和访问内部变量的装饰器,比如统计调用次数、缓存上次结果、限制调用频率等。核心思路是让装饰器本身是一个类,或返回一个闭包,其中包含可变的状态变量。

用类实现带状态的装饰器

类是最清晰、最易扩展的方式。通过实现 __call__ 方法,让实例可调用;状态(如计数器)作为实例属性保存。

例如,实现一个统计函数调用次数的装饰器:

class CallCounter:
    def __init__(self, func):
        self.func = func
        self.count = 0
<pre class="brush:python;toolbar:false;">def __call__(self, *args, **kwargs):
    self.count += 1
    print(f"{self.func.__name__} 已被调用 {self.count} 次")
    return self.func(*args, **kwargs)

@CallCounter def greet(name): return f"Hello, {name}!"

使用时:

greet("Alice")  # 输出:greet 已被调用 1 次
greet("Bob")    # 输出:greet 已被调用 2 次
print(greet.count)  # 输出:2

用闭包实现带状态的装饰器

闭包方式利用嵌套函数+ nonlocal 声明来修改外层变量。适合轻量、单状态场景。

同样实现调用计数:

def call_counter(func):
    count = 0
    def wrapper(*args, **kwargs):
        nonlocal count
        count += 1
        print(f"{func.__name__} 已被调用 {count} 次")
        return func(*args, **kwargs)
    wrapper.count = lambda: count  # 提供访问接口(可选)
    return wrapper
<p>@call_counter
def say_hi():
return "Hi!"</p>

注意:闭包中的 count 是不可直接赋值修改的(除非用 nonlocal),且无法像类那样自然支持多个独立实例。

支持参数的带状态装饰器(类方式更推荐)

如果需要初始化状态(比如设定最大调用次数),类方式更直观:

class RateLimiter:
    def __init__(self, max_calls=3):
        self.max_calls = max_calls
        self.calls = 0
<pre class="brush:python;toolbar:false;">def __call__(self, func):
    def wrapper(*args, **kwargs):
        if self.calls >= self.max_calls:
            raise RuntimeError(f"超出调用上限 {self.max_calls}")
        self.calls += 1
        print(f"第 {self.calls} 次调用")
        return func(*args, **kwargs)
    return wrapper

使用

@RateLimiter(max_calls=2) def do_work(): return "done"

这种写法中,@RateLimiter(...) 先构造实例,再用该实例装饰函数,状态随实例隔离,不同装饰器互不影响。

注意事项与技巧

  • functools.wraps 包装内层函数(无论是类的 __call__ 还是闭包的 wrapper),避免丢失原函数的 __name____doc__ 等元信息
  • 类装饰器天然支持多实例状态隔离;闭包装饰器若带参数,需再嵌套一层函数,容易混淆作用域
  • 状态变量应避免全局变量——否则所有被装饰函数共享同一状态,通常不是预期行为
  • 若需重置状态(如测试中),类方式可通过实例方法暴露 reset 接口,闭包方式则较难做到

今天关于《带状态的Python装饰器实现方法》的内容就介绍到这里了,是不是学起来一目了然!想要了解更多关于的内容请关注golang学习网公众号!

前往漫画官网入口并下载 ➜
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>