装饰器

 

装饰器：

　　定义：本质是函数，功能是用来装饰其他函数（就是为其他函数添加附加功能）。

　　原则：1、不能修改被装饰的函数的源代码

　　　   　2、不能修改被装饰的函数的调用方式

装饰器对于被装饰的函数是透明的（既函数本身不知道自己被装饰）

 

 

实现装饰器知识储备：

　　1、函数即“变量”

　　2、高阶函数

　　　　a.把一个函数名当作参数传递给另外一个函数

　　　　b.返回值中包含函数名

　　3、函数嵌套

 

小结：利用高阶函数+嵌套函数完成装饰器的功能

 

例子：

user = 'daxin'
passwd = '123456'

def auth_type(auth_type):
    def auth(func):
        def auth_page(*args):
            if auth_type == 'local':
                username = input('Please input your user:')
                password = input("Please input your passwd:")
                if username == user and passwd == password:
                    func(*args)
                else:
                    print('Username or password Error!')
            else:
                print('LDAP')
                func(*args)
        return auth_page
    return auth

def index():
    print('in the index page')
@auth_type('local')      --> local传递到最外层的func
def home():
    print("in the home page")
@auth_type('ldap')
def bbs():
    print('in the bbs page')

index()
home()
bbs()

例子2：

import functools

def decorator(logintype='3'):
    def decorator1(func):
        @functools.wraps(func)
        def wrapper(*args,**kwargs):
            if logintype == '1':
                print('begin call')
                func(*args,**kwargs)
                print('end call')
            elif logintype == '2':
                print('start call')
                func(*args,**kwargs)
                print('over call')
            else:
                print('test start')
                func(*args,**kwargs)
                print('test start')
        return wrapper
    return decorator1


@decorator()
def a(name):
    print('in the F name is %s' % name)

@decorator('2')
def b():
    print('in the B')
@decorator('1')
def c():
    print("in the C")

c()
b()
a('123')