lambda_formatter.py

import logging
from typing import Callable


class LambdaFormatter(logging.Formatter):
    def __init__(
            self,
            fmt,
            pre_format_func: Callable[[logging.LogRecord], logging.LogRecord] = lambda text: text,
            post_format_func: Callable[[str, logging.LogRecord], str] = lambda text, _: text
    ):
        super().__init__(fmt)
        self.pre_format_func = pre_format_func
        self.post_format_func = post_format_func

    def format(self, record: logging.LogRecord) -> str:
        log_fmt = self.post_format_func(super().format(self.pre_format_func(record)), record)
        return log_fmt