Python装饰器(decorator)系列 --- 编写带参数的装饰器

装饰器系列：

[1]. Python装饰器(decorator)系列 — 面向对象以及装饰器
[2]. Python装饰器(decorator)系列 — 编写无参数的装饰器
[3]. Python装饰器(decorator)系列 — 编写带参数的装饰器

---

对于上一节的 @log 装饰器：

from functools import reduce

def log(f):
    def fn(x):
        print('call ' + f.__name__ + '()...')
        return f(x)
    return fn

@log
def factorial(n):
    return reduce(lambda x, y: x*y, range(1, n+1))

print(factorial(10))

发现对于被装饰的函数，log打印的语句是不能变的（除了函数名）。

但如果有的函数非常重要，要求打印出’[INFO] call xxx()…’
有的函数不太重要，要求打印出’[DEBUG] call xxx()…’
这时，log函数本身就需要传入’INFO’或’DEBUG’这样的参数，类似这样：

@log('DEBUG')
def factorial(n):
    return reduce(lambda x, y: x*y, range(1, n+1))

由于需要接受参数，因此装饰器需要增添一个函数
方法一(初级调用)：

def log(prefix):  # 这里的prefix接受的是‘DEBUG’参数
    def log_decorator(f):
        def wrapper(*args, **kw):
            print('[%s] %s()...' % (prefix, f.__name__))
            return f(*args, **kw)
        return wrapper
    return log_decorator

# @log('DEBUG')
def test():
    pass

test1 = log('DEBUG')
test = test1(test)

print(test())

方法二(升级调用)：

def log(prefix):
    def log_decorator(f):
        def wrapper(*args, **kw):
            print('[%s] %s()...' % (prefix, f.__name__))
            return f(*args, **kw)
        return wrapper
    return log_decorator

def test():
    pass
test = log('DEBUG')(test)
print(test())

方法三(装饰器调用)：

def log(prefix):
    def log_decorator(f):
        def wrapper(*args, **kw):
            print('[%s] %s()...' % (prefix, f.__name__))
            return f(*args, **kw)
        return wrapper
    return log_decorator

@log('DEBUG')
def test():
    pass
print(test())

结果为：

[DEBUG] test()…
None

练习：
上一节的@performance只能打印秒，请给 @performace 增加一个参数，允许传入’s’或’ms’：

import time
from functools import reduce

def performance(unit):
    def time_decorator(f):
        def wrapper(*args, **kw):
            print('call'+' '+'%s() in %s%s' % (f.__name__, time.time(), unit))
            return f(*args, **kw)
        return wrapper
    return time_decorator

@performance('ms')
def factorial(n):
    return reduce(lambda x,y: x*y, range(1, n+1))

print(factorial(10))

结果为：

call factorial() in 1563515091.1575737ms
3628800

---

大家加油！
学习链接：https://www.imooc.com/code/6066