装饰器
装饰器就是闭包函数的一种应用场景
一 为何要用装饰器
#开放封闭原则:对修改封闭,对扩展开放
二 什么是装饰器
装饰器他人的器具,本身可以是任意可调用对象,被装饰者也可以是任意可调用对象。
强调装饰器的原则:1 不修改被装饰对象的源代码 2不修改被装饰对象的调用方式
装饰器的目标:在遵循1和2的前提下,为被装饰对象添加上新功能
三 装饰器的使用
importtimedeftimmer(func):def wrapper(*args,**kwargs):
start_time=time.time()
res=func(*args,**kwargs)
stop_time=time.time()print('run time is %s' %(stop_time-start_time))returnresreturnwrapper
@timmerdeffoo():
time.sleep(3)print('from foo')
foo()
无参装饰器
def auth(driver='file'):defauth2(func):def wrapper(*args,**kwargs):
name=input("user:")
pwd=input("pwd:")if driver == 'file':if name == 'egon' and pwd == '123':print('login successful')
res=func(*args,**kwargs)returnreselif driver == 'ldap':print('ldap')returnwrapperreturnauth2
@auth(driver='file')deffoo(name):print(name)
foo('egon')
有参装饰器
四 装饰器语法
被装饰函数的正上方,单独一行
@deco1
@deco2
@deco3deffoo():passfoo=deco1(deco2(deco3(foo)))
五 装饰器补充:wraps
from functools importwrapsdefdeco(func):
@wraps(func)#加在最内层函数正上方
def wrapper(*args,**kwargs):return func(*args,**kwargs)returnwrapper
@decodefindex():'''哈哈哈哈'''
print('from index')print(index.__doc__)
六 练习题
一:编写函数,(函数执行的时间是随机的)
二:编写装饰器,为函数加上统计时间的功能
三:编写装饰器,为函数加上认证的功能
四:编写装饰器,为多个函数加上认证的功能(用户的账号密码来源于文件),要求登录成功一次,后续的函数都无需再输入用户名和密码
注意:从文件中读出字符串形式的字典,可以用eval('{"name":"egon","password":"123"}')转成字典格式
五:编写装饰器,为多个函数加上认证功能,要求登录成功一次,在超时时间内无需重复登录,超过了超时时间,则必须重新登录
六:编写下载网页内容的函数,要求功能是:用户传入一个url,函数返回下载页面的结果
七:为题目五编写装饰器,实现缓存网页内容的功能:
具体:实现下载的页面存放于文件中,如果文件内有值(文件大小不为0),就优先从文件中读取网页内容,否则,就去下载,然后存到文件中
扩展功能:用户可以选择缓存介质/缓存引擎,针对不同的url,缓存到不同的文件中
八:还记得我们用函数对象的概念,制作一个函数字典的操作吗,来来来,我们有更高大上的做法,在文件开头声明一个空字典,然后在每个函数前加上装饰器,完成自动添加到字典的操作
九 编写日志装饰器,实现功能如:一旦函数f1执行,则将消息2017-07-21 11:12:11 f1 run写入到日志文件中,日志文件路径可以指定
注意:时间格式的获取
import time
time.strftime('%Y-%m-%d %X')
#题目一:略#题目二:略#题目三:略#题目四:
db='db.txt'login_status={'user':None,'status':False}def auth(auth_type='file'):defauth2(func):def wrapper(*args,**kwargs):if login_status['user'] and login_status['status']:return func(*args,**kwargs)if auth_type == 'file':
with open(db,encoding='utf-8') as f:
dic=eval(f.read())
name=input('username:').strip()
password=input('password:').strip()if name in dic and password ==dic[name]:
login_status['user']=name
login_status['status']=True
res=func(*args,**kwargs)returnreselse:print('username or password error')elif auth_type == 'sql':pass
else:pass
returnwrapperreturnauth2
@auth()defindex():print('index')
@auth(auth_type='file')defhome(name):print('welcome %s to home' %name)#index()#home('egon')
#题目五
importtime,random
user={'user':None,'login_time':None,'timeout':0.000003,}deftimmer(func):def wrapper(*args,**kwargs):
s1=time.time()
res=func(*args,**kwargs)
s2=time.time()print('%s' %(s2-s1))returnresreturnwrapperdefauth(func):def wrapper(*args,**kwargs):if user['user']:
timeout=time.time()-user['login_time']if timeout < user['timeout']:return func(*args,**kwargs)
name=input('name>>:').strip()
password=input('password>>:').strip()if name == 'egon' and password == '123':
user['user']=name
user['login_time']=time.time()
res=func(*args,**kwargs)returnresreturnwrapper
@authdefindex():
time.sleep(random.randrange(3))print('welcome to index')
@authdefhome(name):
time.sleep(random.randrange(3))print('welcome %s to home' %name)
index()
home('egon')#题目六:略#题目七:简单版本
importrequestsimportos
cache_file='cache.txt'
defmake_cache(func):def wrapper(*args,**kwargs):if notos.path.exists(cache_file):
with open(cache_file,'w'):pass
ifos.path.getsize(cache_file):
with open(cache_file,'r',encoding='utf-8') as f:
res=f.read()else:
res=func(*args,**kwargs)
with open(cache_file,'w',encoding='utf-8') as f:
f.write(res)returnresreturnwrapper
@make_cachedefget(url):returnrequests.get(url).text#res=get('https://www.python.org')
#print(res)
#题目七:扩展版本
importrequests,os,hashlib
engine_settings={'file':{'dirname':'./db'},'mysql':{'host':'127.0.0.1','port':3306,'user':'root','password':'123'},'redis':{'host':'127.0.0.1','port':6379,'user':'root','password':'123'},
}def make_cache(engine='file'):if engine not inengine_settings:raise TypeError('egine not valid')defdeco(func):defwrapper(url):if engine == 'file':
m=hashlib.md5(url.encode('utf-8'))
cache_filename=m.hexdigest()
cache_filepath=r'%s/%s' %(engine_settings['file']['dirname'],cache_filename)if os.path.exists(cache_filepath) andos.path.getsize(cache_filepath):return open(cache_filepath,encoding='utf-8').read()
res=func(url)
with open(cache_filepath,'w',encoding='utf-8') as f:
f.write(res)returnreselif engine == 'mysql':pass
elif engine == 'redis':pass
else:pass
returnwrapperreturndeco
@make_cache(engine='file')defget(url):returnrequests.get(url).text#print(get('https://www.python.org'))
print(get('https://www.baidu.com'))#题目八
route_dic={}defmake_route(name):defdeco(func):
route_dic[name]=funcreturndeco
@make_route('select')deffunc1():print('select')
@make_route('insert')deffunc2():print('insert')
@make_route('update')deffunc3():print('update')
@make_route('delete')deffunc4():print('delete')print(route_dic)#题目九
importtimeimportosdeflogger(logfile):defdeco(func):if notos.path.exists(logfile):
with open(logfile,'w'):pass
def wrapper(*args,**kwargs):
res=func(*args,**kwargs)
with open(logfile,'a',encoding='utf-8') as f:
f.write('%s %s run\n' %(time.strftime('%Y-%m-%d %X'),func.__name__))returnresreturnwrapperreturndeco
@logger(logfile='aaaaaaaaaaaaaaaaaaaaa.log')defindex():print('index')
index()
View Code