log.py

import logging


# create logger
def getLogger(name='logger', level='INFO', mode='w', fh=True, ch=True, prefix=""):
    logger = logging.getLogger(name)

    fh = logging.FileHandler(name + '.log', mode)

    ch = logging.StreamHandler()

    if level == "INFO":
        logger.setLevel(logging.INFO)
        fh.setLevel(logging.INFO)
        ch.setLevel(logging.INFO)
    elif level == "DEBUG":
        logger.setLevel(logging.DEBUG)
        fh.setLevel(logging.DEBUG)
        ch.setLevel(logging.DEBUG)
    elif level == "ERROR":
        logger.setLevel(logging.ERROR)
        fh.setLevel(logging.ERROR)
        ch.setLevel(logging.ERROR)

    formatter = logging.Formatter(prefix + ' ' + '%(filename)s:%(lineno)s %(levelname)s: %(message)s')
    fh.setFormatter(formatter)
    ch.setFormatter(formatter)

    if fh:
        logger.addHandler(fh)
    if ch:
        logger.addHandler(ch)

    return logger