python 日志打印

output some logs into console(screen) and some logs into a log file.

e.g. warning and higher logs need output into screen, and all logs dump into a logfile.

import logging
import sys
import os

def setup_test_logging(logfile = None, console_min_level = logging.WARN, file_min_level = logging.DEBUG):
""" setup logging for test, output logs on console (default logging.WARN) and logfile(default logging.DEBUG)"""
if logfile is None:
filename = os.path.basename(sys.argv[0]).replace('.py', '.log')
else:
filename = logfile
logging.basicConfig(format='%(asctime)s %(filename)s:%(lineno)d %(levelname)-8s %(message)s', level=file_min_level, filename = filename)
console = logging.StreamHandler(stream = sys.stdout)
console.setLevel(console_min_level)
formatter = logging.Formatter('%(asctime)s %(filename)s:%(lineno)d %(levelname)-8s %(message)s')
console.setFormatter(formatter)
logging.getLogger('').addHandler(console)