首页 > > 网络编程 > 其它 >

python 装饰器（二）：加参数

2019-02-20 00:47:14来源：博客园阅读 ()

接上篇python 闭包&装饰器（一）

一、功能函数加参数：实现一个可以接收任意数据的加法器

源代码如下：

def show_time(f):


    def inner(*x, **y):     # 形参
        start = time.time()
        f(*x, **y)     # 相当于add()
        end = time.time()
        print('spend %s' % (end - start))


    return inner


@show_time  # @show_time 等于 add = show_time(add)
def add(*a, **b):   # 实参  可以接收任意参数
    sums = 0
    for i in a:
        sums += i
    print(sums)
    time.sleep(1)

add(1, 2, 3, 4, 5)    # 15

二、装饰器函数加参数：根据需求打印日志

源代码如下：

def logger(flag=''):      # flag 默认为空，这一层函数的作用是应用标志位
    def show_time(f):
        def inner(*x, **y):
            start = time.time()
            f(*x, **y)           # 相当于add()
            end = time.time()
            print('spend %s' % (end - start))
            if flag == 'true':
                print('打印日志')

        return inner
    return show_time


@logger('true')     # @show_time
def add(*a, **b):    # 不管有名参数和无名参数都可以接收到
    sums = 0
    for i in a:
        sums += i
    print(sums)
    time.sleep(1)

add(1, 2, 3, 4, 5)


@logger('true')
def bar():
    print('bar.......')
    time.sleep(3)    # 停留3秒

bar()

注意：@logger('true') 的执行机制是：先执行 logger('true') 函数，然后返回 show_time 函数的内存地址，所以 @logger('true') 就相当于 @show_time，两者的区别就是 @logger('true') 要了一个变量flag, 之后的装饰器函数里面就可以使用falg这个标志位了

原文链接:https://www.cnblogs.com/ss-long/p/10398925.html
如有疑问请与原作者联系

标签：

版权申明：本站文章部分自网络，如有侵权，请联系：west999com@outlook.com
特别注意：本站所有转载文章言论不代表本站观点，本站所提供的摄影照片，插画，设计作品，如需使用，请与原作者联系，版权归原作者所有