新增使用imap来代替临时邮箱获取验证码

config.py

@@ -23,15 +23,36 @@ class Config:
         # 加载 .env 文件
         load_dotenv(dotenv_path)
 
+        self.imap = False
         self.temp_mail = os.getenv("TEMP_MAIL", "").strip().split("@")[0]
         self.domain = os.getenv("DOMAIN", "").strip()
 
+        # 如果临时邮箱为null则加载IMAP
+        if self.temp_mail == 'null':
+            self.imap = True
+            self.imap_server = os.getenv("IMAP_SERVER", "").strip()
+            self.imap_port = os.getenv("IMAP_PORT", "").strip()
+            self.imap_user = os.getenv("IMAP_USER", "").strip()
+            self.imap_pass = os.getenv("IMAP_PASS", "").strip()
+            self.imap_dir = os.getenv("IMAP_DIR", "inbox").strip()
+
         self.check_config()
 
     def get_temp_mail(self):
 
         return self.temp_mail
 
+    def get_imap(self):
+        if not self.imap:
+            return False
+        return {
+            "imap_server": self.imap_server,
+            "imap_port": self.imap_port,
+            "imap_user": self.imap_user,
+            "imap_pass": self.imap_pass,
+            "imap_dir": self.imap_dir
+        }
+
     def get_domain(self):
         return self.domain
 
@@ -40,12 +61,30 @@ class Config:
             raise ValueError("临时邮箱未配置，请在 .env 文件中设置 TEMP_MAIL")
         if not self.check_is_valid(self.domain):
             raise ValueError("域名未配置，请在 .env 文件中设置 DOMAIN")
+        if not self.imap_server == 'null' and not self.check_is_valid(self.imap_server):
+            raise ValueError("IMAP服务器未配置，请在 .env 文件中设置 IMAP_SERVER")
+        if not self.imap_port == 'null' and not self.check_is_valid(self.imap_port):
+            raise ValueError("IMAP端口未配置，请在 .env 文件中设置 IMAP_PORT")
+        if not self.imap_user == 'null' and not self.check_is_valid(self.imap_user):
+            raise ValueError("IMAP用户名未配置，请在 .env 文件中设置 IMAP_USER")
+        if not self.imap_pass == 'null' and not self.check_is_valid(self.imap_pass):
+            raise ValueError("IMAP密码未配置，请在 .env 文件中设置 IMAP_PASS")
+        if not self.imap_dir == 'null' and not self.check_is_valid(self.imap_dir):
+            raise ValueError("IMAP收件箱目录未配置，请在 .env 文件中设置 IMAP_DIRECTORY")
 
     def check_is_valid(self, str):
         return len(str.strip()) > 0
 
     def print_config(self):
-        logging.info(f"\033[32m临时邮箱: {self.temp_mail}\033[0m")
+        # logging.info(f"\033[32m临时邮箱: {self.temp_mail}\033[0m")
+        if self.imap:
+            logging.info(f"\033[32mIMAP服务器: {self.imap_server}\033[0m")
+            logging.info(f"\033[32mIMAP端口: {self.imap_port}\033[0m")
+            logging.info(f"\033[32mIMAP用户名: {self.imap_user}\033[0m")
+            logging.info(f"\033[32mIMAP密码: {'*' * len(self.imap_pass)}\033[0m")
+            logging.info(f"\033[32mIMAP收件箱目录: {self.imap_dir}\033[0m")
+        if self.temp_mail != 'null':
+            logging.info(f"\033[32m临时邮箱: {self.temp_mail}@{self.domain}\033[0m")
         logging.info(f"\033[32m域名: {self.domain}\033[0m")