proxy pool retry added

freeplay2api.py

@@ -159,13 +159,6 @@ class FreeplayClient:
 
     def register(self) -> Optional[Dict]:
         url = "https://app.freeplay.ai/app_data/auth/signup"
-        payload = {
-            "email": self.faker.email(),
-            "password": f"aA1!{uuid.uuid4().hex[:8]}",
-            "account_name": self.faker.name(),
-            "first_name": self.faker.first_name(),
-            "last_name": self.faker.last_name(),
-        }
         headers = {
             "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/137.0.0.0 Safari/537.36",
             "Accept": "application/json",
@@ -173,35 +166,59 @@ class FreeplayClient:
             "origin": "https://app.freeplay.ai",
             "referer": "https://app.freeplay.ai/signup",
         }
-        try:
-            proxy_info = self.proxy_pool.get_proxy() if self.proxy_pool else None
-            proxies = {"http": proxy_info['full'], "https": proxy_info['full']} if proxy_info else None
-            response = requests.post(
-                url,
-                data=json.dumps(payload),
-                headers=headers,
-                proxies=proxies,
-                timeout=20,
-            )
-            response.raise_for_status()
-            project_id = response.json()["project_id"]
-            session = response.cookies.get("session")
-            if project_id and session:
-                return {
-                    "email": payload["email"],
-                    "password": payload["password"],
-                    "session_id": session,
-                    "project_id": project_id,
-                    "balance": 5.0,  # 新注册账号默认5刀
+
+        while True:
+            proxy_info = None
+            try:
+                payload = {
+                    "email": self.faker.email(),
+                    "password": f"aA1!{uuid.uuid4().hex[:8]}",
+                    "account_name": self.faker.name(),
+                    "first_name": self.faker.first_name(),
+                    "last_name": self.faker.last_name(),
                 }
-        except requests.exceptions.ProxyError as e:
-            logging.warning(f"Proxy error during registration: {e}")
-            if self.proxy_pool and proxy_info:
-                self.proxy_pool.remove_proxy(proxy_info['ip'], proxy_info['port'])
-            return None
-        except Exception as e:
-            logging.error(f"Account registration failed: {e}")
-            return None
+                
+                proxy_info = self.proxy_pool.get_proxy() if self.proxy_pool else None
+                proxies = {"http": proxy_info['full'], "https": proxy_info['full']} if proxy_info else None
+                
+                response = requests.post(
+                    url,
+                    data=json.dumps(payload),
+                    headers=headers,
+                    proxies=proxies,
+                    timeout=20,
+                )
+                
+                if response.status_code == 200:
+                    data = response.json()
+                    project_id = data.get("project_id")
+                    session = response.cookies.get("session")
+                    if project_id and session:
+                        logging.info(f"Successfully registered account: {payload['email']}")
+                        return {
+                            "email": payload["email"],
+                            "password": payload["password"],
+                            "session_id": session,
+                            "project_id": project_id,
+                            "balance": 5.0,
+                        }
+                
+                logging.warning(f"Registration attempt failed with status {response.status_code}: {response.text}")
+
+            except requests.exceptions.ProxyError as e:
+                logging.warning(f"Proxy error during registration: {e}. Retrying with a new proxy...")
+                if self.proxy_pool and proxy_info:
+                    self.proxy_pool.remove_proxy(proxy_info['ip'], proxy_info['port'])
+            
+            except requests.exceptions.RequestException as e:
+                logging.warning(f"Request exception during registration: {e}. Retrying...")
+                if self.proxy_pool and proxy_info:
+                    self.proxy_pool.remove_proxy(proxy_info['ip'], proxy_info['port'])
+
+            except Exception as e:
+                logging.error(f"An unexpected error occurred during registration: {e}. Retrying...")
+
+            time.sleep(1)
 
     def chat(
         self,