关于Python装饰器的一些理解

今天写后台接口的登录逻辑的时候，遇到了对登录状态的处理，用户登录之后服务器返回给用户token，用户每次调用需要登录的API的时候，就可以根据这个token判断用户是否登录，登录是否过期啥子的了。但是每个路由都对token进行一次验证着实太麻烦了，，而我在java中知道有装饰者模式可以解决这个问题，，所以我查了下，发现python的装饰器也可以解决，而且它也是一个非常重要的东西，所以我就学了下，这里把自己目前的理解写出来，做个记录，纪念我充实的一天。。。。。。

准备知识

1.python函数中可以创建函数

对，没错，就是函数中创建函数，理论上可以一直嵌套

def test():print("来做个加法叭")x=1y=2def add(a,b):return a+bprint(add(x,y))test()

结果：

2.一切皆为对象，函数也是

顾名思义，一切都是对象，函数也是！


def test():return "一切都是对象！"a = test
print(type(a))
print(a())

结果：

我们把函数test（）当成对象赋给了a，可以发现，我们执行a时和执行test时，没啥区别。

3.函数可以当成参数

由上面可以知道，一切都可是对象，函数也是，，所以函数也可以当成别的函数的参数啦

def wrap(fun):print("嘿嘿，我在fun前面执行")print(fun())print("嘿嘿，我在fun后面执行")def my():return "我就是一个卑微的被当成参数的函数"wrap(my)

结果：

装饰器基本功能

装饰器的功能嘛，就是装饰用的呗，你看，一般的照片不都有个框子把它装起来么，框子上面都有花边撒，花边就是装饰用的。装饰器也是一样的，当你的函数执行之前要处理或者验证啥子事情的时侯，就可以用装饰器来标记啦。
就拿我开头说的，，我的每个API的函数执行之前，我都要知道它有没有收到token，，我要是每个函数前面都写一遍认证的代码，，我可能要疯。。所以只要用上装饰器，就ok啦。

装饰器的实现

结合刚刚的准备知识，我们可以发现把那3点特性结合起来，就可以实现装饰器的功能啦

def decorator(fun):    #接收要装饰的函数def wrap():        #在decorator中新建wrap函数，来执行逻辑，进行“装饰”print("嘿嘿，我在fun前面执行")print(fun())print("嘿嘿，我在fun后面执行")return wrap        #返回这个定义好的函数def my():return "我就是一个卑微的被当成参数的函数"a = decorator(my)
a()

结果

这样，我们就可以说，把my()函数装饰起来了。可是每次都这样写着实麻烦死了。
但是python里有快速的方法实现：@

def decorator(fun):    #接收要装饰的函数def wrap():        #在decorator中新建wrap函数，来执行逻辑，进行“装饰”print("嘿嘿，我在fun前面执行")print(fun())print("嘿嘿，我在fun后面执行")return wrap        #返回这个定义好的函数@decorator             #给my函数加上装饰器decorator
def my():return "我就是一个卑微的被当成参数的函数"my()

结果：

嘿嘿，是不是一样的效果？这样就可以大量的简化我们的代码了。
真是太妙了，简直是妙蛙种子进了米奇妙妙屋——妙到家了。

扩展

装饰器的蓝本规范

from functools import wraps
def decorator_name(f):@wraps(f)def decorated(*args, **kwargs):if not can_run:return "Function will not run"return f(*args, **kwargs)return decorated@decorator_name
def func():return("Function is running")can_run = True
print(func())
# Output: Function is runningcan_run = False
print(func())
# Output: Function will not run

同时它也可以写成类

from functools import wrapsclass logit():def __init__(self,x):print(x)def __call__(self,func):@wraps(func)def wrapped_function(*args, **kwargs):log_string = func.__name__ + " was called"print(log_string)return func(*args, **kwargs)return wrapped_function@logit("你好")
def a():print("AAAAA")a()

这里的__call__是python的类的一种特殊方法，和__init__类似，感兴趣的可以查下。它调用类名()的时侯触发。

它可以用来打印日志：

class logit():def __init__(self):self.count=0def __call__(self,func):@wraps(func)def wrapped_function(*args, **kwargs):self.count+=1log_string = func.__name__ + " was called,the count is"print(log_string,self.count)return func(*args, **kwargs)return wrapped_function@logit()
def a():print()while True:a()time.sleep(2)