Python 装饰器写法

需求：想要计算每个模块执行的时间。

> 传统方法，在每一个模块里面书写一个time时间模块即可。

import time
def count():
    t1 = time.time()
    sum = 0
    for i in range(0,1000):
        sum = sum + i
    t2 = time.time()
    print(t2-t1)
    return sum

def count1():
    t1 = time.time()
    sum = 0
    for i in range(0,1000):
        sum = sum + i
    t2 = time.time()
    print(t2-t1)
    return sum

> 装饰器：实现一个模块计算时间，一个模块书写应用，两个功能拆分开

import time

# 书写装饰器实现计算func()方法执行的时间,wrapper翻译为包装纸，包装材料
"""第一步：定义装饰器函数"""
def time1(func):
    def wrapper():
        t1 = time.time()
        func()
        t2 = time.time()
        print(t2 - t1)
    return wrapper

"""第二步：将转装饰器函数放在需要进行计算的模块上方"""

@time1
def count():
    sum = 0
    for i in range(0,1000):
        sum = sum + i
    return sum


# 拓展一下需求：需要给count传递参数，如何实现

"""
核心在于参数的携带方式
*args表示任意多个参数

"""
def time2(func):
    def wrapper(*args):
        t1 = time.time()
        res = func(*args)
        t2 = time.time()
        print(t2 - t1)
        """因为func即就是count具有返回值，因此需要显性的赋值给res变量"""
        print(res)
    return wrapper


@time1
def count(x,y):
    sum = 0
    for i in range(x,y):
        sum = sum + i
    return sum

if __name__ == '__main__':
    #将参数携带进来
    count(2,10000)