使用Python的装饰器实现功能增强
在Python中,装饰器是一种特殊类型的函数,它用于修改其他函数的功能。以下是一个简单的例子,展示如何使用装饰器来实现功能增强。
假设我们有一个函数func
,它只是打印一条消息。我们可以使用装饰器来添加新的功能,比如计算消息的长度。
def lengthen(func):
def wrapper(*args, **kwargs):
message = func(*args, **kwargs))
return len(message)
return wrapper
@lengthen
def func(msg):
print(msg)
print(func("Hello, World!"))) # 输出:13
在这个例子中,lengthen
装饰器接收一个函数并返回一个新的函数。新的函数在调用原始函数后,会计算并返回消息的长度。
还没有评论,来说两句吧...