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

This commit is contained in:
chengchongzhen
2025-01-13 13:23:50 +08:00
parent 15ffc5e07a
commit fa313d699a

View File

@@ -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__":