From d2ac2e716d5e4e11a6a137b19cb6e1ae4ad36d57 Mon Sep 17 00:00:00 2001
From: 0Xiao0 <511201264@qq.com>
Date: Thu, 6 Feb 2025 09:53:47 +0800
Subject: [PATCH] fixup

---
 main.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

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__)