File size: 693 Bytes
40a3ea8 | 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 | import logging
import sys
def setup_logging(args=None, log_level=None, reset=False):
if logging.root.handlers:
if reset:
for handler in logging.root.handlers[:]:
logging.root.removeHandler(handler)
else:
return
if log_level is None and args is not None:
log_level = getattr(args, "console_log_level", None)
if log_level is None:
log_level = "INFO"
log_level = getattr(logging, str(log_level).upper())
handler = logging.StreamHandler(sys.stdout)
handler.setFormatter(logging.Formatter("%(message)s"))
logging.root.setLevel(log_level)
logging.root.addHandler(handler)
setup_logging()
|