Python 装饰器详解：概念、类型与应用场景 | 极客日志

Python 装饰器详解：概念、类型与应用场景 | 极客日志

def say_hello():
    print("你好！")
say_hello()
# 输出：你好！

def decorator(func):
    # 这个就是装饰器
    def wrapper():
        # 这个就是具体的装饰
        print("准备开始啦！")
        func()  # 这里调用原来的函数
        print("结束啦！")
    return wrapper

def say_hello():
    print("你好！")

# 给函数「加上装饰」
# 把 say_hello 这个礼物，放进 decorator 这个包装盒里，然后把包装好的新东西命名为 decorated_say_hello。
decorated_say_hello = decorator(say_hello)
# 运行
decorated_say_hello()
# 输出结果
# 准备开始啦！
# 你好！
# 结束啦！

装饰器 └── 包装纸 └── 苹果

def decorator(func):
    def wrapper():
        print("准备开始啦！")
        func()
        print("结束啦！")
    return wrapper

@decorator
# 就像给函数戴上一顶魔法帽子
def say_hello():
    print("你好！")

say_hello()
# 直接调用就会自动被装饰
# 输出结果
# 准备开始啦！
# 你好！
# 结束啦！

# 这是一个'加工机器'（装饰器），它会给任何函数加上'说你好'的功能
def say_hello_machine(func):
    def wrapper():
        print("你好！")  # 这是加上的'巧克力酱'
        func()  # 这是原来的圆筒冰淇淋
        print("再见！")  # 这是加上的'坚果碎'
    return wrapper

@say_hello_machine
# 直接把'圆筒'放进'机器'
def my_name():
    print("我是小明")

# 现在直接调用，它就已经是被装饰后的'豪华冰淇淋'了
my_name()
# 输出结果
# 你好！
# 我是小明
# 再见！

# 这个机器自己需要一个参数，来决定说多少次你好
def say_hello_machine_times(times):
    # 机器的调节按钮
    def real_decorator(func):
        # 这才是真正的'加工器'
        def wrapper():
            for i in range(times):
                print("你好！")
            func()
        return wrapper
    return real_decorator

# 使用这个高级机器，把'调节按钮'拧到 3
@say_hello_machine_times(3)
# 告诉机器，要说 3 次你好
def my_name():
    print("我是小明")

my_name()
# 输出结果
# 你好！
# 你好！
# 你好！
# 我是小明

# 这是一个用'类'做的机器
class MyDecorator:
    def __init__(self, func):
        self.func = func

    def __call__(self):
        print("类装饰器说：开始！")
        self.func()
        print("类装饰器说：结束！")

# 使用类装饰器
@MyDecorator
def my_name():
    print("我是小明")

my_name()
# 输出结果
# 类装饰器说：开始！
# 我是小明
# 类装饰器说：结束！

class Circle:
    def __init__(self, radius):
        self.radius = radius

    def area(self):
        print(self.radius * self.radius * 3.14)

circle = Circle(10)
circle.area()  # 需要加括号
# 输出 314.0

class Circle:
    def __init__(self, radius):
        self.radius = radius

    @property
    def area(self):
        print(self.radius * self.radius * 3.14)

circle = Circle(10)
circle.area  # 直接当属性用，不用加括号！
# 输出 314.0

class ThirdGradeClass:
    total_students = 0  # 这是一个全班共有的变量

    def __init__(self, name):
        self.name = name
        ThirdGradeClass.total_students += 1  # 每创建一个新学生，总数 +1

    @classmethod
    def get_total_students(cls):  # 注意这里的参数是 cls，代表'类'本身
        return cls.total_students

# 创建几个学生
xiaoming = ThirdGradeClass("小明")
xiaohong = ThirdGradeClass("小红")

# 调用类方法，不需要通过某个具体的学生，通过类名就可以调用！
print(ThirdGradeClass.get_total_students())  # 输出：2
# 通过某个学生对象也可以调用，但它知道的仍然是全班的信息
print(xiaoming.get_total_students())  # 输出：2

class ThirdGradeClass:
    @staticmethod
    def add_numbers(a, b):  # 这个函数完全不知道班级、学生的存在
        # 它只是一个实现特定功能的普通函数
        return a + b

# 使用这个静态方法，既可以通过类调用，也可以通过对象调用
# 但它就是个计算工具，和班级、学生没关系
result1 = ThirdGradeClass.add_numbers(5, 3)  # 输出：8
result2 = xiaoming.add_numbers(10, 2)  # 输出：12

def build_house():
    print("记录：玩家建造了房子！")
    # 每次都要写这个 print
    print("砰！砰！砰！一栋房子盖好了！")

def mine_ore():
    print("记录：玩家挖矿了！")
    # 这个也要写
    print("镐子咣咣咣，挖到了钻石！")

build_house()
mine_ore()

# 这是一个'记录魔法装备'（装饰器）
def log_action(func):
    def wrapper():
        print(f"记录：玩家进行了 {func.__name__} 操作！")  # 自动获取函数名
        func()  # 执行原本的操作
    return wrapper

# 给函数穿上魔法装备！
@log_action
def build_house():
    print("砰！砰！砰！一栋房子盖好了！")

@log_action
def mine_ore():
    print("镐子咣咣咣，挖到了钻石！")

# 现在直接调用，就会自动记录啦！
build_house()
mine_ore()

记录：玩家进行了 build_house 操作！
砰！砰！砰！一栋房子盖好了！
记录：玩家进行了 mine_ore 操作！
镐子咣咣咣，挖到了钻石！

import time

# '计时魔法装备'
def timer(func):
    def wrapper():
        start_time = time.time()  # 开始计时
        func()  # 执行动作
        end_time = time.time()  # 结束计时
        print(f"【{func.__name__}】动作耗时：{end_time - start_time:.2f} 秒")
    return wrapper

# 给砍树函数穿上计时装备
@timer
def chop_tree():
    print("咔嚓...咔嚓...")
    time.sleep(2)  # 假装砍树花了 2 秒钟
    print("获得木头*5！")

chop_tree()

咔嚓...咔嚓...
获得木头*5！
【chop_tree】动作耗时：2.00 秒

player_level = 8  # 假设玩家现在 8 级

# '权限检查魔法装备'
def check_level(required_level):
    def decorator(func):
        def wrapper():
            if player_level >= required_level:
                func()  # 等级足够，执行操作
            else:
                print(f"权限不足！需要 {required_level} 级，你只有 {player_level} 级。")
        return wrapper
    return decorator

# 使用末影箱需要 10 级，给它穿上检查装备！
@check_level(required_level=10)
def use_ender_chest():
    print("成功打开末影箱，存取物品！")

use_ender_chest()

权限不足！需要 10 级，你只有 8 级。