python带参数装饰器的两种写法

前言

最近在实现一个装饰器的过程中发现了一个很有意思的地方，在博客里面分享出来

不同的写法

三层函数嵌套，实现了可传参数的一个装饰器。

import logging
import functools

def logger(msg=None):
    """日志"""
    def dector(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            result = func(*args, **kwargs)
            logging.info(func.__name__ + func.__doc__)
            logging.info(f"参数：{args}{kwargs}")
            if msg:
                log.info(msg.format(result))
            return result
        return wrapper
    return dector

使用functools.partial函数实现了一个比较抽象的三层带参数装饰器。

import logging
import functools

def loggers(func=None, msg=None):
    """日志"""
    if func is None:
        return functools.partial(loggers, msg=msg)

    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        result = func(*args, **kwargs)
        logging.info(func.__name__ + func.__doc__)
        logging.info(f"参数：{args}{kwargs}")
        if msg:
            log.info(msg.format(result))
        return result

    return wrapper