python - 如何在 Python 的方法装饰器中使用类属性？

Question

我有一个负责记录的装饰器。例如这样的：

import logging
import functools
from typing import Callable


def log_container(func: Callable, logger=None):
    if logger is None:
        logger = logging.getLogger("default")
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        logger.info(f"{func.__qualname__} started.")
        result = func(*args, **kwargs)
        logger.info(f"{func.__qualname__} ended.")
        return result

    return wrapper

我有一个具有不同记录器作为属性的类，我想将该记录器作为装饰器参数。这可能吗？如果是这样，如何？

class Dummy:
    logger = logging.getLogger("Dummy")
    
    @log_container(logger=Dummy.logger)
    def do_something(self):
        pass

标签： pythonloggingpython-decoratorspython-class

Answer 1

你的装饰器，正如它目前定义的那样，不起作用。您不能将装饰函数与参数一起传递，因为装饰器本身只接受一个参数：它func本身。如果你想让你的装饰器接受一个参数，你需要将装饰器嵌套更深一层。现在变成一个返回装饰器的函数：

def log_container(logger=None):
    if logger is None:
        logger = logging.getLogger("default")

    def inner(func: Callable):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            logger.info(f"{func.__qualname__} started.")
            result = func(*args, **kwargs)
            logger.info(f"{func.__qualname__} ended.")
            return result

        return wrapper
    return inner

然后在为类装饰方法时Dummy，不要使用Dummy.logger因为Dummy直到类定义结束才定义，而是可以logger直接访问类属性。

class Dummy:
    logger = logging.getLogger("Dummy")
    
    @log_container(logger=logger)
    def do_something(self):
        pass

    # Use default logger
    @log_container()
    def do_something_else(self):
        pass