python 带参数的装饰器

from functools import wraps import logging

def logged(level, name=None, message=None): """ Add logging to a function. level is the logging level, name is the logger name, and message is the log message. If name and message aren't specified, they default to the function's module and name. """ def decorate(func): 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
return decorate

Example use

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

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