Python 如何实现装饰器？

Question

Python 如何实现装饰器？

仓乃文 · Accepted Answer

装饰器本质是一个接收函数并返回新函数的高阶函数，用于在不修改原函数的前提下增加额外功能，如日志、计时、权限校验。

基础装饰器（无参数）：
```python
import time
from functools import wraps

def timer(func):
    @wraps(func)  # 保留原函数的元信息（__name__、__doc__）
    def wrapper(*args, **kwargs):
        start = time.time()
        result = func(*args, **kwargs)
        print(f"{func.__name__} 耗时: {time.time() - start:.3f}s")
        return result
    return wrapper

@timer
def slow_function():
    time.sleep(1)
    return "done"
```

带参数的装饰器（三层嵌套）：
```python
def retry(max_attempts=3, delay=1):
    def decorator(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            for attempt in range(max_attempts):
                try:
                    return func(*args, **kwargs)
                except Exception as e:
                    if attempt == max_attempts - 1:
                        raise  # 最后一次仍失败则抛出
                    time.sleep(delay)
        return wrapper
    return decorator

@retry(max_attempts=3, delay=2)
def fetch_data():
    return requests.get("https://api.example.com").json()
```

补充说明：
- @wraps 保留原函数的 __name__、__doc__ 等元信息
- 无参数装饰器两层嵌套，有参数装饰器三层嵌套
- 装饰器在函数定义时执行，不是在调用时
- 常见用途：日志记录、性能计时、权限校验、缓存、重试