Create app.py

app.py

@@ -0,0 +1,133 @@
+from fastapi import FastAPI, HTTPException
+from selenium import webdriver
+from selenium.webdriver.chrome.service import Service
+from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.common.by import By
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+import time
+import random
+import string
+import re
+from pydantic import BaseModel
+from typing import Optional
+import logging
+
+# 配置日志
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(levelname)s - %(message)s'
+)
+logger = logging.getLogger(__name__)
+
+app = FastAPI()
+
+class EmailResponse(BaseModel):
+    email: str
+    verification_code: Optional[str] = None
+    error: Optional[str] = None
+
+def setup_driver():
+    chrome_options = Options()
+    chrome_options.add_argument('--no-sandbox')
+    chrome_options.add_argument('--headless')
+    chrome_options.add_argument('--disable-dev-shm-usage')
+    chrome_options.add_argument('--disable-gpu')
+    chrome_options.binary_location = "/usr/bin/chromium"
+    
+    driver = webdriver.Chrome(options=chrome_options)
+    return driver
+
+def generate_email():
+    """生成随机邮箱地址"""
+    random_prefix = ''.join(random.choices(string.ascii_letters + string.digits, k=8))
+    return f"{random_prefix}@youxiang.dev"
+
+def get_verification_code(driver, email, max_retries=10, delay=2):
+    try:
+        email_url = f'https://linshiyou.com/#/{email}'
+        
+        # 打开新标签页并访问邮箱
+        driver.execute_script(f"window.open('{email_url}', '_blank')")
+        driver.switch_to.window(driver.window_handles[-1])
+        logger.info(f"已打开邮箱页面: {email}")
+        
+        for attempt in range(max_retries):
+            try:
+                # 刷新页面
+                driver.refresh()
+                time.sleep(2)
+                
+                # 获取邮件内容
+                body_content = driver.find_element(By.CLASS_NAME, "body").get_attribute('outerHTML')
+                logger.info(f"获取到邮件内容: {body_content}")
+                
+                # 匹配验证码
+                match = re.search(r'\b\d{6}\b', body_content)
+                if match:
+                    verification_code = match.group(0)
+                    logger.info(f"成功获取验证码: {verification_code}")
+                    
+                    # 关闭邮箱标签页
+                    driver.close()
+                    driver.switch_to.window(driver.window_handles[0])
+                    
+                    return verification_code
+                    
+            except Exception as e:
+                logger.error(f"第 {attempt + 1} 次尝试失败: {str(e)}")
+                if attempt == max_retries - 1:
+                    raise
+                
+            time.sleep(delay)
+            logger.info(f"等待验证码，重试第 {attempt + 1} 次")
+        
+        # 如果获取失败，确保回到主标签页
+        try:
+            driver.close()
+            driver.switch_to.window(driver.window_handles[0])
+        except:
+            pass
+            
+        return None
+        
+    except Exception as e:
+        logger.error(f"获取验证码时发生错误: {str(e)}")
+        # 确保清理标签页
+        try:
+            driver.close()
+            driver.switch_to.window(driver.window_handles[0])
+        except:
+            pass
+        raise
+
+@app.get("/")
+def read_root():
+    return {"message": "临时邮箱服务运行中"}
+
+@app.post("/generate-email")
+def create_email():
+    email = generate_email()
+    logger.info(f"生成新邮箱: {email}")
+    return EmailResponse(email=email)
+
+@app.get("/get-verification-code/{email}")
+def get_code(email: str):
+    driver = None
+    try:
+        logger.info(f"开始获取邮箱验证码: {email}")
+        driver = setup_driver()
+        code = get_verification_code(driver, email)
+        if code:
+            return EmailResponse(email=email, verification_code=code)
+        raise HTTPException(status_code=404, detail="未找到验证码")
+    except Exception as e:
+        logger.error(f"获取验证码失败: {str(e)}")
+        raise HTTPException(status_code=500, detail=str(e))
+    finally:
+        if driver:
+            driver.quit()
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)