用Python类实现装饰器

用Python类实现装饰器

为什么使用类实现装饰器?

在某些情况下,我们可能希望装饰器保存状态、初始化一些参数,或者通过对象的方法操作。这时,用类实现装饰器会显得更为灵活和合理。类装饰器允许我们:

  • 通过类的属性来存储状态。
  • 利用类的构造函数传递参数。
  • 通过类方法提供复杂的行为逻辑。

类装饰器的实现

python 复制代码
class MyDecorator:
    def __init__(self, func):
        self.func = func

    def __call__(self, *args, **kwargs):
        print("Function is being called from class-based decorator")
        return self.func(*args, **kwargs)

@MyDecorator
def say_hello():
    print("Hello, world!")

say_hello()

在这个例子中,MyDecorator 类的 init 方法接收被装饰的函数,并将其保存为类的一个属性。call 方法则使该类的实例可以像函数一样被调用,这就是为什么我们可以像函数装饰器一样使用 @MyDecorator 来装饰 say_hello 函数。

增加状态的类装饰器

通过类装饰器,我们可以轻松地维护状态。假设我们想跟踪某个函数被调用的次数,可以通过以下方式实现:

python 复制代码
class CallCounter:
    def __init__(self, func):
        self.func = func
        self.count = 0

    def __call__(self, *args, **kwargs):
        self.count += 1
        print(f"Function has been called {self.count} times")
        return self.func(*args, **kwargs)

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

greet("Alice")
greet("Bob")

带参数的类装饰器

类装饰器的另一个强大功能是能够接收参数。为了实现这一点,我们可以通过 init 方法接受更多的参数。假设我们想实现一个带有自定义前缀的日志记录器,可以这样写:

python 复制代码
class Logger:
    def __init__(self, prefix):
        self.prefix = prefix

    def __call__(self, func):
        def wrapper(*args, **kwargs):
            print(f"{self.prefix}: Function {func.__name__} is called")
            return func(*args, **kwargs)
        return wrapper

@Logger("INFO")
def process_data(data):
    print(f"Processing {data}")

process_data("Dataset 1")
相关推荐
浪裡遊3 分钟前
Sass详解:功能特性、常用方法与最佳实践
开发语言·前端·javascript·css·vue.js·rust·sass
真实的菜10 分钟前
JVM类加载系统详解:深入理解Java类的生命周期
java·开发语言·jvm
代码讲故事30 分钟前
多种方法实现golang中实现对http的响应内容生成图片
开发语言·chrome·http·golang·图片·快照·截图
天天爱吃肉821835 分钟前
ZigBee通信技术全解析:从协议栈到底层实现,全方位解读物联网核心无线技术
python·嵌入式硬件·物联网·servlet
虾球xz1 小时前
CppCon 2018 学习:EFFECTIVE REPLACEMENT OF DYNAMIC POLYMORPHISM WITH std::variant
开发语言·c++·学习
Allen_LVyingbo1 小时前
Python常用医疗AI库以及案例解析(2025年版、上)
开发语言·人工智能·python·学习·健康医疗
小哈龙1 小时前
裸仓库 + Git Bash 搭建 本地 Git 服务端与客户端
开发语言·git·bash
智能砖头1 小时前
LangChain 与 LlamaIndex 深度对比与选型指南
人工智能·python
G探险者2 小时前
《如何在 Spring 中实现 MQ 消息的自动重连:监听与发送双通道策略》
java·开发语言·rpc
weixin_437398212 小时前
转Go学习笔记
linux·服务器·开发语言·后端·架构·golang