装饰器

最新推荐文章于 2024-09-21 00:01:32 发布

daimagui

最新推荐文章于 2024-09-21 00:01:32 发布

阅读量86

点赞数

本文链接：https://blog.csdn.net/daimagui/article/details/98115922

版权

装饰器

@func1
def func():
    print('aaa')

装饰器存在的意义

不影响原有函数的功能
可以添加新功能

一般常见的，比如拿到第三方的API接口，第三方不允许修改这个接口。这个时候，装饰器就派上用场了。

装饰器本身也是一个函数，作用就是为现有存在的函数，在不改变函数的基础上，增加一些功能进行装饰。

它是以闭包的形式去实现的。

在使用装饰器函数时，在被装饰的函数的前一行，使用@装饰器函数名形式来进行装饰。

Demo：

现在在一个项目中，有很多函数，由于我们的项目越来越大，功能也越来越多，导致程序越来越慢。

其中一个功能函数的功能是实现一百万次的累加。

计算函数的运行时间：

import time

def my_count():
    s=0
    for i in range(1000001):
        s+=i
        print('sum:,',s)

start=time.time()
my_count()
end=time.time()

print('执行时间为：，',(end-start)) #执行时间为：， 12.710999965667725

将上面时间计算函数封装

import time

def my_count():
    s=0
    for i in range(1000001):
        s+=i
    print('sum:,',s)

# start=time.time()
# my_count()
# end=time.time()
#
# print('执行时间为：，',(end-start))

def count_time(func):
    start=time.time()
    func
    end=time.time()
    print('执行时间为，',(end-start))

count_time(my_count())

经过修改后，定义一个函数来实现时间计算功能。

初看上去，比之前好很多。

只是在使用时，需要将对应的函数传入到时间计算函数中。

import time

def count_time(func):
    def wrapper():
        start=time.time()
        func()
        end=time.time()
        print((end-start))
    return wrapper


def my_count():
    s=0
    for i in range(1000001):
        s+=i
    print('sum:',s)
my_count=count_time(my_count)
my_count()

import time

def count_time(func):
    def wrapper():
        start=time.time()
        func()
        end=time.time()
        print((end-start))
    return wrapper

@count_time
def my_count():
    s=0
    for i in range(1000001):
        s+=i
    print('sum:',s)
# my_count=count_time(my_count)
my_count()

这样实现的好处，定义闭包函数后，只需要通过@装饰器函数名形式的装饰器语法，就可以将

@装饰器函数名加到要装饰的函数前即可。

这种不改变原有函数功能，对函数进行拓展的形式，就称为装饰器

在执行@装饰器函数名时，就是将原函数传递到闭包中。然后，原函数的引用指向闭包返回的装饰过的内部函数的引用。

装饰器的几种形式

1.无参无返回值

def setFunc(func):
    def wrapper():
        print('Start')
        func()
        print('end')
    return wrapper

@setFunc
def show():
    print('show')

show()
# Start
# show
# end

2.无参有返回值

def setFunc(func):
    def wrapper():
        print('Start')
        func()
        print('end')
    return wrapper

@setFunc
def show():
    return 'show'

print(show())
# Start
# end
# None

def setFunc(func):
    def wrapper():
        print('Start')
        return func()
        print('end') #return后面不再执行
    return wrapper

@setFunc
def show():
    return 'show'

print(show())
#Start
#show

3.有参无返回值

def setFunc(func):
    def wrapper(s):
        print('Start')
        func(s)
        print('end') 
    return wrapper

@setFunc
def show(s):
    print('Hello %s'%s)

show('City College')
#Start
#Hello City College
#end

4.有参有返回值

def setFunc(func):
    def wrapper(x,y):
        print('Start')
        return func(x,y)
        # print('end')
    return wrapper

@setFunc
def myAdd(x,y):
    return x+y

print(myAdd(3,3))
#Start
#6

万能装饰器

通过可变参数（*args/**kwargs）来接受不同的参数类型。

def setFunc(func):

    def wrapper(*args,**kwargs):
        print('Wrapper context.')
        return func(*args,**kwargs)

    return wrapper

@setFunc
def func(name,age,job='IT'):
    print(name,age,job)

func('Tom',18)

@setFunc
def demo(a,b,*c,**d):
    print((a,b))
    print(c)
    print(d)

demo('city','college',1999,1,1,school='zhejiang university')

Wrapper context.
Tom 18 IT
Wrapper context.
('city', 'college')
(1999, 1, 1)
{'school': 'zhejiang university'}

多个装饰器

#装饰器1
def setFunc1(func):
    def wrapper(*args,**kwargs):
        print(000)
        func(*args,**kwargs)
        print(111)

    return wrapper
#装饰器2
def setFunc2(func):

    def warpper2(*args,**kwargs):
        print(000)
        func(*args,**kwargs)
        print(111)

    return warpper2

@setFunc1                  #F
@setFunc2                  #g
def show(*args,**kwargs):  #f
    print('show run')

show() #F(g(f))

#从下往上去装饰（程序语句运行顺序来看）
#从内往外去装饰（从函数的调用来看）
0
0
show run
111
111