12 知识点总结 装饰器进阶
⼀. 通⽤装饰器的回顾 1.开闭原则: 对增加功能开放. 对修改代码封闭 2.装饰器的作⽤: 在不改变原有代码的基础上给⼀个函数增加功能 3.通⽤装饰器的写法: def wrapper(fn): def inner(*args,**kwargs): print("目标函数前一行") ret=fn(*args,**kwargs) print("目标函数后一行") return ret return inner @wrapper def target_func(*args,**kwargs): print("我是目标函数体") target_func() 4.执⾏过程: (1) 程序从上向下, 当执⾏到@wrapper的时候. 把函数作为参数传递给wrapper函数. 得到inner函数. 重新赋值给target_func (2) 当执⾏到target_func的时候. 我们实际上执⾏的是inner函数. inner函数会先执⾏⽬标函数之前的代码. 然后再执⾏你的⽬标函数. 执⾏完⽬标函数最后执⾏的是⽬标函数 之后的代码 ⼆. 函数的有⽤信息 1. 如何给函数添加注释 用三个引号来表示 def eat(food,drink): """ eat:把传递进来的吃掉 :param food: 参数food是什么意思 :param drink: 参数drink是什么意思 :return: None 返回什么 """ print(food,drink) return "very good" 按住ctrl 点内置函数名,可以查看函数的注释如int,str 等 2.如何获取到函数的相关信息 def eat(food,drink): """ :param food: 参数food是什么意思 :param drink: 参数drink是什么意思 :return: None 返回什么 """ print(food,drink) # print(eat.__name__) 读取不出来 # print(eat.__doc__) 读取不出来 return "very good" eat("水果","可乐") print(eat.__name__) # 查询函数名 print(eat.__doc__) #查询函数文档注释 函数名.__name__可以查看函数的名字 (双下划线) 函数名.__doc__ 可以查看函数的⽂档注释 (1) 一个被装饰器装饰过的函数:查询目标函数的函数名 def wrapper(fn): def inner(*args,**kwargs): # 聚合 print("目标函数前一行") ret=fn(*args,**kwargs) # 打散 这⾥的作⽤. 其实就是为了保证我可以装饰所有函数⽽准备的 print("目标函数后一行") return ret return inner @wrapper def target_func(*args,**kwargs): print("我是目标函数体") target_func() print(target_func.__name__) # 被装饰过的函数函数名是inner. (2) 把上述查询函数名修改为原函数名 from functools import wraps # 加 引入函数模块 def wrapper(fn): @wraps(fn) # 加 使用函数原来的名字 def inner(*args,**kwargs): print("目标函数前一行") ret=fn(*args,**kwargs) print("目标函数后一行") return ret return inner @wrapper def target_func(*args,**kwargs): print("我是目标函数体") target_func() print(target_func.__name__) # 查询结果不再是inner 而是target_func @wrapper def new_target_func(): print("我是另⼀个⽬标函数") new_target_func() print(new_target_func.__name__) 三.装饰器传参 def wrapper_out(flag): def wrapper(fn): def inner(*args,**kwargs): if flag==True: # 设定条件,满足执行下一步 print("目标函数前一行") ret=fn(*args,**kwargs) print("目标函数后一行") else: # 不满足执行这一步 ret=fn(*args,**kwargs) return ret return inner return wrapper @wrapper_out(True) def target_func(): print("我是目标函数体") target_func() # 目标函数前一行,我是目标函数体,目标函数后一行 @wrapper_out(False) def target_func(): print("我是目标函数体") target_func() # 我是目标函数体 执行步骤: 先执⾏wrapper(True) 然后再@返回值. 返回值恰好是wrapper. 结果就是@wrapper 四.多个装饰器装饰一个函数 def wrapper(fn): def inner(*args,**kwargs): print("我是a") ret=fn(*args,**kwargs) print("我是b") return ret return inner def wrapper1(fn): def inner(*args,**kwargs): print("我是c") ret=fn(*args,**kwargs) print("我是d") return ret return inner def wrapper2(fn): def inner(*args,**kwargs): print("我是e") ret=fn(*args,**kwargs) print("我是f") return ret return inner @wrapper2 @wrapper1 @wrapper def eat(*args,**kwargs): print("我是目标函数") eat() 执行步骤: 执⾏顺序: ⾸先@wrapper1装饰起来. 然后获取到⼀个新函数是wrapper1中的inner. 然后执⾏@wrapper2.这个时候. wrapper2装饰的就是wrapper1中的inner. 第⼆层装饰器前(第⼀层装饰器前(⽬标)第⼀层装饰器后)第⼆层装饰器后. 程序从左到右执⾏ 五.补充知识点 1枚举 (同时拿元素和索引) lst=["金","木","水","火","土"] for index,element in enumerate(lst): print(index,element) # 解构 for element in enumerate(lst): print(element) # 结果是元组(0,"金) 2.存盘 , 网络传输 s="提前祝大家端午节快乐" bys=s.encode("UTF-8") # 编码 print(bys) # 结果 字节 s1=bys.decode("UTF-8") # 解码 print(s1) # 结果 字符串 应用: 把UTF-8 转成 GBK bys=b'\xe6\x8f\x90' # 一个字符的utf-8 形式 s=bys.decode("UTF-8") # 用UTF-8 解码成 字符串s gbk=s.encode("GBK") # 再把字符串转化成GBK (两个字节) print(gbk) # 结果 b'\xcc\xe1' 作业:
1.给每个函数写一个记录日志的功能，功能要求：每一次调用函数之前，要将函数名称，时间节点记录到log的日志中
所需模块
import time
struct_time = time.localtime()# localtime() : localtime是 把从1970-1-1零点零分到当前时间系统所偏移的秒数时间转换为本地时间
print(time.strftime("%Y-%m-%d %H:%M:%S",struct_time))
# strftime是一种，根据格式化本地时间/日期，函数的功能将时间格式化，或者说格式化一个时间字符串
import time struct_time = time.localtime() print(time.strftime("%Y-%m-%d %H:%M:%S",struct_time)) import time def wrapper(fn): def inner(*args,**kwargs): struct_time = time.localtime() #时间 f=open("log",mode="a",encoding="UTF-8") # 文件 f.write(fn.__name__+"\t"+time.strftime("%Y-%m-%d %H:%M:%S", struct_time)+"\n") #文件名和和格式化后的时间添加到文件中 ret=fn(*args,**kwargs)# 目标函数 return ret return inner @wrapper def func1(*args,**kwargs): print("函数一") func1() 2.京东,淘宝登陆系统 # 1,用户有两套账号密码,一套为京东账号密码，一套为淘宝账号密码分别保存在两个文件中。 # 2,设置四个函数，分别代表 京东首页，京东超市，淘宝首页，淘宝超市。 # 3,启动程序后,呈现用户的选项为: # 1,京东首页 # 2,京东超市 # 3,淘宝首页 # 4,淘宝超市 # 5,退出程序 # 4,四个函数都加上认证功能，用户可任意选择,用户选择京东超市或者京东首页,只要输入一次京东账号和密码并成功,则这两个函数都可以任意访问; # 用户选择淘宝超市或者淘宝首页,只要输入一次淘宝账号和密码并成功,则这两个函数都可以任意访问. flag1 = True flag2 = True def wrapper_out(aaa): def wrapper(fn): def inner(*args,**kwargs): flag = True global flag1,flag2 if aaa=="jd" : while flag and flag1==True: user=input("请输入您的用户名:") pwd=input("请输入您的密码:") with open("jd-user",mode="r",encoding='utf-8')as f: for i in f.readlines(): username,password=i.strip().split(",") if user==username and pwd==password: print("登陆成功") flag = False flag1 = False break else: print("登陆失败") elif aaa=="tb": while flag and flag2==True: user2=input("请输入您的用户名:") pwd2=input("请输入您的密码:") with open("tb_user",mode="r",encoding="utf-8")as f: for i in f.readlines(): username2,password2=i.strip().split(",") if user2==username2 and pwd2==password2: print("登陆成功") flag=False flag2=False break else: print("登陆失败") ret=fn() return ret return inner return wrapper @wrapper_out("jd") def jd_index(): print("京东首页") @wrapper_out("jd") def jd_market(): print("京东超市") @wrapper_out("tb") def tb_index(): print("淘宝首页") @wrapper_out("tb") def tb_market(): print("淘宝超市") def logout(): print("退出程序") global flag1, flag2 flag1=False flag2=False def start(): with open("菜单",mode="r",encoding="utf-8")as f: for line in f: print(line.strip()) def choic(): while 1: choose=input("请输入序号;") if choose=="1": jd_index() elif choose=="2": jd_market() elif choose=="3": tb_index() elif choose=="4": tb_market() elif choose=="5": logout() flag=False start() choic() 3.明星投票系统: # 1. ⽤户注册. 将⽤户名和密码保存在⽂件中. # 2. 登录功能. 不需要三次登录. 正常反复⽆限登录就可以. 直到登录成功为⽌ # 3. ⽤户登录后可以查看所有明星信息. 明星信息保存在明星⽂件中. # 例如: # 1 林俊杰 1 # 2 周杰伦 0 # 3 ⿇花藤 0 # 第⼀位是序号, 第⼆位是名字, 第三位是得票数 # 4. 给明星进⾏投票 # 5. 每个⼈每24⼩时只能投⼀次票(扩展功能, 装饰器) # 思路: 每次⽤户投票之后. 都从⽂件中查询⼀下⽤户上⼀次投票时间. 让⽤户上⼀ # 次投票时间和当前系统时间做时间差. 如果⼤于24⼩时. 则可以进⾏投票. 如果⼩于24⼩时, # 则提⽰⽤户不能进⾏投票. 时间处理请⾃⾏百度. # 使⽤知识点: # 1. 函数, 请将以上功能尽可能的封装成函数. ⽤函数来描述每⼀个功能是我们以后开发 # 经常使⽤的⼀种写法和思路 import os import time def wrapper(fn): def inner(): t = time.time() t2 = os.path.getmtime("stra_info") if t - t2 <86400.0: print('24小时内不要重复登陆') return ret=fn() return ret return inner def start(): with open("main",mode="r",encoding="utf-8")as f: for line in f.readlines(): print(line.strip()) start() def regist(): with open("user_info",mode="r+",encoding="utf-8")as f1: for i in f1.readlines(): user=input("请输入用户名:") pad=input("请输入密码:") if user==i.strip().split(",")[0]: print("该用户名已存在") else: with open("user_info",mode="a",encoding="utf-8")as f2: f2.write("\n"+user+","+pad) print("注册成功") return # regist() def logon (): while 1: username1 = input("请输入用户名:") password1 = input("请输入密码:") with open("user_info",mode="r",encoding="utf-8")as f3: for i in f3.readlines(): user1,pwd1=i.strip().split(',') if username1==user1 and password1==pwd1: print("登陆成功") return else: print("登陆失败请重新登陆") # logon() def star_info(): with open("star_info", mode="r",encoding="utf-8")as f4: for line in f4.readlines(): print(line.strip()) @wrapper def vote(): vote_num=input('请输入明星id:') with open("stra_info",mode="r",encoding="utf-8")as f5,open("star_info1",mode="w",encoding="utf-8")as f6: for line in f5.readlines(): num,name,poll=line.strip().split(" ") if vote_num==num: f6.write(num+" "+name+" "+str(int(num)+1)+'\n') else: f6.write(line) os.remove("stra_info") os.rename("star_info1","stra_info") def logout(): print('您已退出系统') def choice(): while 1: choose=input('请输入项目序号:') if choose=="1": regist() elif choose=="2": logon() elif choose=="3": star_info() elif choose=="4": vote() elif choose=="5": logout() start() choice()