Initial commit

2026-05-24 21:47:44 +08:00
commit 2ce35563c4
20 changed files with 3698 additions and 0 deletions
@@ -0,0 +1,139 @@
+#!/usr/bin/env python3
+"""
+腾讯混元3D 邮箱验证码登录 CLI 工具 (CloakBrowser 持久化版本)
+第一次输入邮箱回车后自动发送验证码
+第二次输入验证码后回车自动点击登录按钮
+登录状态会自动保存到 ./hunyuan3d_profile，下次运行无需重新登录
+"""
+
+import os
+import sys
+import time
+from cloakbrowser import launch_persistent_context
+from ..config import get_profile_dir, get_cookie_path
+
+PROFILE_DIR = str(get_profile_dir())
+
+
+def _extract_and_save_cookies(context):
+    """从浏览器上下文提取 Cookie 并保存到文件"""
+    try:
+        cookies = context.cookies()
+        relevant_names = {"hunyuan_user", "hunyuan_token", "hunyuan_source", "hy_user"}
+        cookie_dict = {c["name"]: c["value"] for c in cookies if c["name"] in relevant_names}
+        if cookie_dict:
+            cookie_str = "; ".join(f"{k}={v}" for k, v in cookie_dict.items())
+            cookie_path = get_cookie_path()
+            cookie_path.write_text(cookie_str, encoding="utf-8")
+            print(f"Cookie 已自动保存到: {cookie_path}")
+            return True
+        else:
+            print("警告: 未找到 hunyuan 相关 Cookie，API 客户端可能无法使用")
+    except Exception as e:
+        print(f"自动提取 Cookie 失败: {e}")
+    return False
+
+
+def main():
+    # 使用持久化上下文，cookie 和登录状态会保存到 PROFILE_DIR
+    context = launch_persistent_context(PROFILE_DIR, headless=False)
+    page = context.new_page()
+
+    print("正在打开腾讯混元3D...")
+    page.goto("https://3d.hunyuan.tencent.com/")
+    time.sleep(2)
+
+    # 检查是否已经是登录状态（没有登录按钮说明已登录）
+    login_btn = page.locator("button").filter(has_text="登录").first
+    if login_btn.count() == 0:
+        print("检测到已有登录状态，无需重新登录。")
+        print(f"当前页面: {page.url}")
+    else:
+        # 未登录，走登录流程
+        login_btn.click()
+        time.sleep(1.5)
+
+        # 切换到邮箱登录
+        email_tab = page.locator("text=邮箱").first
+        if email_tab.count() > 0:
+            email_tab.click()
+            time.sleep(1.5)
+        else:
+            print("未找到邮箱登录选项")
+            context.close()
+            return
+
+        # 定位元素
+        email_input = page.locator('input[placeholder="请输入邮箱地址"]')
+        code_input = page.locator('input[type="number"][placeholder="请输入邮箱验证码"]')
+        send_code_btn = page.locator("a.hyc-email-login__send-code")
+        login_submit_btn = page.locator("button.hyc-email-login__btn")
+        checkbox = page.locator(".t-checkbox__former")
+
+        # 第一次交互：输入邮箱并发送验证码
+        print("\n============================================")
+        email = input("请输入邮箱地址，按回车发送验证码: ").strip()
+        if not email:
+            print("邮箱不能为空，退出")
+            context.close()
+            return
+
+        email_input.fill(email)
+        time.sleep(0.5)
+
+        # 勾选协议（如果未勾选）
+        if checkbox.count() > 0:
+            is_checked = checkbox.evaluate("el => el.checked")
+            if not is_checked:
+                checkbox.evaluate("el => el.click()")
+                time.sleep(0.3)
+
+        if send_code_btn.count() > 0:
+            send_code_btn.click()
+            print("已点击发送验证码，请查收邮件...")
+        else:
+            print("未找到发送验证码按钮")
+            context.close()
+            return
+
+        # 第二次交互：输入验证码并登录
+        print("\n============================================")
+        code = input("请输入邮箱验证码，按回车登录: ").strip()
+        if not code:
+            print("验证码不能为空，退出")
+            context.close()
+            return
+
+        code_input.fill(code)
+        time.sleep(0.5)
+
+        # 点击登录
+        if login_submit_btn.count() > 0:
+            login_submit_btn.click()
+            print("已点击登录按钮，等待跳转...")
+        else:
+            print("未找到登录提交按钮")
+            context.close()
+            return
+
+        # 等待登录成功跳转
+        try:
+            page.wait_for_url(lambda url: "login" not in url, timeout=30000)
+            print(f"\n登录成功！当前页面: {page.url}")
+            print(f"登录状态已保存到: {os.path.abspath(PROFILE_DIR)}")
+            _extract_and_save_cookies(context)
+        except Exception:
+            print("\n登录可能仍在处理中，或出现错误。请检查浏览器状态。")
+
+    # 保持浏览器打开
+    print("\n按 Enter 键关闭浏览器并退出...")
+    input()
+    context.close()
+
+
+if __name__ == "__main__":
+    try:
+        main()
+    except KeyboardInterrupt:
+        print("\n已取消")
+        sys.exit(0)