logger.py

import logging
import logging.handlers
import os

def get_logger(log_name, logdir=None, loglevel=logging.INFO):
    _logger = logging.getLogger(log_name)
    fmt = '%(asctime)s [%(levelname)s] %(message)s'
    formatter = logging.Formatter(fmt)

    if not _logger.hasHandlers():
        if logdir is None:
            handler = logging.StreamHandler()
        else:
            handler = logging.handlers.RotatingFileHandler(
                filename=os.path.join(logdir, log_name),
                maxBytes=2 * 1024 * 1024 * 1024,
                backupCount=10)
        handler.setFormatter(formatter)
        _logger.addHandler(handler)

    _logger.setLevel(loglevel)
    return _logger