装饰器详解

装饰器详解

简单实例教程

需要装饰的函数

def login(username, password):
    print(username, password)

装饰器的参数是方法名,就是被装饰的函数

def decorator(func):
    # 在定义一个函数用于实现装饰,参数要和被装饰的函数一样
    def new_login(username, password):
        # 共性代码
        print(username)
        # login函数特有代码
        func(username, password)
        # 共性代码
        print(password)
    # 执行new_login函数
    return new_login

被装饰的函数需要这样写

@decorator
def login(username, password):
    print(username, password)

调用

login("你打球真像蔡徐坤", "你鸡太美")

结果

在这里插入图片描述

装饰器原理

装饰器函数

def decorator(func):
    # 在定义一个函数用于实现装饰,参数要和被装饰的函数一样
    def new_login(username, password):
        # 共性代码
        print(username)
        # login函数特有代码
        func(username, password)
        # 共性代码
        print(password)
    # 执行new_login函数
    return new_login

被装饰函数

def login(username, password):
    print(username, password)
# 此操作等于在函数名上加@decorator
login = decorator(login)
login("你打球真像蔡徐坤", "你鸡太美")

结果

在这里插入图片描述

小结

==A:login B:decorator C:new_login==
装饰器的原理是:将函数A的函数名作为函数B的参数传入,函数B的函数体中重新定义一个函数C,函数C是函数A的重写,在C的函数体中调用函数A,前后都可以扩充代码,达到重写函数A的目的,最后函数B将函数C返回赋予A,使得A被重写成C函数,调用被装饰的A函数,实质就是调用C函数