亲宝软件园·资讯

展开

Python中的Decorator装饰器的使用示例

漂泊的小森 人气:0

#最近在网上看代码时,出现了@???的代码,看了好久也不知道是什么意思,经过了解原来是装饰器,我给大家举个例子讲解一下,帮助大家快速理解:

##大家在写函数的时候,可能会想测试这个函数的运行时间,如果每个函数都加一个类似于:

start=time.time()
函数主体
over=time.time()
Time=over-start
print(Time)
def haha():
    start=time.time()
    for i in range(1200):
        if i%2==0:
            print('这是个偶数:',i)
    over=time.time()
    print("总共花费的时间:",over-start)
haha()

这样函数的可读性也不高,如果要测试好几个函数还会大量浪费时间。

下面引入装饰器:

1简单的装饰器操作:

(先定义一个函数1(里面传递一个参数),然后在函数1里面再定义一个函数2负责计算时间,并在时间中间调用定义函数1的参数,然后返回这个函数2):

!!!当我们想调用装饰器时,只需@函数1的名字,放在想要使用的函数前面即可。

import time
def decorator(fun):
    def wrapper():
        t1=time.time()
        fun()
        t2=time.time()
        print('总共花费的时间:',t2-t1)
    return wrapper
@decorator
def haha():
    #start=time.time()
    for i in range(1200):
        if i%2==0:
            print('这是个偶数:',i)
    #over=time.time()
    #print("总共花费的时间:",over-start)
haha()

2,装饰器升级版(如果我们想套用的函数中,具有返回对应的值这个功能共时使用):

例如这个函数:

def haha():
    a=0
    #start=time.time()
    for i in range(1200):
        if i%2==0:
            print('这是个偶数:',i)
            a=a+i
    return a

#我们如果还用1中的装饰器就不能返回对应的值,这就需要改动装饰器:

(在装饰器中也要把函数中定义参数的返回值给表示出来,并且利用return 给返回值,输出出来)

import time
def decorator1(fun):
    def wrapper2():
        t1=time.time()
        result=fun()
        t2=time.time()
        print('总共花费的时间:',t2-t1)
        return result
    return wrapper2
@decorator1
def haha():
    a=0
    #start=time.time()
    for i in range(1200):
        if i%2==0:
            print('这是个偶数:',i)
            a=a+i
    return a
    #over=time.time()
    #print("总共花费的时间:",over-start)
aaa=haha()
print(aaa)

可能有同学会问,为什么后面要用到aaa=haha(),

这是因为我们要接受返回值result,并将其用print()打印出来。

3,装饰器的升级升级版(当我们需要使用的函数haha中,不但有返回值,还有传入的参数时使用。)

(只需要在我们的函数2和调用的函数1中的参数()里面加上关键字传参*args,即可)

import time
def decorator1(fun):
    def wrapper2(*args):
        t1=time.time()
        result=fun(*args)
        t2=time.time()
        print('总共花费的时间:',t2-t1)
        return result
    return wrapper2
@decorator1
def haha(key):
    a=0
    #start=time.time()
    for i in range(key):
        if i%2==0:
            print('这是个偶数:',i)
            a=a+i
    return a
    #over=time.time()
    #print("总共花费的时间:",over-start)
aaa=haha(1200)
print(aaa)

可不要忘了,调用函数haha时要传入参数哦。

加载全部内容

相关教程
猜你喜欢
用户评论