From fa313d699a27042bf12ee0549b36d041f7f66d4b Mon Sep 17 00:00:00 2001
From: chengchongzhen <15939054361@163.com>
Date: Mon, 13 Jan 2025 13:23:50 +0800
Subject: [PATCH] Refactor configuration validation in config.py by introducing
 check_config and check_is_valid methods for improved clarity and
 maintainability

---
 config.py | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/config.py b/config.py
index 37348b5a..23625aad 100644
--- a/config.py
+++ b/config.py
@@ -26,10 +26,7 @@ def __init__(self):
         self.temp_mail = os.getenv("TEMP_MAIL", "").strip()
         self.domain = os.getenv("DOMAIN", "").strip()
 
-        if not self.temp_mail:
-            raise ValueError("临时邮箱未配置，请在 .env 文件中设置 TEMP_MAIL")
-        if not self.domain:
-            raise ValueError("域名未配置，请在 .env 文件中设置 DOMAIN")
+        self.check_config()
 
     def get_temp_mail(self):
         logging.info(f"\033[32m临时邮箱: {self.temp_mail}\033[0m")
@@ -39,6 +36,15 @@ def get_domain(self):
         logging.info(f"\033[32m域名: {self.domain}\033[0m")
         return self.domain
 
+    def check_config(self):
+        if not self.check_is_valid(self.temp_mail):
+            raise ValueError("临时邮箱未配置，请在 .env 文件中设置 TEMP_MAIL")
+        if not self.check_is_valid(self.domain):
+            raise ValueError("域名未配置，请在 .env 文件中设置 DOMAIN")
+
+    def check_is_valid(self, str):
+        return len(str.strip()) > 0
+
 
 # 使用示例
 if __name__ == "__main__":