python笔记之装饰器

python中函数的参数也可以是函数，在多个函数需要实现相同代码时也可以考虑使用装饰器进行处理。

假设以下场景，需要进行查询及查看个人主页操作，那么在此操作之前一般需要先登录系统，此时就可以将登录函数写成装饰器函数

def login(fun): #装饰器函数格式，函数内层在定义一个函数，同时函数返回内层函数本身，相当于是执行内层函数msg
    def msg():
        print('登录成功')
        fun()
    return msg

@login    #装饰器login
def search1():  #一旦函数被装饰，自动执行装饰器函数login,且该函数相当于装饰器的参数
    print('查询成功')

@login
def profile():
    print('欢迎你查看个人主页')

以上代码，login为装饰器函数，search1、profile为被装饰器函数，详细的解释这个代码的意思:
1、自动执行login函数并且将被装饰的函数当作login函数的参数来处理
2、将login函数的返回值msg重新赋值给被装饰的函数
3、一旦结合了装饰器，调用被装饰的函数本质上是执行msg()函数
4、函数被装饰之后，被装饰的函数重新赋值给装饰器的内层函数

下面看一下被装饰函数有参数的情况

def loginxt(fun):
    def msg(*args,**kwargs):
        r = fun(*args,**kwargs)
        if r:
            return r
    return msg

@loginxt
def order(name,pwd):
    if name=='admin' and pwd=='123456':
        return ('欢迎你登录系统')
    else:
        return ('请先登录系统')


print(order('admin','123456'))