Python作为一门多范式的编程语言，提供了丰富的工具和特性来帮助开发者简化和优化代码。其中，装饰器和上下文管理器是两种非常重要的工具，它们可以在函数和语句级别上扩展功能，使得代码更加灵活和强大。
装饰器
装饰器是一种特殊的函数，它可以在不修改原函数的情况下，为函数添加额外的功能。装饰器通常用于在函数执行前后执行一些额外的代码，比如记录日志、权限验证、性能监控等。在Python中，装饰器通常使用@符号加上装饰器函数的名称来应用到目标函数上。
def log(func):
def wrapper(*args, **kwargs):
print(f'Calling function {func.__name__}')
result = func(*args, **kwargs)
print(f'Function {func.__name__} finished')
return result
return wrapper
@log
def greet(name):
return f'Hello, {name}!'
print(greet('Alice'))
在上面的例子中，log函数是一个装饰器函数，它将函数调用的日志记录功能添加到了greeet函数上。当我们调用greet函数时，实际上是调用了wrapper函数，而wrapper函数又会调用原函数，并在调用前后记录了日志。
上下文管理器
上下文管理器是一种对象，它实现了__enter__和__exit__方法，可以被with语句使用。上下文管理器可以在代码块执行前后执行一些额外的代码，比如资源的申请和释放、事务的管理等。使用上下文管理器可以确保资源的正确释放，避免资源泄露和错误处理。
class FileHandler:
def __init__(self, filename, mode):
self.filename = filename
self.mode = mode
def __enter__(self):
self.file = open(self.filename, self.mode)
return self.file
def __exit__(self, exc_type, exc_value, exc_tb):
self.file.close()
with FileHandler('file.txt', 'w') as f:
f.write('Hello, world!')
在上面的例子中，FileHandler实现了__enter__和__exit__方法，可以在with语句中被使用。当with语句块执行时，首先会调用__enter__方法获取资源，然后执行代码块，最后调用__exit__方法释放资源。这样可以确保资源的正确释放，避免忘记释放资源和错误处理。
总结
装饰器和上下文管理器是Python中非常强大的工具，它们可以在函数和语句级别上扩展功能，使得代码更加灵活和强大。合理使用装饰器和上下文管理器，能够提高代码的可维护性和可扩展性，是Python编程中不可或缺的工具。希望通过本文的介绍，你能更好地理解和运用装饰器和上下文管理器，为你的代码增添新的可能性。
本文来自极简博客，作者：网络安全守护者，转载请注明原文链接：Python的装饰器与上下文管理器：扩展函数和语句的功能
