x

services/capsolver.py

@@ -0,0 +1,93 @@
+import asyncio
+import aiohttp
+from loguru import logger
+from typing import Optional
+import time
+
+class Capsolver:
+    def __init__(self, api_key: str, website_url: str, website_key: str):
+        self.api_key = api_key
+        self.website_url = website_url
+        self.website_key = website_key
+        self.base_url = "https://api.capsolver.com"
+
+    async def create_task(self) -> Optional[str]:
+        """创建验证码任务"""
+        async with aiohttp.ClientSession() as session:
+            payload = {
+                "clientKey": self.api_key,
+                "task": {
+                    "type": "AntiTurnstileTaskProxyLess",
+                    "websiteURL": self.website_url,
+                    "websiteKey": self.website_key,
+                }
+            }
+            
+            async with session.post(f"{self.base_url}/createTask", json=payload) as resp:
+                result = await resp.json()
+                if result.get("errorId") > 0:
+                    logger.error(f"创建任务失败: {result.get('errorDescription')}")
+                    return None
+                return result.get("taskId")
+
+    async def get_task_result(self, task_id: str) -> Optional[dict]:
+        """获取任务结果"""
+        async with aiohttp.ClientSession() as session:
+            payload = {
+                "clientKey": self.api_key,
+                "taskId": task_id
+            }
+            
+            async with session.post(f"{self.base_url}/getTaskResult", json=payload) as resp:
+                result = await resp.json()
+                if result.get("errorId") > 0:
+                    logger.error(f"获取结果失败: {result.get('errorDescription')}")
+                    return None
+                    
+                if result.get("status") == "ready":
+                    return result.get("solution", {})
+                return None
+
+    async def solve_turnstile(self) -> Optional[str]:
+        """
+        解决 Turnstile 验证码
+        """
+        task_id = await self.create_task()
+        if not task_id:
+            raise Exception("创建验证码任务失败")
+        
+        # 增加重试次数限制和超时时间控制
+        max_retries = 5  # 减少最大重试次数
+        retry_delay = 2  # 设置重试间隔为2秒
+        timeout = 15     # 设置总超时时间为15秒
+        
+        start_time = time.time()
+        for attempt in range(1, max_retries + 1):
+            try:
+                logger.debug(f"第 {attempt} 次尝试获取验证码结果")
+                result = await self.get_task_result(task_id)
+                
+                if result and "token" in result:
+                    token = result["token"]
+                    logger.success(f"成功获取验证码 token: {token[:40]}...")
+                    return token
+                    
+                # 检查是否超时
+                if time.time() - start_time > timeout:
+                    logger.error("验证码请求总时间超过15秒")
+                    break
+                    
+                await asyncio.sleep(retry_delay)
+                
+            except Exception as e:
+                logger.error(f"获取验证码结果失败: {str(e)}")
+                if attempt == max_retries:
+                    raise
+                
+                if time.time() - start_time > timeout:
+                    logger.error("验证码请求总时间超过15秒")
+                    break
+                    
+                await asyncio.sleep(retry_delay)
+                
+        raise Exception("验证码解决失败: 达到最大重试次数或超时")