"""
裝飾器（decorators）是 Python 中的一種高級功能，它允許你動(dòng)態(tài)地修改函數(shù)或類的行為。
裝飾器是一種函數(shù)，它接受一個(gè)函數(shù)作為參數(shù)并返回一個(gè)新的函數(shù)或修改原來的函數(shù)。
本質(zhì)上是一個(gè)接收函數(shù)作為輸入并返回一個(gè)新的包裝過后的函數(shù)的對象。
"""
# 1 基礎(chǔ)使用
# my_decorator 是一個(gè)裝飾器函數(shù)，它接受say_hello函數(shù)作為參數(shù)并返回 wrapper 函數(shù)。
def my_decorator(original_function):
    def wrapper():
        print("在原函數(shù)之前執(zhí)行")
        original_function()
        print("在原函數(shù)之后執(zhí)行")
    return wrapper
# my_decorator注解將 say_hello 替換為 wrapper
@my_decorator
def say_hello():
    print("Hello!")
say_hello()
print("*" * 10)
# 2 帶參數(shù)的裝飾器
def my_decorator_v2(original_function):
    def wrapper(*args, **kwargs):
        print("在原函數(shù)之前執(zhí)行")
        original_function(*args, **kwargs)
        print("在原函數(shù)之后執(zhí)行")
    return wrapper
@my_decorator_v2
def say_bye(name):
    print(f"Hello, {name}!")
say_bye("李四")
import time
# 步驟1：定義裝飾器函數(shù)（接收被裝飾的函數(shù)作為參數(shù)）
def timer_decorator(original_function):
    # 步驟2：定義嵌套的包裝函數(shù)（封裝額外功能，調(diào)用原函數(shù)）
    def wrapper():
        # 額外功能：記錄開始時(shí)間
        start_time = time.time()
        print("在原函數(shù)之前執(zhí)行")
        # 調(diào)用被裝飾的原函數(shù)，保留其核心功能
        result = original_function()
        print("在原函數(shù)之后執(zhí)行")
        # 額外功能：計(jì)算并打印耗時(shí)
        end_time = time.time()
        print(f"函數(shù) {original_function.__name__} 執(zhí)行耗時(shí)：{end_time - start_time:.6f} 秒")
        # 返回原函數(shù)的執(zhí)行結(jié)果，保證被裝飾函數(shù)的返回值不丟失
        return result
    # 步驟3：返回包裝函數(shù)（而非直接執(zhí)行，保留包裝邏輯供后續(xù)調(diào)用）
    return wrapper
# 步驟4：定義被裝飾的目標(biāo)函數(shù)（核心業(yè)務(wù)邏輯）
def my_business_func():
    time.sleep(0.5)  # 模擬耗時(shí)操作
    print("核心業(yè)務(wù)邏輯執(zhí)行完成")
    return "執(zhí)行成功"
# 步驟5：手動(dòng)應(yīng)用裝飾器
my_business_funcVar = timer_decorator(my_business_func)
print(my_business_funcVar)
# 調(diào)用裝飾后的函數(shù)
result = my_business_funcVar()
print("函數(shù)返回值：", result)
def timer_decorator(original_function):
    # 包裝函數(shù)添加 *args 和 **kwargs，兼容任意參數(shù)
    def wrapper(*args, **kwargs):
        start_time = time.time()
        # 調(diào)用原函數(shù)時(shí)傳入?yún)?shù)，保證核心邏輯正常執(zhí)行
        result = original_function(*args, **kwargs)
        end_time = time.time()
        print(f"函數(shù) {original_function.__name__} 執(zhí)行耗時(shí)：{end_time - start_time:.6f} 秒")
        return result
    return wrapper
@timer_decorator
def my_business_func(name, age=20):
    time.sleep(0.5)
    print(f"核心業(yè)務(wù)邏輯：姓名{name}，年齡{age}")
    return f"{name} 業(yè)務(wù)執(zhí)行成功"
# 帶參數(shù)調(diào)用，裝飾器正常工作
result = my_business_func("張三", age=25)
print("函數(shù)返回值：", result)

def log(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        print(f"Log: {func.__name__}")
        return func(*args, **kwargs)
    return wrapper
def timer(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        start = time.time()
        result = func(*args, **kwargs)
        print(f"Timer: {time.time() - start:.4f}s")
        return result
    return wrapper
@log
@timer
def add(a, b):
    return a + b
add(3, 4)  # 先執(zhí)行 log，再執(zhí)行 timer