标签:模块 概念 传参数 保留 before span sleep 插入 测试
装饰器的应用场景,较为经典的有插入日志、性能测试、事务处理等。 上代码:
def foo(): print(‘foo‘)
此段代码,如果我们想在不改变原有功能的基础上添加新功能,可以这样做:
def foo(): print(‘foo‘) def wrapper(func): def inner(): print(‘before‘) func() print(‘after‘) return inner foo=wrapper(foo) foo()
打印结果:
before
foo
after
Python提供了一个语法糖来降低字符输入量。
def wrapper(func): def inner(): print(‘before‘) func() print(‘after‘) return inner @wrapper #wrapper=wrapper(foo) def foo(): print(‘foo‘) foo()
打印结果:
before
foo
after
让我们来关注一下@wrapper的写法,在foo函数定义上加上这一行与另外写foo = wrapper(foo)完全等价,@并没有另外的魔力。
除了字符输入少了一些,还有一个额外的好处:这样看上去更有装饰器的感觉。 这样还不能满足我们的需求,我们想给foo函数传参数的话,如何处理?
上代码:
def wrapper(func): def inner(sth): print(‘before‘) func(sth) print(‘after‘) return inner @wrapper #wrapper=wrapper(foo) def foo(sth): print(sth) foo(‘apple‘)
打印结果:
before
apple
after
这个是传一个参数的情况,如果我们想传的参数很多,除了多个位置参数之外,还想要传关键字参数,这个时候怎么办呢? func函数通过*args,**kwargs接收实参即可完美实现!
def wrapper(func): def inner(*args,**kwargs): print(‘before‘) func(*args,**kwargs) print(‘after‘) return inner @wrapper #wrapper=wrapper(foo) def foo(sth1,sth2,sth3,sth4): print(sth1,sth2,sth3,sth4) foo(‘apple‘,‘orange‘,‘banana‘,‘lemon‘)
打印结果:
before
apple orange banana lemon
after
进一步,我们的foo函数有时候需要使用到返回值,然后如何处理呢?
import time def wrapper(func): def inner(*args,**kwargs): print(‘before‘) ret = func(*args,**kwargs) print(‘after‘) print(ret) return ret return inner @wrapper #foo=wrapper(foo) def foo(arg): print(arg) return 111 res=foo(‘apple‘) time.sleep(1) print(res) 打印结果: before apple after 111 111(1秒后打印)
此外,functools模块提供有装饰器:wraps(wrapped,assigned,updated)。
函数是有几个特殊属性比如函数名,在被装饰后,上例中的函数名foo会变成包装函数的名字inner,我们可以看下例:
def wrapper(func): def inner(*args, **kwargs): print("before...") ret = func(*args, **kwargs) print("after/...") return ret return inner @wrapper # foo = wrapper(foo) -->inner def foo(sth): print(sth) return "aaa" print(foo.__name__) 打印结果: inner
这时候wraps就体现了它的作用:它能将装饰过的函数的特殊属性保留。
import functools def wrapper(func): @functools.wraps(func) def inner(*args, **kwargs): print("before...") ret = func(*args, **kwargs) print("after/...") return ret return inner @wrapper # foo = wrapper(foo) -->inner def foo(sth): print(sth) return "aaa" print(foo.__name__) 打印结果: foo
关于装饰器,一些有意思的小练习可以拿来做。
一:编写函数,(函数执行的时间是随机的) 二:编写装饰器,为函数加上统计时间的功能 三:编写装饰器,为函数加上认证的功能 四:编写装饰器,为多个函数加上认证的功能(用户的账号密码来源于文件),要求登录成功一次,后续的函数都无需再输入用户名和密码 注意:从文件中读出字符串形式的字典,可以用eval(‘{"name":"egon","password":"123"}‘)转成字典格式 五:编写下载网页内容的函数,要求功能是:用户传入一个url,函数返回下载页面的结果 六:为题目五编写装饰器,实现缓存网页内容的功能: 具体:实现下载的页面存放于文件中,如果文件内有值(文件大小不为0),就优先从文件中读取网页内容,否则,就去下载,然后存到文件中 七:还记得我们用函数对象的概念,制作一个函数字典的操作吗,来来来,我们有更高大上的做法,在文件开头声明一个空字典,然后在每个函数前加上装饰器,完成自动添加到字典的操作
标签:模块 概念 传参数 保留 before span sleep 插入 测试
原文地址:http://www.cnblogs.com/metianzing/p/7740114.html