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

有何异同

可以看到从代码的阅读层面来讲，第一个写法是比较易读的，第二种写法不容易阅读。从执行调用方面第二种更简单点。

先看以下第一种调用执行的方式：

第一种写法
- 装饰器传参数
- 装饰器不传参数
第二种写法
- 装饰器不带参数
- 装饰器带参数

可以看到使用了functools.partial函数的装饰器，在调用非必填参数时，可以不用使用括号。

除了写法理解比较抽象，这样的调用还是比较友好的。