需求：实现一个可以统计代码的运行时间

需求：有一个做加法计算的函数，要统计执行这个加法函数代码运行了多久

import time


def add(a, b):
    time.sleep(1)
    return a + b


st = time.time()
add(100, 200)
et = time.time()
print("该函数运行时间为：", et - st)

学了闭包＋装饰器后：

import time

#  通过函数实现装饰器
def count_time(func):
    def wrapper(*args, **kwargs):
        st = time.time()
        res = func(*args, **kwargs)
        et = time.time()
        print("该函数运行时间为：", et - st)
        return res

    return wrapper


@count_time
def add(a, b):
    time.sleep(1)
    return a + b


print(add(200, 300))

又学了魔术方法后：

import time

# 通过类实现装饰器
class CountTime:
    def __init__(self, func):
        self.func = func

    def __call__(self, *args, **kwargs):
        st = time.time()
        res = self.func(*args, **kwargs)
        et = time.time()
        print("该函数运行时间为：", et - st)
        return res


@CountTime  # add = CountTime(add)
def add(a, b):
    time.sleep(1)
    return a + b


print(add(200, 300))

需求升级：

装饰器接收一个int类型的参数n，可以用来装饰任何函数，如果函数运行时间大于n，则

打印“请耐心等待，马上回来”和统计函数的运行时间

# 通过函数实现装饰器
import time

def count_time_1(n):
    def count_time(func):
        def wrapper(*args, **kwargs):
            st = time.time()
            res = func(*args, **kwargs)
            et = time.time()
            print("该函数运行时间为：", et - st)
            if et - st > n:
                print("请耐心等待，马上回来")
            return res

        return wrapper

    return count_time


@count_time_1(2)  # add = count_time(2)(add)
def add(a, b):
    time.sleep(2)
    return a + b


print(add(200, 300))

# 通过类实现装饰器
class CountTime:
    def __init__(self, n):
        self.n = n

    def __call__(self, func):
        def wrapper(*args, **kwargs):
            st = time.time()
            res = func(*args, **kwargs)
            et = time.time()
            print("该函数运行时间为：", et - st)
            if et - st > self.n:
                print(f"运行时间超过{self.n}s，请耐心等待，马上回来")
            return res

        return wrapper


@CountTime(1)  # add = CountTime(1)(add)
def add(a, b):
    time.sleep(1)
    return a + b


print(add(100, 200))