Python小结

可迭代对象、迭代器、生成器

装饰器 Decorators

装饰器（Decorator）核心是高阶函数（函数可以接收函数作为参数，并返回一个新函数，包裹了原逻辑 + 新功能）。装饰器不应偷偷改变返回类型和异常语义，否则调用者难以推理。

def log_decorator(func):
    def wrapper(*args, **kwargs):
        print(f"调用 {func.__name__}()")
        return func(*args, **kwargs)
    return wrapper

@log_decorator
def greet():
    print("Hello!")

语法糖 @log_decorator 这行相当于：greet = log_decorator(greet)

扩展：带参数的装饰器
装饰器本身也能接收参数（会多包一层函数，用于传递其它的参数）：

def repeat(num_times): 
    def decorator(func):
        @functools.wraps(func) # 保留原函数的元信息:functools.wraps 复制 __name__、__doc__ 等元数据，并通过 __wrapped__ 保留原函数入口
        def wrapper(*args, **kwargs):
            for _ in range(num_times):
                func(*args, **kwargs)
        return wrapper
    return decorator

@repeat(num_times=3)
def greet():
    print("Hello!")

# @repeat(num_times=3) 装饰器参数
# 等价于
# decorator = repeat(num_timies=3) 先调用并返回结果（传递了参数）
# @decorator 然后再装饰

总结：带参数装饰器会多一层函数：第一层接收配置，第二层接收被装饰函数，第三层才处理实际调用。