修复bug

main.py

@@ -175,6 +175,20 @@ async def register_account(
 
         # 保存成功的账号
         if result["status"] == "success":
+            # 注册成功后，立即登录获取 token
+            logger.info(f"[Task {task_id}] Registration done, now logging in to get token...")
+
+            from core.login_flow import LoginFlow
+            login_flow = LoginFlow(session, result["email"], result["password"])
+            login_result = await login_flow.run()
+
+            if login_result.get("status") == "success":
+                result["access_token"] = login_result.get("access_token")
+                result["session_token"] = login_result.get("session_token")
+                logger.success(f"[Task {task_id}] ✅ Token obtained!")
+            else:
+                logger.warning(f"[Task {task_id}] ⚠️ Login failed: {login_result.get('error')}")
+
             await save_account(result, config.accounts_output_file)
             logger.success(
                 f"[Task {task_id}] ✅ Account created: {result['email']}:{result['password']}"