装饰器可用于扩展类方法功能而不修改原代码，1. 使用函数装饰器如@log_calls可添加日志记录；2. 使用@validate_input可在调用前验证参数类型；3. 装饰@classmethod时需确保装饰器正确处理cls参数且顺序正确；4. 使用类装饰器如callcounter可维护调用状态；5. 常见问题包括遗漏@functools.wraps导致元数据丢失、参数处理错误及装饰器顺序不当；6. 装饰器可与继承结合，子类继承或重写装饰方法；7. 实际应用包括权限验证、缓存、事务管理和性能监控。
给类方法添加功能，装饰器是个好帮手。它允许你在不修改原类方法代码的情况下，扩展或修改其行为。
解决方案
装饰器本质上是一个 Python 函数，它接受另一个函数作为输入并返回一个新的函数。对于类方法，我们需要使用一些特殊的装饰器，例如
@classmethod或
@staticmethod，或者自定义装饰器来正确处理
self或
cls参数。
立即学习“Python免费学习笔记（深入）”；
让我们看几个例子：
- 简单的功能扩展：
假设你想在每次调用某个类方法时记录日志。你可以创建一个装饰器来实现：
import functools
def log_calls(func):
@functools.wraps(func)
def wrapper(*args, **kwargs):
print(f"Calling {func.__name__} with args: {args}, kwargs: {kwargs}")
result = func(*args, **kwargs)
print(f"{func.__name__} returned: {result}")
return result
return wrapper
class MyClass:
def __init__(self, value):
self.value = value
@log_calls
def my_method(self, arg1):
return self.value + arg1
# Example usage
obj = MyClass(10)
result = obj.my_method(5) # Output will include log messages
print(result)这里，
log_calls装饰器包装了
my_method，在方法调用前后打印日志。
functools.wraps用于保留原始函数的元数据，如
__name__和
__doc__。
- 修改类方法的行为：
有时候，你可能需要修改类方法的行为。例如，你可能想在方法执行前进行一些验证。
def validate_input(func):
@functools.wraps(func)
def wrapper(self, arg):
if not isinstance(arg, int):
raise ValueError("Argument must be an integer")
return func(self, arg)
return wrapper
class MyClass:
def __init__(self, value):
self.value = value
@validate_input
def update_value(self, new_value):
self.value = new_value
# Example usage
obj = MyClass(10)
try:
obj.update_value("abc") # Raises ValueError
except ValueError as e:
print(e)在这个例子中，
validate_input装饰器确保
update_value方法的参数是整数。
- 装饰
@classmethod
和@staticmethod
：
装饰类方法和静态方法稍微复杂一些，因为你需要确保装饰器正确处理
cls或没有隐式第一个参数的情况。
def classmethod_decorator(func):
@functools.wraps(func)
def wrapper(cls, *args, **kwargs):
print(f"Calling class method {func.__name__} with cls: {cls}, args: {args}, kwargs: {kwargs}")
result = func(cls, *args, **kwargs)
return result
return wrapper
class MyClass:
@classmethod_decorator
@classmethod
def my_class_method(cls, arg):
return f"Class method called with {arg}"
print(MyClass.my_class_method("test"))注意装饰器的顺序。
@classmethod应该放在最靠近方法定义的位置，然后是你的自定义装饰器。
- 更复杂的场景：使用类作为装饰器
类装饰器允许你维护状态，这在某些情况下非常有用。
class CallCounter:
def __init__(self, func):
self.func = func
self.call_count = 0
def __call__(self, *args, **kwargs):
self.call_count += 1
print(f"Call count for {self.func.__name__}: {self.call_count}")
return self.func(*args, **kwargs)
class MyClass:
@CallCounter
def my_method(self):
print("Method called")
obj = MyClass()
obj.my_method()
obj.my_method()CallCounter类记录了
my_method被调用的次数。
如何选择合适的装饰器方法？
选择哪种方法取决于你的具体需求。如果只是简单地添加日志或验证，简单的函数装饰器就足够了。如果需要维护状态或进行更复杂的操作，类装饰器可能更合适。
装饰器有哪些常见的坑？
-
忘记
@functools.wraps
： 这会导致原始函数的元数据丢失，影响调试和文档。 -
参数处理不当： 确保装饰器正确处理
self
、cls
以及其他参数。 -
装饰器顺序错误： 特别是与
@classmethod
和@staticmethod
一起使用时。
装饰器与继承有什么关系？
装饰器可以很好地与继承结合使用。你可以装饰父类的方法，然后子类会继承装饰后的方法。或者，你可以选择在子类中重新装饰方法，以覆盖父类的装饰器。
装饰器在实际项目中的应用案例？
- 权限验证： 检查用户是否有权访问某个方法。
- 缓存： 缓存方法的结果，避免重复计算。
- 事务管理： 在方法执行前后启动和提交事务。
- 性能监控： 记录方法的执行时间。
