装饰器

定义：

  器字代表函数，装饰器本质是函数；装饰：装饰其他函数，就是为其他函数添加附加功能

原则：

  1.不能修改被装饰函数的源代码(在不修改被装饰函数源代码的情况下为其添加功能)

  2.不能修改被装饰的函数的调用函数，被装饰函数不知道被装饰了

实现装饰器需要了解：

  1.函数及变量：内存存储函数方式与存储变量一样：内存中存储函数体，再指向函数名

  2.高阶函数

a.把一个函数名当作实参传给另外个函数(在不修改被装饰函数源代码的情况下为其添加功能)

b.返回值中包含函数名(不修改函数的调用方式)

import time
def bar():#bar是这个函数的门牌号
    time.sleep(2)
    print('in the bar')
def test1(func):#不是装饰器，是因为函数的调用方式test1(bar)被更改了
    start_time=time.time()
    func()
    stop_time=time.time()
    print('the func fun time is %s' % (stop_time-start_time))

test1(bar) #func=bar

 运行结果：

  

import time
def bar():#bar是这个函数的门牌号
    time.sleep(2)
    print('in the bar')

def test2(func):
    print(func)
    return func

print(test2(bar))
print('--------------------')
bar=test2(bar)
print('--------------------')
bar()

运行结果：

  3.嵌套函数：在已定义的函数内部再定义函数

def foo():
    print('in the foo')
    def bar():#bar具有局部变量的特性
        print('in the bar')

通过高阶函数和嵌套函数实现装饰器功能：

import time

def timer(func):#在被装饰函数之外嵌套装饰器函数，添加装饰功能
    def deco(*args,**kwargs):#不固定参数参数
        start_time=time.time()
        result=func(*args,**kwargs)#被装饰的函数在这运行
        stop_time=time.time()
        print('the func run time is %s' % (stop_time-start_time))

        return result#加入了返回值
    return deco#timer这个函数返回了deco函数的地址

@timer#@timer等于test1=timer(test1)：返回了deco函数的内存地址赋给了test1，此时test1是新的test1，这时的test1已经有了装饰器的功能
def test1():
    time.sleep(3)
    print('in the test1')

@timer#test2=timer(test2),deco=test2
def test2(name):
    time.sleep(3)
    print('test2:',name)
    print('in the test2')
    return 'from test2'#加入了返回，装饰器内的函数也要加返回

test1()
print('-----------------')
print(test2('tao'))#print打印函数的返回值

运行结果：

  所以说，装饰器就是将被装饰函数在装饰器函数内执行，再将装饰器函数赋予被装饰函数名，之后再执行被装饰函数时就相当于执行了装饰器函数，即在没有改变被装饰函数的源代码和调用方式的情况下，实现了装饰的功能。

实现加入参数的装饰器：

user,passwd='Tao','abc'
def auth(auth_type):#与未加参数的装饰器区别：装饰器多加了一个参数，要多嵌套一层函数；auth返回的仍是wrapper这个函数
    def outer_wrapper(func):#注意加入的参数和func的位置
        def wrapper(*args, **kwargs):#auth函数返回out_wrapper,out_wrapper函数返回wrapper，所以auth函数返回的仍是wrapper这个函数

            #判断参数
            if auth_type=='local':
                username = input('username:').strip()
                password = input('passwd:').strip()

                if user == username and passwd == password:
                    print('欢迎')
                    res = func(*args, **kwargs)  #当被装饰函数有返回值时
                    return res
                else:
                    exit('错误')

            #判断参数
            elif auth_type=='ldap':
                print('ldap无法实现')

        return wrapper

    return outer_wrapper



def index():
    print('welcome to index page')

@auth(auth_type='local')#auth返回了wrapper这个函数赋给了home
def home():
    print('welcome to home page')
    return "from home" #加入了返回，装饰器内的函数也要加返回

@auth(auth_type='ldap')#auth返回了wrapper这个函数赋给了bbs
def bbs():
    print('welcome to bbs page')

index()
print('--------------------')
print(home())#打印函数的返回值即from home
print('--------------------')
bbs()

运行结果：