Python-装饰器详解

装饰器

介绍

在Python中，装饰器是一种特殊的语法，用于修改或增强函数的功能。装饰器是Python的高级特性之一，它允许我们通过在不修改原函数代码的情况下，添加额外的功能或行为。

装饰器是一个函数，它接受一个函数作为参数，并返回一个新的函数。装饰器函数通常使用@符号放在要修饰的函数定义之前

python中的装饰器(decorator)一般采用语法糖的形式，是一种语法格式。比如：@classmethod，@staticmethod，@property，@wraps()，等都是python中的装饰器。

#装饰器: 是一种程序设计模式,主要用于给函数或者类添加一些额外的功能
#   又不希望通过继承或者修改源代码的方式去实现，那就使用装饰器

# 装饰器：不改变函数或类或者源代码的基础上，添加额外功能。
# 装饰器的本质就是闭包，它需要把一个callable对象（能够打括号调用的对象）作为参数转递进来。

使用

闭包

闭包：需要满足三个条件

• 有内外函数
• 外函数要返回内函数,而不是返回函数的调用结果
• 内函数有引用外函数的变量

函数体内的局部变量，一般情况下，调用的时候生成，结束的使用释放
在闭包的条件下，局部变量不会随着调用的结束而释放

变量生命周期
局部变量 -- 函数的调用生成,函数调用完成释放
def func1():
    print("i am func1")


def outer():
    x = 30
    func1()
    print(f"x is {x}")

outer()
print(x)

def outer(x):
    a = 200
    def inner():
        print(x+a)
    return inner

d = outer(100)
#此时，d就是内函数，如果不调用d，就不会有输出
# d()
# 我们需要调用内函数才可以看到调用结果。

装饰器

看一段代码执行时间：

定义func1函数

def func1():
    time.sleep(1)
    print("i am func1")

在函数前后查看时间。

func1()

# 时间戳
start = time.time()
f = func1()
end = time.time()
print(f"执行函数花了{end-start}s")

使用装饰器查看时间

定义装饰器

def runtime(func):
    def inner(*args,**kwargs):          # *args, **kwargs  让装饰器更加通用。
        start = time.time()
        result = func(*args,**kwargs)
        end = time.time()
        print(f"函数执行花了{end-start}s")
        print(f"执行了{func.__name__}函数")
        return result                   # inner 函数返回原函数的返回值。
    return inner

定义函数

@runtime
def func2(a,b):
    time.sleep(2)
    print("i am func2")
    return a+b

func2(1,2)

注意：

• 装饰器只能装饰函数或者类
• 被装饰过的函数，不是原函数了，是装饰器返回的内函数
• 添加的功能放在装饰器的内函数里面。

练习：

添加额外功能
统计运行时间
权限控制
日志记录

执行了什么函数  -- 写入日志，记录INFO 等级 ，写到文件和屏幕
计时功能装饰器

import functools
from logset import logger

def log(func):
    def inner_log(*args, **kwargs):
        logger.info(f"开始执行{func.__name__}函数")
        result = func(*args, **kwargs)
        logger.info(f"{func.__name__}函数，执行结束！")
        return result
    return inner_log

def runtime(func):
    @functools.wraps(func)
    def inner(*args,**kwargs):
        start = time.time()
        result = func(*args,**kwargs)
        end = time.time()
        print(f"函数执行花了{end-start}s")
        return result

    return inner

@log                # test_func1 = log ( runtime( test_func1() ) )
@runtime
def test_func1():
    time.sleep(1)
    print("this is func1")


test_func1()

装饰器带参数

自身不转入参数的装饰器，使用两层函数定义
自身转入参数的装饰器，使用三层函数定义

import time

user = {"root":"123456","admin":"admin123"}

def deco(username,passwd):
    def runtime(func):
        def inner(*args,**kwargs):
            if username in user and user[username] == passwd:
                if username == "root":
                    start = time.time()
                    result = func(*args,**kwargs)
                    end = time.time()
                    print(f"{func.__name__}执行花了{end-start}s")
                    return result
                else:
                    print("用户名密码不正确")
            else:
                print(f"用户:{username}没有权限")
        return inner
    return runtime

@deco(username="r",passwd="123456")
def add(a,b):
    time.sleep(1)
    return a+b

add(1,2)

类实现装饰器

import time

class A():
    def __init__(self,func):
        self.func = func

    def __call__(self, *args, **kwargs):
        start = time.time()
        result = self.func()
        end = time.time()
        print(f"{self.func.__name__}函数执行花了{end - start}s")
        return  result


@A              #   func1 = A(func1)   这是实例化
def func1():
    time.sleep(1)
    print("i am func1")


func1()

带参数的装饰器,用类实现

user = {"root":"123456","admin":"admin123"}
class A:
    def __init__(self,username,passwd):
        self.username = username
        self.passwd = passwd

    def __call__(self, func):
        def inner(*args, **kwargs):
            if self.username in user and user[self.username] == self.passwd:
                if self.username == "root":
                    start = time.time()
                    result = func(*args, **kwargs)
                    end = time.time()
                    print(f"{func.__name__}执行花了{end - start}s")
                    return result
                else:
                    print("用户名密码不正确")
            else:
                print(f"用户:{self.username}没有权限")
        return inner





@A(username = "root",passwd="123456")       #   a = A(username = "root"),func1 = a(func1)   这是实例化
def func1():
    time.sleep(1)
    print("i am func1")

func1()