将装饰器设计为类:Python实现实例化与灵活应用

需积分: 32 108 下载量 58 浏览量 更新于2024-08-08 收藏 5.68MB PDF 举报
本篇文章主要讨论了如何在Python中将装饰器定义为类,以便更好地适应在函数内部和外部使用的需求。装饰器作为Python编程中的一种高级特性,通常用于修改或增强函数的行为。这里介绍了一种名为`Profiled`的装饰器类,它实现了`__call__()`和`__get__()`方法,这两个方法对于确保装饰器实例化并正确地应用到目标函数或类的方法上至关重要。 `__call__()`方法是在调用装饰器时被自动调用,它负责计数被装饰函数的调用次数,并调用原始函数(`__wrapped__(*args, **kwargs)`)来执行实际的功能。而`__get__()`方法则用于处理装饰器作为类属性的情况,当在类的上下文中访问装饰器时,会返回一个方法类型实例,这样可以在实例化该类时自动应用装饰器。 以下是一个具体的例子: ```python import types from functools import wraps class Profiled: def __init__(self, func): wraps(func)(self) # 保存原始函数的元信息 self.ncalls = 0 def __call__(self, *args, **kwargs): self.ncalls += 1 return self.__wrapped__(*args, **kwargs) def __get__(self, instance, cls): if instance is None: # 如果装饰器用于类定义而非方法 return self else: return types.MethodType(self, instance) # 为类方法提供实例化的装饰器 # 使用方式: @Profiled def add(x, y): return x + y class Spam: @Profiled def bar(self, x): ... ``` 在这个例子中,`@Profiled`可以用于函数`add`或类`Spam`的方法`bar`,无论是在类定义内部还是外部使用,装饰器都能正确地工作并记录函数调用次数。 Python Cookbook是一个广泛使用的指南,提供了丰富的实用技巧和代码示例,包括上述装饰器类的设计。如果你在处理Python中的数据结构、字符串操作、数值和日期时间处理、迭代器和生成器等方面遇到问题,这本书是极好的参考资源。通过学习这些章节,你可以深入理解如何灵活运用Python语言特性和库函数,提高代码的可读性和效率。