Python编程之基础知识练习_005

练习内容：使用两种不同方式，编写一个可接受参数的logging装饰器函数

方式一：

__author__ = 'Orcsir'
from functools import wraps
import logging


def logged(level, name=None, message=None):
    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

# Test
@logged(logging.DEBUG)
def add(x, y):
    print("add func")
    return x + y


@logged(logging.WARNING, message='example')
def spam():
    print("Spam")

add(1, 2)
spam()

方式二：使用partial函数，减少装饰器内部函数定义

__author__ = 'Orcsir'

from functools import wraps, partial
import logging


def logged(func=None, level=logging.INFO, name="", message=""):
    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


# Test
@logged(level=logging.WARNING)
def add(x, y):
    print("add func")
    return x + y


@logged(level=logging.WARNING, message='example')
def spam():
    print("Spam")

add(1, 2)
spam()