Python---装饰器详解
2017-10-24 本文已影响29人
coderST
定义：
本质上是一个函数。作用是用来装饰另一个函数（即被装饰函数），给被装饰函数添加功能。前提是不能改变被装饰函数的源代码和调用方式。这样的一个函数称之为装饰器。
解析：
下面我们话不多说，直接用代码说明。下面是一个函数。
1 def add():
2 b=1+2
3 print(b)
4
5 add（）
程序输出：
————————
3
————————
- 现在我要给这个函数增加一个解释性的句子，如下，我们可以编写一个装饰器：
#原函数
def add():
a=1+2
print(a)
#装饰器
def decorator(func):
def warpper():
print("1+2的结果是：")
func()
return warpper
#注意此句
add=decorator(add)
#调用函数
add()
程序输出：
——————————
1+2的结果是：
3
——————————
- 这样我们就成功的达成了我们的目的。这里要注意第12行的这一句，这一句是将add这个函数对象传入了decorator（）函数，返回的是一个新函数变量，这个新函数对象又重新赋值给add，这样就可以保证不改变被装饰函数的调用方式不变。在Python语法中有一种更优雅的方式可以代替第十二行的语句。如下：
#装饰器
def decorator(func):
def warpper():
print("1+2的结果是：")
func()
return warpper
#add=decorator(add)
#原函数
@decorator#换成@符号
def add():
a=1+2
print(a)
#调用函数
add()
在被装饰函数前面直接加上“@xxx”（xxx为装饰器函数名）即可
被装饰函数有参数怎么办？
- 如果被装饰器函数有参数呢？该怎们班？不用担心，我们可以用不定参数的形式来收集参数。实例代码如下：
def decorator(func):
def warpper(*args,**kwargs):
print("相加的结果是：")
func(*args,**kwargs)
return warpper
@decorator
def add(x,y):
a=x+y
print(a)
add(2,3)
程序输出：
——————————————————
相加的结果是：
5
——————————————————
如上，我们给包装函数加上接收参数，然后传给func（）函数就行了。这样不管被装饰函数有怎样的参数都不怕了。
下面写一个页面验证的装饰器。
- 大家知道有些网站的一部分页面是要求用户登录之后才可以访问的，比如下面的三个函数（分别代表三个页面）：
def index():
print("welcome to the index page")
def home():
print("welcome to the home page")
def bbs():
print("welcome to the bbs page")
return "I am the return contents"
- 假如说现在我们要给home页面和bbs页面加上验证，显然现在更改源代码是不可行的。这个时候我们可以用装饰器，如下:
username,passwd="jack","abc123"#模拟一个已登录用户
def decorator(func):
def warpper(*args,**kwargs):
Username=input("Username:").strip()
password=input("Password:").strip()
if username==Username and passwd==password:
print("Authenticate Success!")
func(*args,**kwargs)
else:
exit("Username or password is invalid！")
return warpper
def index():
print("welcome to the index page")
@decorator
def home():
print("welcome to the home page")
@decorator
def bbs():
print("welcome to the bbs page")
return "I am the return contents"
index()
home()
bbs()
程序结果：
————————
welcome to the index page #index页面未验证直接可以登入
Username:jack
Password:abc123
Authenticate Success! #登录的而情形
welcome to the home page
Username:jack #密码或用户名错误的情形
Password:123
Username or password is invalid！
————————
- 我们注意到bbs（）是有返回值的，如果我们把上述代码的最后一句（第25行）改为“print（bbs（））”之后再看看他的输出结果：
————————
welcome to the index page
Username:jack
Password:abc123
Authenticate Success!
welcome to the home page
Username:jack
Password:abc123
Authenticate Success!
welcome to the bbs page
None #返回值能么成None了？？？
————————
-
What happened！ bbs（）的返回值打印出来竟然是None。怎么会这样？这样的话不就改变了被装饰函数的源代码了吗？怎样才能解决呢？
-
我们来分析一下：
-
我们执行bbs函数其实就相当于执行了装饰器里的wrapper函数，仔细分析装饰器发现wrapper函数却没有返回值，所以为了让他可以正确保证被装饰函数的返回值可以正确返回，那么需要对装饰器进行修改：
-
username,passwd="jack","abc123"#模拟一个已登录用户
def decorator(func):
def warpper(*args,**kwargs):
Username=input("Username:").strip()
password=input("Password:").strip()
if username==Username and passwd==password:
print("Authenticate Success!")
return func(*args,**kwargs)#在这里加一个return就行了
else:
exit("Username or password is invalid！")
return warpper
def index():
print("welcome to the index page")
@decorator
def home():
print("welcome to the home page")
@decorator
def bbs():
print("welcome to the bbs page")
return "I am the return contents"
index()
home()
bbs()
- 如图加上return就可以解决了。下面我们在看看改后的程序输出：
————————
welcome to the index page
Username:jack
Password:abc123
Authenticate Success!
welcome to the home page
Username:jack
Password:abc123
Authenticate Success!
welcome to the bbs page
I am the return contents #bbs（）的返回值得到了正确的返回
——-——————
好了，返回值的问题解决了.
既然装饰器是一个函数，那装饰器可以有参数吗？
- 答案是肯定的。我们同样可以给装饰器加上参数。比如还是上面的三个页面函数作为例子，我们可以根据不同页面的验证方式来给程序不同的验证，而这个验证方式可以以装饰器的参数传入，这样我们就得在装饰器上在嵌套一层函数 了：
username,passwd="jack","abc123"#模拟一个已登录用户
def decorator(auth_type):
def out_warpper(func):
def warpper(*args,**kwargs):
Username=input("Username:").strip()
password=input("Password:").strip()
if auth_type=="local":
if username==Username and passwd==password:
print("Authenticate Success!")
return func(*args,**kwargs)
else:
exit("Username or password is invalid！")
elif auth_type=="unlocal":
print("HERE IS UNLOCAL AUTHENTICATE WAYS")
return warpper
return out_warpper
def index():
print("welcome to the index page")
@decorator(auth_type="local")
def home():
print("welcome to the home page")
@decorator(auth_type="unlocal")
def bbs():
print("welcome to the bbs page")
return "I am the return contents"
index()
home()
bbs()
输出：
————————
welcome to the index page
Username:jack
Password:abc123
Authenticate Success!
welcome to the home page
Username:jack
Password:abc123
HERE IS UNLOCAL AUTHENTICATE WAYS
————————
-
可见，程序分别加入了第2行和第16行和中间的根据auth_type参数的判断的相关内容后， 就解决上述问题了。对于上面的这一个三层嵌套的相关逻辑，大家可以在 pycharm里头加上断点，逐步调试，便可发现其中的道理。
-
总结
-
要想学好迭代器就必须理解一下三条：
-
1.函数即变量（即函数对象的概念）
-
2.函数嵌套
-
3.函数式编程
-
