thomasbyr/bioinfo-genome

🧬 large scale genes data analysis software

bioinfo-genome's Issues

proper logging

I may (or may not) change the way we log shit to stdout/stderr
Hopefully this wont break the proxied threads 😇

import os

from dotenv import load_dotenv

import logging
from typing_extensions import override
from termcolor import colored

__all__ = ['logger', 'log_lvl', 'console_handler', 'default_formatter']


def formatter(c: str, attrs: list[str] = []) -> str:
  return f"{colored('%(asctime)s', 'grey', attrs=['bold'])} {colored('%(levelname)8s', c, attrs=attrs)} {colored('%(name)s', 'magenta')} (%(filename)s:%(lineno)d) %(message)s"


class UsefulFormatter(logging.Formatter):

  dt_fmt = '%Y-%m-%d %H:%M:%S'
  FORMATS = {
    logging.DEBUG: formatter('green'),
    logging.INFO: formatter('blue'),
    logging.WARNING: formatter('yellow'),
    logging.ERROR: formatter('red'),
    logging.CRITICAL: formatter('red', ['bold']),
  }

  @override
  def format(self, record: logging.LogRecord) -> str:
    log_fmt = self.FORMATS.get(record.levelno)
    formatter = logging.Formatter(log_fmt, self.dt_fmt, style='%')
    return formatter.format(record)


load_dotenv()
DEBUG = os.getenv('DEBUG', 'False').lower() in {'true', '1', 'yes'}

logger = logging.getLogger('me 🙂')
logger.setLevel(log_lvl := logging.DEBUG if DEBUG else logging.INFO)

# create console handler with a higher log level
console_handler = logging.StreamHandler()
console_handler.setLevel(log_lvl)
console_handler.setFormatter(default_formatter := UsefulFormatter())

logger.addHandler(console_handler)