十三：装饰器函数

一：装饰器的作用就是在原来函数的基础上增加了一些扩展的功能，因为我们开始写的函数已经投入应用了，所以不能对原函数进行修改，也不能修改对它的调用方式，即“开放封闭原则”，封闭：就是说不能对已经使用的函数进行任何修改；开放：就是说虽然不能对原函数进行修改，但是我们可以进行扩展，以增加新的功能。当然我们也可以在执行某些函数之前需要执行其它固定函数，也可以作为装饰器函数的应用，比如登陆操作，无论进入那个功能函数，都需要提前进行登陆，所以可以把登陆功能作为其它功能函数的装饰器来用。

example 1：需求：对已经投入使用的函数 foo() 增加了一个函数运行时间的功能，并且不能改变其原来的调用方式。

# 需求：在不改变调用方式的情况下，给要求的函数加上计算执行时间的功能

import time

def show_time(f):
    def inner():
        start = time.time()
        f()
        end = time.time()
        print('spend %s' % (end - start) )
    return inner

@show_time    # foo = show_time(foo)
def foo():
    print('foo...')
    time.sleep(2)

foo()  # 仍以原来的方式进行调用

 

example 2：对功能函数和装饰器函数加入参数，对装饰器加参数，我们需要再嵌套一层循环，这样就可以引入一个参数。

# 需求：在不改变调用方式的情况下，给要求的函数加上计算执行时间的功能

import time

def logger(flag = 'Flase'):  # 装饰器函数传入参数
    def show_time(f):    # 不能直接在这个函数中加入参数，因为装饰器格式是定好的
        def inner(x,y):    # 给功能函数加参数要传入形参
            start = time.time()
            f(x,y)         # 对 f 的访问为闭包，给功能函数加参数要传入形参
            end = time.time()
            print('spend %s' % (end - start) )
            if flag == 'true':
                print('日志记录')
        return inner
    return show_time      #需要多加一个返回，返回值为嵌套的最外层函数

@logger('true')    #  == @show_time     add = show_time(add)
def add(x,y):
    time.sleep(2)
    print(x + y)

add(1,2)

 

example 3：如果装饰器函数不需要参数的话，那么只需要给功能函数加如参数就可以了。

import time

def show_time(f):
    def inner(x,y):   # 给功能函数加参数要传入形参
        start = time.time()
        f(x,y)         # 对 f 的访问为闭包，给功能函数加参数要传入形参
        time.sleep(1)
        end = time.time()
        print('spend %s' % (end - start) )
    return inner

@show_time   # foo = show_time(foo)
def foo(x,y):
    print('result:',x + y)

foo(1,2)