log.py example

example-log.py

# -*- coding: utf-8 -*-
# vim: set et sw=4 ts=4:


import logging
import logging.config

import flask


class RequestIDLogger(logging.getLoggerClass()):

    def _log(self, level, msg, args, exc_info=None, extra=None):
        try:
            request_id = getattr(flask.g, "request_id", None)
        except RuntimeError:  # outside of Flask context
            request_id = None

        if request_id is not None:
            msg = "<request:{0}> | {1}".format(request_id, msg)

        return super(RequestIDLogger, self)._log(
            level, msg, args, exc_info, extra
        )


logging.setLoggerClass(RequestIDLogger)

getLogger = logging.getLogger
configure_logging = logging.config.dictConfig

DEBUG = logging.DEBUG
INFO = logging.INFO