Explorar el Código

Refactor configuration loading and logging in config.py; update .env.example formatting

chengchongzhen hace 1 año
padre
commit
15ffc5e07a
Se han modificado 2 ficheros con 18 adiciones y 7 borrados
  1. 3 3
      .env.example
  2. 15 4
      config.py

+ 3 - 3
.env.example

@@ -1,6 +1,6 @@
-DOMAIN='xxxxx.me'
-TEMP_MAIL='xxxxxx'
-BROWSER_USER_AGENT='Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/130.0.6723.92 Safari/537.36'
+DOMAIN=xxxxx.me
+TEMP_MAIL=xxxxxx
+BROWSER_USER_AGENT=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/130.0.6723.92 Safari/537.36
 
 # 代理
 # BROWSER_PROXY='http://127.0.0.1:2080' 

+ 15 - 4
config.py

@@ -1,6 +1,7 @@
 from dotenv import load_dotenv
 import os
 import sys
+from logger import logging
 
 
 class Config:
@@ -22,11 +23,21 @@ class Config:
         # 加载 .env 文件
         load_dotenv(dotenv_path)
 
+        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")
+
     def get_temp_mail(self):
-        return os.getenv("TEMP_MAIL")
+        logging.info(f"\033[32m临时邮箱: {self.temp_mail}\033[0m")
+        return self.temp_mail
 
     def get_domain(self):
-        return os.getenv("DOMAIN")
+        logging.info(f"\033[32m域名: {self.domain}\033[0m")
+        return self.domain
 
 
 # 使用示例
@@ -34,7 +45,7 @@ if __name__ == "__main__":
     try:
         config = Config()
         print("环境变量加载成功!")
-        print(f"临时邮箱: {config.get_temp_mail()}")
-        print(f"域名: {config.get_domain()}")
+        config.get_temp_mail()
+        config.get_domain()
     except ValueError as e:
         print(f"错误: {e}")