[Python知识库]python中的装饰器

python中的装饰器

1.写一个知道函数的执行时间的函数

import time #引入time模块
def tim():
	start = time.time() #记录起始时间
	time.sleep(2)   #让程序休眠2秒
	print('该函数的执行时间是：') 
	end = time.time()  #记录截至时间
	print(end - start) #计算时间差
tim()  #执行函数

2.如果我想对多个程序都要进行执行时间测量。都要写这样的程序岂不是很麻烦，所以把公共部分抽象出来

def  func():
	print('函数执行时间是')
def tim(f):
	start = time.time() #记录起始时间
	time.sleep(2)   #让程序休眠2秒
	f()
	end = time.time()  #记录截至时间
	print(end - start) #计算时间差
tim(func)  #这里将函数func传入到tim函数从而实现对func函数的装饰功能

3.有没有可以直接执行函数就实现了对函数的装饰呢？

#装饰器函数的基本写法
def tim(f):
	def inner():
		start = time.time() #记录起始时间
		f()
		end = time.time()  #记录截至时间
		print(end - start) #计算时间差
	return inner
def func():
	 print('执行该该函数')
func = tim(func)#这样就是inner和func是一个相同的地址了
func() #这样就实现了上面的要求

4.还有一种更加简便的方式实现装饰器函数

def tim(f):
	def inner():
		start = time.time() #记录起始时间
		f()
		end = time.time()  #记录截至时间
		print(end - start) #计算时间差
	return inner

@tim    # func = tim(func) 这样就是inner和func是一个相同的地址了
def func():
	 print('执行该该函数')

func() #这样就实现了上面的要求

5.带有返回值的装饰器

import time
def tim(f):
    def inner():
        start = time.time()
        ret1 = f()
        end = time.time()
        print(ret1)
        print(end - start)
    return inner
@tim
def func():
    time.sleep(2)
    print('执行func函数')
    return 'func函数的返回值'
func()

#下面是输出结果

执行func函数
func函数的返回值
2.006819486618042

Process finished with exit code 0

6.func函数带参数的形式

import time
def tim(f):
    def inner(a,b): #注意这里传的是func函数所需要的参数
        start = time.time()
        ret1 = f(a,b) #注意这里就是上面inner里面的参数
        end = time.time()
        print(ret1)
        print(end - start)
    return inner
@tim
def func(a,b):
    time.sleep(2)
    print(a+b)
    return 'func函数的返回值'
func(12,2)

7.万能参数和有返回值形式的装饰器

import time
def tim(f):
    def inner(*args,**kwargs):
        start = time.time()
        ret = f(*args,**kwargs)
        end = time.time()
        print(ret)
        print(end - start)
    return inner
@tim
def func(*args,**kwargs):
    time.sleep(2)
    print('这里是func打印的内容')
    print(*args,**kwargs) #前面是可以传字符串，int数据类型的 后面是针对字典类型的数据{'s':2}
    return '这里是func函数返回的内容'
func('akjs',(1,5),2,{'s':2})

#输出形式
这里是func打印的内容
akjs (1, 5) 2 {'s': 2}
这里是func函数返回的内容
2.01223087310791

Process finished with exit code 0

总结装饰器的模板如下

装饰器的模板

def timmer(f):
    def inner(*args,**kwarge):
        '''被装饰函数执行之前要做的事'''
        ret = f(*args,**kwarge)
        '''被装饰函数执行之后要做的事'''
        return ret
    return inner

@timmer   # func = timmer(func)
def func(*args,**kwarge):
    time.sleep(0.01)
    print('老板哈',*args,**kwarge)
    return 'haha'

func(实参)