python基础装饰器_python基础之装饰器

装饰器

装饰器就是闭包函数的一种应用场景

一 为何要用装饰器

#开放封闭原则：对修改封闭，对扩展开放

二 什么是装饰器

装饰器他人的器具，本身可以是任意可调用对象，被装饰者也可以是任意可调用对象。

强调装饰器的原则：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