单独记录装饰器这个知识点是因为这个知识点是非常重要的，必须掌握的（代码大约150行）。
了解装饰器之前要知道三个知识点
- 作用域，上一篇讲到过顺序是L->E->G->B
- 高阶函数：
-
满足下列其中一种即可称之为高阶函数：
a、把一个函数名当做实参传递给另一个函数
b、返回值中包含函数名
3.闭包
闭包是由函数及其相关的引用环境组合而成的实体(即：闭包=函数+引用环境)，通俗讲法是：在一个外函数中定义了一个内函数，内函数里运用了外函数的临时变量，并且外函数的返回值是内函数的引用。这样就构成了一个闭包。，具体看下面代码：
n = 10 # 定义全局作用域变量 def fn(): # 形成闭包
n = 100 # 定义局部变量n def inner():
nonlocal n
n += 1 # 这里定义操作相同变量n无法调用上层作用中的变量，如果只读不写则可以正常访问
# python3 中新增nonlocal 关键字可以调用上层作用域中的变量
print(n) return inner # 返回内嵌函数的地址，从而形成闭包
# 形成闭包的条件
# 1、必须要有一个内嵌函数
# 2、内嵌函数中要对*变量的引用
# 3、外部函数必须返回内嵌函数 t = fn()
t() ->101
print(n) ->10
这里要插播一下，刚刚看到群里面有一道题目，需求如下：
需要将str1 = (('a'),('b')),(('c'),('d')) 变成 [{'a': 'c'}, {'b': 'd'}]
可以先尝试下：
给个提示，用到zip和map，上一篇已经说过map了，这次来补充一下zip是个啥，举个例子给你看下：
#zip() 函数用于将可迭代的对象作为参数，将对象中对应的元素打包成一个个元组，然后返回由这些元组组成的列表
s1 = (('a'),('b'))
s2 = (('c'),('d'))
s3 = zip(s1, s2)
print(list(s3)) ->结果是：[('a', 'c'), ('b', 'd')]
所以应该知道怎么玩了吧，接下来就是答案了：
str1 = (('a'),('b')),(('c'),('d'))
z = zip(str1[0], str1[1]) #返回的是一个迭代器
b = list(map(lambda x: {x[0]: x[1]} , z)) #map这里做的就是将zip类的两个元祖拿出来然后进行元祖变成字典的转换（这个转换不错值得学习），返回的也是一个迭代器，所以用list返回成列表
print(b)
好了，言归正传哈，了解了上面的三个知识点后，接着就来说说装饰器，装饰器的作用主要是啥呢？装饰器就是对已经存在的对象添加新的功能
现在我们一步一步来，原先我们已经有了一些功能了，假设像下面的
#业务中的某一个功能
def f1():
print('f1...')
此时我们有一个需求，就是需要计算这个函数执行花费的时间，那怎么玩呢？直接修改函数吗？那如果成千上百的函数都需要呢？所以不能直接修改，这样我们就重新定义一个函数
import time def count_time(f):
start_time = time.time()
f()
time.sleep(0.1) #由于执行时间太短，所以加了0.1秒
end_time = time.time()
print(end_time-start_time) def f1():
print('f1...') count_time(f1)
似乎已经达到了效果，但是又有一个问题来了，你想想原本大家都是通过f1去调用的，现在是不是都得改成count_time了，别人是不是得砍死你呢？
是不是我们得想办法既加上功能又还是以前的名称调用呢？这个就是装饰器的由来
import time
def count_time(f):
def inner():
start_time = time.time()
f()
time.sleep(0.1)
end_time = time.time()
print(end_time-start_time)
return inner def f1():
print('f1...') f1 = count_time(f1) # 右边就是返回执行时间统计和业务程序的内存对象地址，又将这个对象地址赋值给f1，这样调用f1就是原来功能加上新功能了
f1()
函数count_time就是装饰器，它把真正的业务方法包裹在函数里面，看起来像f1被上下时间函数装饰了，但是每次写f1 = count_time(f1)还是很麻烦的，因此装饰器有相应的语法就是@count_time
@符号是装饰器的语法糖，在定义函数的时候使用，避免再一次赋值操作，像下面一样
import time def count_time(f):
def inner():
start_time = time.time()
f()
time.sleep(0.1)
end_time = time.time()
print(end_time-start_time)
return inner @count_time #就等同于f1 = count_time(f1)，也就是将inner的引用对象引给了f1
def f1():
print('f1...') f1()
接着又有新的问题了，啥问题呢？像f1这种都是无参数的函数，那么如果有多个参数咋办呢？此时我们就需要加上不定长参数
import time def count_time(f):
def inner(*args, **kwargs):
start_time = time.time()
result = f(*args, **kwargs)
time.sleep(0.1)
end_time = time.time()
print(end_time-start_time)
return result
return inner @count_time #就等同于f1 = count_time(f1)
def f1(*args, **kwargs):
sum_num = 0
for i in args:
sum_num += i
return sum_num sum_num = f1(1,2,3)
print(sum_num)
呼，似乎终于写完了，但是呢产品又跑过来，说现在有些函数又要额外加些新功能，比如这些需要加一个日志打印功能，另外的函数又不用，那我们怎么办呢？
我们的思路肯定是将功能添加到count_time函数内但是呢我们需要再传入一个参数作为打印日志功能的开关，此时我们就会想到是不是装饰器可以传入参数呢？
我们发现f1 = count_time(f1)，这里并没有传入参数的入口，那咋么办呢？
import time def logger(flag=0):
def count_time(f):
def inner(*args, **kwargs):
start_time = time.time()
result = f(*args, **kwargs)
time.sleep(0.1)
end_time = time.time()
print(end_time-start_time)
if flag:
print("打印日志")
return result
return inner
return count_time @logger(1) # 因为logger(1)实际上是做了执行操作，所以执行的结果就是将count_time装饰器返回，然后还保留了flag参数，接着就是@count_time了也就是f1=count_time(f1)
def f1(*args, **kwargs):
sum_num = 0
for i in args:
sum_num += i
return sum_num sum_num = f1(1,2,3)
print(sum_num)
上面的logger是允许带参数的装饰器。它实际上是对原有装饰器的一个函数封装，并返回一个装饰器(一个含有参数的闭包函数)。
如果没有如果没有使用@语法，等同于 # f1 = logger(1)(f1)，这下应该明白了吧
学完了这些之后呢，就来做一个作业，就是用装饰器做一个简单的登录判断
qq文本中存了这些账户信息（一般之后都是存到数据库的）
# -*- coding:utf-8 -*-
# @__author__ : Loris
# @Time : 2018/11/6 10:31 """
需求：用装饰器的知识做一个登录功能 要求：
1、给用户展现三个菜单，home、book、clothes
2、用户点击了其中菜单后，判断用户是否已经登录，如果登录了就直接进入到菜单内，如果没登录就让用户选择时用微信还是qq登录
3、qq和微信的账号和密码存在文本中，校验通过后才可以进入，不通过则提示用户输入有误！ """ # 读取账户信息
def read_file(file_name):
user_msg = [] # 存储信息
with open(file_name, encoding="utf-8") as f_msg:
f_msg.readline() # 第一行去掉
for i in f_msg:
user_msg.append(i.strip().split(",")) # 通过,分割，将用户名和密码变成一个列表
user_msg = dict(user_msg) # 将读取出来的信息变成字典
return user_msg #判断三次密码错误就退出
def is_login(f, user_msg):
global login_status
global first_flag
count = 3 # 总共三次
while count > 0:
username = input("username:")
passwd = input("password:")
if username in user_msg:
pswd = user_msg[username]
if passwd == pswd:
print("你已经成功登录")
login_status = True
f()
break
else:
print("账户名和密码错误")
count -= 1
else:
print("账户名和密码错误")
count -= 1
if count == 0:
first_flag = 0 # 定义登录装饰器
def login_style(style="qq"):
def login(f):
def inner():
if not login_status:
if style == "qq":
# 从文件中读取用户信息
user_msg = read_file("qq")
is_login(f, user_msg)
elif style == "weixin":
user_msg = {"liu":"", "long":"haha"}
is_login(f, user_msg)
else:
f()
return inner
return login #定义三个菜单栏内的功能 @login_style("weixin")
def home():
print("我是首页") @login_style()
def book():
print("书本页面") @login_style()
def clothes():
print("衣服页面") # 定义一个全局参数作为是否已经登录的标识位，默认为没登陆
login_status = False #用户的三个菜单栏
menu_list = ["home", "book", "clothes"] first_flag = 1 #定义主函数
def main():
while first_flag:
#首先是循环打印菜单栏
print("菜单栏如下：")
for menu in menu_list:
print(menu) # 用户选择
choice_menu = input("please choice:") # 判断用户选择是否正确
if choice_menu == menu_list[0]:
eval(menu_list[0])()
elif choice_menu == menu_list[1]:
eval(menu_list[1])()
elif choice_menu == menu_list[2]:
eval(menu_list[2])()
else:
print("请重新输入") if __name__ == "__main__":
main()