Discuss / Python / Python Cookbook 9.6 带可选参数的装饰器给出了答案

Python Cookbook 9.6 带可选参数的装饰器给出了答案

Topic source

墨韩

#1 Created at ... [Delete] [Delete and Lock User]
from functools import wraps, partial
import logging

def logged(func=None, *, level=logging.DEBUG, name=None, message=None):
    if func is None:
        return partial(logged, level=level, name=name, message=message)

    logname = name if name else func.__module__
    log = logging.getLogger(logname)
    logmsg = message if message else func.__name__

    @wraps(func)
    def wrapper(*args, **kwargs):
        log.log(level, logmsg)
        return func(*args, **kwargs)

    return wrapper

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

@logged(level=logging.CRITICAL, name='example')
def spam():
    print('Spam!')

  • 1

Reply