一：编写函数，（函数执行的时间用time.sleep(n)模拟） 二：编写装饰器，为函数加上统计时间的功能 三：编写装饰器，为函数加上认证的功能 四：编写装饰器，为多个函数加上认证的功能（用户的账号密码来源于文件），要求登录成功一次，后续的函数都无需再输入用户名和密码 注意：从文件中读出字符串形式的字典，可以用eval('{"name":"egon","password":"123"}')转成字典格式 五：编写装饰器，为多个函数加上认证功能，要求登录成功一次，在超时时间内无需重复登录，超过了超时时间，则必须重新登录 六：选做题 # 思考题（选做），叠加多个装饰器，加载顺序与运行顺序，可以将上述实现的装饰器叠加起来自己验证一下 # @deco1 # index=deco1(deco2.wrapper的内存地址) # @deco2 # deco2.wrapper的内存地址=deco2(deco3.wrapper的内存地址) # @deco3 # deco3.wrapper的内存地址=deco3(index) # def index(): # pass
# 一：编写函数，（函数执行的时间用time.sleep(n)模拟） ''' import time # 源码 def func(): time.sleep(3) # 单位: 秒 func() ''' # 二：编写装饰器，为函数加上统计时间的功能 import time # 源码 # 统计时间装饰器 def timer(func): # func ---> 被装饰对象 def inner(*args, **kwargs): # 被装饰对象需要接收的参数 # 被装饰对象执行之前的时间 start_time = time.time() # *args, **kwargs将接收过来的参数，传给调用的被装饰对象，会得到一个返回值 res = func(*args, **kwargs) # 调用被装饰对象， 这行代码就是被装饰对象的执行全过程 # 被装饰对象执行结束后的时间 end_time = time.time() # 打印被装饰对象的执行时间 # func.__name__ ---> 可以获取函数的名称 print(f'当前被装饰对象[{func.__name__}] 执行时间为: [{end_time - start_time}]') return res return inner @timer # timer = inner = timer(func) def func_demo(): time.sleep(3) # 单位: 秒 # 使用装饰器的目的: 1、为被装饰对象添加新的功能 2、前提: 不修改被装饰对象的源代码, 不修改调用方式 # inner = timer(func) # func = inner # func() # --> inner() # func_demo() # 三：编写装饰器，为函数加上认证的功能 # 全局变量: 用于记录是否有用户登录 login_user = None # 登录功能 def login(): username = input('请输入用户名: ').strip() password = input('请输入密码: ').strip() if username == 'tank' and password == '123': print('登录成功') # 登录成功后，给全局变量赋值，记录当前用户已经登录 global login_user login_user = username else: print('登录失败') # 登录认证装饰器 def login_auth(func): def inner(*args, **kwargs): # 被装饰对象执行前，为其添加新的功能 # 在此处做登录认证 if login_user: # 被装饰对象执行 func ---> play, shopping res = func(*args, **kwargs) return res # 被装饰对象执行后，为其添加新的功能 else: print('未有用户登录，无法使用好玩的功能!') login() return inner # play 与 shopping 功能使用前 必须先登录 否则，不允许使用，让其执行登录功能 @login_auth def play(): # 不使用装饰器 # if not login_user: # login() print('开始play了啊....') pass @login_auth def shopping(): pass # login() # # play() # shopping() # 四：编写装饰器，为多个函数加上认证的功能（用户的账号密码来源于文件），要求登录成功一次，后续的函数都无需再输入用户名和密码 # 注意：从文件中读出字符串形式的字典，可以用eval('{"name":"egon","password":"123"}')转成字典格式 # 1、打开db.txt文件，读取用户数据，转成字典类型 with open('db.txt', 'r', encoding='utf-8') as f: # 直接读取文件中的数据，得到的数据类型是: str user_data = f.read() print(user_data, type(user_data)) ''' {"name": "tank", "pwd": "123"} ''' # eval(str) ---> 从str字符串中，检测python代码的语法， # 如果里面有{key: value}，会将该数据生成对应的类型的内存地址 user_dic = eval(user_data) print(user_dic, type(user_dic)) user_dic = user_dic # print(user_dic) # print(type(user_dic)) login_info = None def login(): username = input('请输入用户名: ').strip() if username in user_dic.get('name'): password = input('请输入密码: ').strip() if password == user_dic.get('pwd'): print('login successful') global login_info login_info = {username: password} else: print('密码错误') else: print('用户不存在') def login_auth(func): def inner(*args, **kwargs): if login_info: res = func(*args, **kwargs) return res else: login() return inner @login_auth def withdraw(): pass @login_auth def shopping(): pass # 五：编写装饰器，为多个函数加上认证功能，要求登录成功一次，在超时时间内无需重复登录，超过了超时时间，则必须重新登录 import time # 全局变量，用于计算函数登录后的执行总时长 sum_time = 0 # ---》 func1 --> 3 ---> 6 def time_wait(func): # 登录成功后，可以使用的功能时长，单位 秒 wait_time = 6 def inner(*args, **kwargs): # 执行前时间 start_time = time.time() res = func(*args, **kwargs) # 执行后时间 end_time = time.time() global sum_time # 第一个函数进来后，func1 执行时长为3, func2 执行时长为3 sum_time += (end_time - start_time) print(f'登录后执行的总时长: {sum_time}') # func1执行的时长 >= wait_time if sum_time >= wait_time: # 需要重新去登录 login() sum_time = 0 else: return res return inner @time_wait def func1(): time.sleep(3) @time_wait def func2(): time.sleep(3) # func1() # func2() # # # 此时总时长为6秒 # func1() # 六：选做题 # 思考题（选做），叠加多个装饰器，加载顺序与运行顺序，可以将上述实现的装饰器叠加起来自己验证一下 # @deco1 # index=deco1(deco2.wrapper的内存地址) # @deco2 # deco2.wrapper的内存地址=deco2(deco3.wrapper的内存地址) # @deco3 # deco3.wrapper的内存地址=deco3(index) # def index(): # pass ''' def wrapper1(func): def inner1(*args, **kwargs): res = func(*args, **kwargs) return res return inner1 def wrapper2(func): def inner2(*args, **kwargs): res = func(*args, **kwargs) return res return inner2 def wrapper3(func): def inner3(*args, **kwargs): res = func(*args, **kwargs) return res return inner3 @wrapper1 @wrapper2 @wrapper3 def index(): pass '''