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)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值