python 装饰器

一.开闭原则:
对功能的扩展开放
对代码的修改是封闭

二.装饰器:

1.通用装饰器语法:

def wrapper(fn):
def inner(*args, **kwargs): # 聚合
在目标函数之前
ret = fn(*arg, **kwargs) # 打散
在目标函数之后
return ret
return inner

@wrapper
def func():
pass

func()

2. 同一个函数被多个装饰器装饰

@wrapper1
@wrapper2
@wrapper3
def func():
pass

1 2 3 func 3 2 1

3. 带参数的装饰器

def wrapper_out(参数):
def wrapper(fn):
def inner(*args, **kwargs): # 聚合
在目标函数之前
ret = fn(*arg, **kwargs) # 打散
在目标函数之后
return ret
return inner
return wrapper

@wrapper_out(实参) # 执行的时候. 先执行函数的调用然后使用返回值和前面的@组合成装饰器语法糖
def func():
pass

装饰器应用

flag = False  # 没登录

def login():
    global flag
    username = input('请输入用户名')
    password = input('请输入密码')
    if username == 'alex' and password == '123':
        flag = True
        print('登录')
    else:
        flag = False
        print('用户名密码错误')
# 登录验证装饰器
def wrapper(fn):
    def inner(*args, **kwargs):
        # 登录校验
        while 1:
            if flag == True:
                ret = fn(*args, **kwargs)
            else:
                print('对不起, 您没有登录')
                login()
    return inner
menu = ['我是被装饰1','我是被装饰2','我是被装饰3']
@wrapper
def chakan():
    print('我是被装饰1')

def cha():
    print('我是被装饰2')

def so():
    print('我是被装饰3')

while 1:
     for i in range(len(menu)):
         print(i+1, menu[i])
     n = input('请输入要调戏的装饰')
     if n == '1':
         chakan()
     elif n == '2':
         cha()
     elif n == '3':
         so()
     else:
         print('您输入的对象有误')