diff --git a/lib/common.py b/lib/common.py index 42dbb57..2ef9416 100644 --- a/lib/common.py +++ b/lib/common.py @@ -1,5 +1,6 @@ # common.py import os, yaml +import logging def load_config(): """ @@ -8,3 +9,13 @@ def load_config(): path = os.path.join(os.path.dirname(__file__), '..', 'conf', 'config.yaml') with open(path, encoding='utf-8') as f: return yaml.safe_load(f) + +def get_logger(name): + logger = logging.getLogger(name) + if not logger.handlers: + handler = logging.StreamHandler() + formatter = logging.Formatter('[%(asctime)s] %(levelname)s: %(message)s') + handler.setFormatter(formatter) + logger.addHandler(handler) + logger.setLevel(logging.INFO) + return logger