Decorator装饰器定义可选参数

一般带参数装饰器。
使用装饰器时需要加上括号@logged()。
@logged将会提示参数错误。

import logging
logging.basicConfig(level=logging.DEBUG)


def logged(level=logging.INFO):
    def decorator(func):
        def wrapper(*args, **kwargs):
            log = logging.getLogger(func.__module__)
            log.log(level, ('execute %s!' % func.__name__))
            return func(*args, **kwargs)
        return wrapper
    return decorator

@logged()
def add(x, y):
    return x + y

add(1, 1)

---

利用functools.partial定义可选参数装饰器。
此时若无参，@logged()和@logged都是可以的。

from functools import partial

def new_logged(func=None, *, level=logging.INFO):
    if not func:
        return partial(new_logged, level=level)
    def wrapper(*args, **kwargs):
        log = logging.getLogger(func.__module__)
        log.log(level, ('execute %s!' % func.__name__))
        return func(*args, **kwargs)
    return wrapper

@new_logged
def minus(x, y):
    return x - y

@new_logged(level=logging.WARNING)
def plus(x, y):
    return x + y

---

对于@new_logged，因为@new_logged后没有()，这时候被装饰函数minus会被作为第一个参数直接传入new_logged装饰器，即new_logged(minus)。

@new_logged
def minus(x, y):
    return x - y

相当于：

def minus(x, y):
    return x - y
minus = new_logged(minus)

---

对于@new_logged()，因为@new_logged存在()，这是传入的参数为空，将会执行判断语句返回partial(new_logged, level=level)。

@new_logged()
def minus(x, y):
    return x - y

相当于:

def minus(x, y):
    return x - y
minus = new_logged()(minus)

---

对于@new_logged(level=logging.WARNING)：即在@new_logged()的基础上传入了参数，第一个func依旧为None，执行判断语句返回partial(new_logged, level=level)。

@new_logged(level=logging.WARNING)
def plus(x, y):
    return x + y

相当于：

def plus(x, y):
    return x + y
plus = new_logged(level=logging.WARNING)(plus)