Python装饰器超详细实例教程

# 定义装饰器：接收一个函数，返回一个新函数
def my_decorator(func):
    def wrapper():
        print("函数执行前：添加额外功能")
        func()  # 调用原函数
        print("函数执行后：添加额外功能")
    return wrapper
# 原函数
def say_hello():
    print("Hello 装饰器！")
# 手动装饰：不修改原函数，给它加功能
say_hello = my_decorator(say_hello)
# 调用
say_hello()

def my_decorator(func):
    def wrapper():
        print("函数执行前")
        func()
        print("函数执行后")
    return wrapper
# 直接用 @ 装饰
@my_decorator
def say_hello():
    print("Hello 装饰器！")
say_hello()

def my_decorator(func):
    # wrapper 接收任意参数，传给原函数
    def wrapper(*args, **kwargs):
        print("执行前")
        # 调用原函数并传参
        result = func(*args, **kwargs)
        print("执行后")
        # 返回原函数的返回值
        return result
    return wrapper
@my_decorator
def add(a, b):
    return a + b
print(add(10, 20))

# 最外层：接收装饰器参数
def decorator_with_param(msg):
    # 中层：真正的装饰器
    def my_decorator(func):
        # 内层：包装函数
        def wrapper(*args, **kwargs):
            print(f"装饰器参数：{msg}")
            result = func(*args, **kwargs)
            return result
        return wrapper
    return my_decorator
# 使用：装饰器带参数
@decorator_with_param("我是自定义参数")
def test():
    print("原函数执行")
test()

import functools
def my_decorator(func):
    # 保留原函数信息
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        print("前置功能")
        return func(*args, **kwargs)
    return wrapper
@my_decorator
def add(a, b):
    """两数相加"""
    return a + b
# 打印原函数名和文档，不会被装饰器覆盖
print(add.__name__)  # 输出 add
print(add.__doc__)   # 输出 两数相加

import time
import functools
def timer(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        start = time.time()
        res = func(*args, **kwargs)
        end = time.time()
        print(f"函数 {func.__name__} 耗时：{end - start:.4f}s")
        return res
    return wrapper
@timer
def long_task():
    time.sleep(1)
long_task()

import functools
def logger(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        print(f"调用函数：{func.__name__}，参数：{args} {kwargs}")
        res = func(*args, **kwargs)
        print(f"函数返回：{res}")
        return res
    return wrapper
@logger
def add(a, b):
    return a + b
add(5, 3)

def dec1(func):
    def wrapper():
        print("装饰器1 前置")
        func()
        print("装饰器1 后置")
    return wrapper
def dec2(func):
    def wrapper():
        print("装饰器2 前置")
        func()
        print("装饰器2 后置")
    return wrapper
# 执行顺序：dec1前置 → dec2前置 → 原函数 → dec2后置 → dec1后置
@dec1
@dec2
def test():
    print("原函数")
test()