diff --git a/main.py b/main.py index 3786ef3..5bb2e7d 100644 --- a/main.py +++ b/main.py @@ -11,10 +11,13 @@ import yaml @singleton class Main(): - def __init__(self) -> None: - + def init_log_handlers(self): with open(".env.yaml", "r") as file: config = yaml.safe_load(file) + if config["log"] is None: + return + if config["log"]["handlers"] is None: + return log_handler_confg = config["log"]["handlers"] log_handlers = [] for handler_config in log_handler_confg: @@ -31,6 +34,9 @@ class Main(): )) logging.basicConfig(level=logging.DEBUG, handlers=log_handlers) logging.info("logging init finish") + + def __init__(self) -> None: + self.init_log_handlers() def run(self): logger = logging.getLogger(__name__)