Bladeren bron

Refactor configuration validation in config.py by introducing check_config and check_is_valid methods for improved clarity and maintainability

chengchongzhen 1 jaar geleden
bovenliggende
commit
fa313d699a
1 gewijzigde bestanden met toevoegingen van 10 en 4 verwijderingen
  1. 10 4
      config.py

+ 10 - 4
config.py

@@ -26,10 +26,7 @@ class Config:
         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 @@ class Config:
         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__":