Print colored logs

color_logger.py

from colorama import Fore, Back, Style
import logging

logger = logging.getLogger(__name__)
logger.setLevel(logging.DEBUG)
logging.basicConfig(
    format='%(asctime)s %(levelname)-8s %(message)s',
    level=logging.INFO,
    datefmt='%Y-%m-%d %H:%M:%S')


def warning(msg):
    logger.warning(Fore.YELLOW + msg + Style.RESET_ALL)


def error(msg):
    logger.error(Fore.RED + msg + Style.RESET_ALL)


def info(msg):
    logger.info(Fore.GREEN + msg + Style.RESET_ALL)


def debug(msg):
    logger.debug(Fore.CYAN + msg + Style.RESET_ALL)


def critical(msg):
    logger.critical(Fore.MAGENTA + msg + Style.RESET_ALL)


def highlight(msg):
    logger.info(Fore.BLACK + Back.YELLOW + msg + Style.RESET_ALL)


if __name__ == '__main__':
    warning('This is a warning')
    error('This is an error')
    info('This is info')
    debug('This is debug')
    critical('This is critical')
    highlight('This is highlighted info')