JSON Logger example

logs.py

import os
import json
import logging
from unittest.mock import Mock


LOG_LEVEL = os.environ.get('LOG_LEVEL', 'DEBUG').upper()


if not (LOG_LEVEL in ('DEBUG', 'INFO', 'WARN', 'WARNING', 'ERROR', 'CRITICAL', 'FATAL')):
    raise ValueError(f'Invalid LOG_LEVEL {LOG_LEVEL}')


class JsonFormatter(logging.Formatter):
    def format(self, record: logging.LogRecord):
        rdict = vars(record)
        if isinstance(rdict['msg'], Mock):
            rdict['msg'] = repr(rdict['msg'])
        return json.dumps(rdict)


def get_logger(name, level=None) -> logging.Logger:
    if level is None:
        level = LOG_LEVEL
    log = logging.getLogger(name)
    log.setLevel(level)
    handler = logging.StreamHandler()
    handler.setFormatter(JsonFormatter())
    log.addHandler(handler)
    return log