Python 装饰器是一种高级功能，允许你在不修改原始函数代码的情况下，动态地修改或扩展函数的行为。

基本语法

装饰器是一种特殊的函数，其基本语法如下：

def decorator_function(func):def wrapper(*args, **kwargs):# 在调用原始函数之前的操作result = func(*args, **kwargs)# 在调用原始函数之后的操作return resultreturn wrapper

常用命令

• @decorator_function：将装饰器应用到函数上的语法糖。
• decorator_function：装饰器函数本身。

示例

示例 1：简单装饰器

def my_decorator(func):def wrapper():print("Something is happening before the function is called.")func()print("Something is happening after the function is called.")return wrapper@my_decorator
def say_hello():print("Hello!")say_hello()

示例 2：带参数的装饰器

def repeat(num_times):def decorator_repeat(func):def wrapper(*args, **kwargs):for _ in range(num_times):result = func(*args, **kwargs)return resultreturn wrapperreturn decorator_repeat@repeat(num_times=3)
def greet(name):print(f"Hello {name}")greet("Alice")

应用场景

1. 日志记录

装饰器在日志记录中发挥着重要作用，它可以捕获函数的输入参数、执行时间以及输出结果，从而方便开发人员跟踪函数的执行过程和调试代码。通过装饰器记录日志，可以提高代码的可读性和可维护性。

示例代码：

import logging
import timedef log_decorator(func):def wrapper(*args, **kwargs):logging.info(f"Calling function {func.__name__} with args: {args}, kwargs: {kwargs}")start_time = time.time()result = func(*args, **kwargs)end_time = time.time()logging.info(f"Function {func.__name__} executed in {end_time - start_time} seconds with result: {result}")return resultreturn wrapper@log_decorator
def add(x, y):return x + yresult = add(3, 5)
print("Result:", result)

2. 性能监控

装饰器在性能监控方面也具有重要作用，它可以帮助开发人员监控函数的执行时间，发现潜在的性能瓶颈并进行优化。通过装饰器进行性能监控，可以提高代码的效率和性能。

示例代码：

import timedef performance_decorator(func):def wrapper(*args, **kwargs):start_time = time.time()result = func(*args, **kwargs)end_time = time.time()print(f"Function {func.__name__} executed in {end_time - start_time} seconds")return resultreturn wrapper@performance_decorator
def calculate_factorial(n):factorial = 1for i in range(1, n + 1):factorial *= ireturn factorialresult = calculate_factorial(10)
print("Factorial:", result)

3. 权限验证

装饰器可以用于权限验证，例如检查用户是否具有执行特定操作的权限。这种方式使得权限验证逻辑与业务逻辑分离，提高了代码的模块化和可维护性。

示例代码：

def permission_required(permission):def decorator(func):def wrapper(*args, **kwargs):if check_permission(permission):return func(*args, **kwargs)else:raise PermissionError("Permission denied")return wrapperreturn decoratordef check_permission(permission):# 检查用户是否具有指定权限的逻辑return True  # 此处仅为示例，实际需根据业务逻辑实现@permission_required("admin")
def delete_user(user_id):# 删除用户的逻辑print(f"User {user_id} deleted successfully")delete_user(123)

注意事项

1. 装饰器顺序

当多个装饰器应用于同一个函数时，它们的执行顺序与它们在代码中的顺序相反。这意味着最先定义的装饰器实际上会最后执行，而最后定义的装饰器会最先执行。

示例代码：

def decorator1(func):def wrapper():print("Decorator 1 executed")func()return wrapperdef decorator2(func):def wrapper():print("Decorator 2 executed")func()return wrapper@decorator1
@decorator2
def greet():print("Hello!")greet()

输出结果为：

Decorator 1 executed
Decorator 2 executed
Hello!

2. 装饰器的参数

如果装饰器本身需要接受参数，则需要在装饰器函数外再包裹一层函数。这样的装饰器称为带参数的装饰器。在定义带参数的装饰器时，外层函数接受装饰器参数，内层函数接受被装饰函数。

示例代码：

def decorator_with_param(param):def decorator(func):def wrapper():print(f"Decorator with parameter {param} executed")func()return wrapperreturn decorator@decorator_with_param("test")
def greet():print("Hello!")greet()

输出结果为：

Decorator with parameter test executed
Hello!

总结

装饰器是 Python 中一种强大的工具，可用于动态修改函数的行为，常用于日志记录、性能监控、权限验证等场景。通过合理使用装饰器，可以提高代码的灵活性、可重用性和可维护性。