ProxyC/Utils/Logger.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import sys
import logging
import colorlog
from Basis.Constant import LogLevel, LogFile

logColor = {  # log color
    'DEBUG': 'white',
    'INFO': 'green',
    'WARNING': 'yellow',
    'ERROR': 'red',
    'CRITICAL': 'bold_red',
}

logLevel = {  # log level
    'debug': logging.DEBUG,
    'info': logging.INFO,
    'warning': logging.WARNING,
    'error': logging.ERROR,
    'critical': logging.CRITICAL
}[LogLevel.lower()]

dateFormat = '%Y-%m-%d %H:%M:%S'
logFormat = '[%(asctime)s.%(msecs)03d] [%(levelname)s] %(message)s (%(module)s.%(funcName)s)'

# load fileHandler -> log file
fileHandler = logging.FileHandler(LogFile, encoding = 'utf-8')
fileHandler.setFormatter(logging.Formatter(
    logFormat, datefmt = dateFormat
))
fileHandler.setLevel(logging.DEBUG)  # debug level for log file

# load stdHandler -> stderr
stdHandler = colorlog.StreamHandler()
stdHandler.setFormatter(colorlog.ColoredFormatter(
    '%(log_color)s' + logFormat,
    datefmt = dateFormat,
    log_colors = logColor,
    stream = sys.stderr
))
stdHandler.setLevel(LogLevel)  # custom level for stderr

logger = logging.getLogger()
logger.addHandler(stdHandler)
logger.addHandler(fileHandler)
logger.setLevel(logging.DEBUG)  # set log level in handler