前言
给下面的函数加上运行时间
deffib(n):
a, b= 0, 1
for i inrange(n):print(b)
a, b= b, a+breturnb
a= fib(50)
修改一:改动函数
importtimedeffib(n):
start=time.time()
a, b= 0, 1
for i inrange(n):print(b)
a, b= b, a+bprint(time.time() -start)returnb
a= fib(50)
修改二:不改动函数
deffib(n):
a, b= 0, 1
for i inrange(n):print(b)
a, b= b, a+breturnbimporttimedefwrapper(func):def inner(*args, **kwargs):
start=time.time()
a= func(*args, **kwargs)print(time.time() -start)returnareturn inner #返回函数名
fib=wrapper(fib)
fib(50)
忙活了这么半天,终于初具规模了!现在已经基本上完美了,唯一碍眼的那句话就是还要在做一次赋值调用。。。
装饰器
装饰器的功能:在不修改原函数及其调用方式的情况下对原函数功能进行扩展
装饰器的本质:就是一个闭包函数
满足开放封闭原则
1.对扩展是开放的
2.对修改是封闭的
通过使用装饰器语法糖来解决这个问题!
importtimedefwrapper(func):def inner(*args, **kwargs):
start=time.time()
a= func(*args, **kwargs)print(time.time() -start)returnareturn inner #返回函数名
@wrapper#fib = wrapper(fib)
deffib(n):
a, b= 0, 1
for i inrange(n):print(b)
a, b= b, a+breturnb
fib(100)
上面的装饰器已经非常完美了,但是正常我们情况下查看函数的一些信息的方法在此处都会失效
importtimedefwrapper(func):def inner(*args, **kwargs):
start=time.time()
a= func(*args, **kwargs)print(time.time() -start)returnareturn inner #返回函数名
@wrapper#fib = wrapper(fib)
deffib(n):
a, b= 0, 1
for i inrange(n):print(b)
a, b= b, a+breturnb
fib(100)print(fib.__doc__)print(fib.__name__) #返回 inner
# 为了不让他们失效,我们还要在装饰器上加上一点来完善它:importtimefrom functools importwrapsdefwrapper(func):
@wraps(func)#加在最内层函数正上方
def inner(*args, **kwargs):
start=time.time()
a= func(*args, **kwargs)print(time.time() -start)returnareturn inner #返回函数名
@wrapper#fib = wrapper(fib)
deffib(n):
a, b= 0, 1
for i inrange(n):print(b)
a, b= b, a+breturnb
fib(100)print(fib.__doc__)print(fib.__name__) #返回 fib
带参数的装饰器
带参数的装饰器:就是给装饰器传参
用处:就是当加了很多装饰器的时候,现在忽然又不想加装饰器了,想把装饰器给去掉了,但是那么多的代码,一个一个的去闲的麻烦,
那么,我们可以利用带参数的装饰器去装饰它,这就他就像一个开关一样,要的时候就调用了,不用的时候就去掉了。给装饰器里面传个参数,
那么那个语法糖也要带个括号。在语法糖的括号内传参。在这里,我们可以用三层嵌套,弄一个标识 为去标识。如下面的代码示例
importtimefrom functools importwraps
flag=Falsedefouter(flag):defwrapper(func):
@wraps(func)#加在最内层函数正上方
def inner(*args, **kwargs):
start=time.time()ifflag:print('执行fib')
a= func(*args, **kwargs)else:print('不执行fib')
a= ''
print(time.time() -start)returnareturn inner #返回函数名
returnwrapper"""带参数的装饰器执行过程
fib = outer(False) (fib)
--分解
fib_decorator = outer(False)
fib = fib_decorator(fib)
--分解
fib_decorator = outer(False)
@fib_decorator
def fib(n)
所以,带参数的outer函数首先返回一个decorator函数,
再让这个decorator函数接收fib并返回新函数:"""@outer(True)deffib(n):
a, b= 0, 1
for i inrange(n):print(b)
a, b= b, a+breturnb
fib(100)print(fib.__doc__)print(fib.__name__) #返回 fib
多个装饰器叠加使用
defdecorator_a(func):print('Get in decorator_a')def inner_a(*args, **kwargs):print('Get in inner_a')return func(*args, **kwargs)returninner_adefdecorator_b(func):print('Get in decorator_b')def inner_b(*args, **kwargs):print('Get in inner_b')return func(*args, **kwargs)returninner_b
@decorator_b
@decorator_adeff(x):print('Get in f')return x * 2
print(f(1))
输出结果分析
"""装饰器执行顺序自下而上,内部函数调用顺序自上而下
所以输出
Get in decorator_a
Get in decorator_b
Get in inner_b
Get in inner_a
Get in f
2
分析
当解释器执行下面这段代码时,实际上按照从下到上的顺序已经依次调用了 decorator_a 和 decorator_b ,
这是会输出对应的 Get in decorator_a 和 Get in decorator_b 。
这时候 f 已经相当于 decorator_b 里的 inner_b 。但因为 f 并没有被调用,
所以 inner_b 并没有调用,依次类推 inner_b 内部的 inner_a 也没有调用,
所以 Get in inner_a 和 Get in inner_b 也不会被输出。
然后最后一行当我们对 f 传入参数1进行调用时, inner_b 被调用了,
它会先打印 Get in inner_b ,然后在 inner_b 内部调用了 inner_a
所以会再打印 Get in inner_a, 然后再 inner_a 内部调用的原来的 f,
并且将结果作为最终的返回。这时候你该知道为什么输出结果会是那样,
以及对装饰器执行顺序实际发生了什么有一定了解了吧。"""
装饰器使用举例
#!/usr/bin/env python#-*- coding: utf-8 -*-#@Time : 2018/4/4 22:38#@Author : hyang#@Site :#@File : 装饰器练习.py#@Software: PyCharm
"""三:编写装饰器,为函数加上认证的功能,即要求认证成功后才能执行函数
四:编写装饰器,为多个函数加上认证的功能(用户的账号密码来源于文件),
要求登录成功一次,后续的函数都无需再输入用户名和密码
提示:从文件中读出字符串形式的字典,可以用eval('{"name":"egon","password":"123"}')转成字典格式"""user_data={'user': None,'is_authenticated': False
}deflogin_required(func):#用户认证装饰器
def inner(*args, **kwargs):if args[0].get('is_authenticated'):print('User is authenticated')
ret= func(*args, **kwargs)else:
exit("User is not authenticated !!!")returnretreturninnerdefget_acct_info():"""get acct_info from USER.TXT
:return:"""with open(r'USER.TXT', 'r') as f:for line inf:
user_dict=eval(line)
USER_INFO_LI.append(user_dict)defauth_user():"""认证用户
:return:"""count=0while count < 3:
user= input('user >>>').strip()
pwd= input('pwd >>>').strip()#print(USER_INFO_LI)
user_flag =False
pwd_flag=Falsefor item inUSER_INFO_LI:if user == item['name']:
user_flag=Trueifuser_flag:if pwd == item['password']:
pwd_flag=Trueif user_flag andpwd_flag:#用户认证成功
user_data['user'] =user
user_data['is_authenticated'] =Trueprint('account login success')returnuser_dataelifuser_flag:#密码不对
print('pwd is error')else:#用户名不存在
print('user is not exists')
count+= 1
else:print("account [%s] too many login attempts" %user)
exit()
@login_requireddefprint_info(p_data):print(p_data)
@login_requireddefenter_x(p_data):print('enter_x', p_data['user'])
@login_requireddefenter_y(p_data):print('enter_x', p_data['user'])deflogout(p_data):print('logout', p_data['user'])
p_data['is_authenticated'] =Falsedefinteractive(p_data):"""interact with user
:return:"""menu= """------- menu ---------
1. 账户信息(功能已实现)
2. 进入x(功能已实现)
3. 进入y(功能已实现)
4. 退出程序"""
print(menu)
menu_dic={'1': print_info,'2': enter_x,'3': enter_y,'4': logout
}
exit_flag=Falsewhile notexit_flag:
user_option= input(">>:").strip()if user_option inmenu_dic:
menu_dic[user_option](p_data)else:print("Option does not exist!")if __name__ == '__main__':
USER_INFO_LI=[]
get_acct_info()
acc_data=auth_user()
interactive(acc_data)