#!/usr/bin/env python3

from pycolib.coloredformatter import ColoredFormatter

import logging

def prepare_logger(name, level=logging.DEBUG):

    logger = logging.getLogger(name)

    logger.setLevel(level)

    return logger

def prepare_console_handler(level=logging.DEBUG):

    handler = logging.StreamHandler()

    formatter = ColoredFormatter('%(asctime)s %(levelname)-5s %(message)s', '%H:%M:%S')

    formatter.set_color('message', 'RECV', 2)

    formatter.set_color('message', 'SEND', 6)

    handler.setFormatter(formatter)

    handler.setLevel(level)

    return handler

def setup():

    logging.addLevelName(21, 'RECV')

    logging.addLevelName(22, 'SEND')

    logger_main = prepare_logger('main')

    handler_console = prepare_console_handler()

    logger_main.addHandler(handler_console)

def finish():

    logging.shutdown()

def main():

    setup()

    log = logging.getLogger('main')

    log.info('test')

    try:

        raise Exception('test exception')

    except:

        log.exception('test exc')

    log.log(21, 'recv')

    log.log(22, 'send')

    log.debug('debug')

    finish()

if __name__ == '__main__':

    main()