File size: 899 Bytes
158b61b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
# -*- coding: utf-8 -*-
import logging
from logging.handlers import RotatingFileHandler
logger = logging.getLogger()
def init_logger(
log_file=None,
log_file_level=logging.NOTSET,
rotate=False,
log_level=logging.INFO,
):
log_format = logging.Formatter("[%(asctime)s %(levelname)s] %(message)s")
logger = logging.getLogger()
logger.setLevel(log_level)
console_handler = logging.StreamHandler()
console_handler.setFormatter(log_format)
logger.handlers = [console_handler]
if log_file and log_file != '':
if rotate:
file_handler = RotatingFileHandler(
log_file, maxBytes=1000000, backupCount=10)
else:
file_handler = logging.FileHandler(log_file)
file_handler.setLevel(log_file_level)
file_handler.setFormatter(log_format)
logger.addHandler(file_handler)
return logger
|