capture more capcha info

core/mexc_webclient/mexc_futures_client.py

@@ -22,6 +22,17 @@ from urllib.parse import urlencode
 
 logger = logging.getLogger(__name__)
 
+class MEXCSessionManager:
+    def __init__(self):
+        self.captcha_token = None
+
+    def get_captcha_token(self) -> str:
+        return self.captcha_token if self.captcha_token else ""
+
+    def save_captcha_token(self, token: str):
+        self.captcha_token = token
+        logger.info("MEXC: Captcha token saved in session manager")
+
 class MEXCFuturesWebClient:
     """
     MEXC Futures Web Client that mimics browser behavior for futures trading.
@@ -30,30 +41,27 @@ class MEXCFuturesWebClient:
     the exact HTTP requests made by their web interface.
     """
     
-    def __init__(self, session_cookies: Dict[str, str] = None):
+    def __init__(self, api_key: str, api_secret: str, user_id: str, base_url: str = 'https://www.mexc.com', headless: bool = True):
         """
         Initialize the MEXC Futures Web Client
         
         Args:
-            session_cookies: Dictionary of cookies from an authenticated browser session
+            api_key: API key for authentication
+            api_secret: API secret for authentication
+            user_id: User ID for authentication
+            base_url: Base URL for the MEXC website
+            headless: Whether to run the browser in headless mode
         """
-        self.session = requests.Session()
-        
-        # Base URLs for different endpoints
-        self.base_url = "https://www.mexc.com"
-        self.futures_api_url = "https://futures.mexc.com/api/v1"
-        self.captcha_url = f"{self.base_url}/ucgateway/captcha_api/captcha/robot"
-        
-        # Session state
+        self.api_key = api_key
+        self.api_secret = api_secret
+        self.user_id = user_id
+        self.base_url = base_url
         self.is_authenticated = False
-        self.user_id = None
-        self.auth_token = None
-        self.fingerprint = None
-        self.visitor_id = None
-        
-        # Load session cookies if provided
-        if session_cookies:
-            self.load_session_cookies(session_cookies)
+        self.headless = headless
+        self.session = requests.Session()
+        self.session_manager = MEXCSessionManager()  # Adding session_manager attribute
+        self.captcha_url = f'{base_url}/ucgateway/captcha_api'
+        self.futures_api_url = "https://futures.mexc.com/api/v1"
         
         # Setup default headers that mimic a real browser
         self.setup_browser_headers()
@@ -142,37 +150,59 @@ class MEXCFuturesWebClient:
         endpoint = f"robot.future.{side}.{symbol}.{leverage}"
         url = f"{self.captcha_url}/{endpoint}"
         
-        # Setup headers for captcha request
+        # Attempt to get captcha token from session manager
+        captcha_token = self.session_manager.get_captcha_token()
+        if not captcha_token:
+            logger.warning("MEXC: No captcha token available, attempting to fetch from browser")
+            captcha_token = self._extract_captcha_token_from_browser()
+            if captcha_token:
+                self.session_manager.save_captcha_token(captcha_token)
+            else:
+                logger.error("MEXC: Failed to extract captcha token from browser")
+                return False
+        
         headers = {
             'Content-Type': 'application/json',
             'Language': 'en-GB',
             'Referer': f'{self.base_url}/en-GB/futures/{symbol}?type=linear_swap',
             'trochilus-uid': self.user_id if self.user_id else '',
-            'trochilus-trace-id': f"{uuid.uuid4()}-{int(time.time() * 1000) % 10000:04d}"
+            'trochilus-trace-id': f"{uuid.uuid4()}-{int(time.time() * 1000) % 10000:04d}",
+            'captcha-token': captcha_token
         }
         
-        # Add captcha token if available (this would need to be extracted from browser)
-        # For now, we'll make the request without it and see what happens
-        
+        logger.info(f"MEXC: Verifying captcha for {endpoint}")
         try:
             response = self.session.get(url, headers=headers, timeout=10)
-            
             if response.status_code == 200:
                 data = response.json()
-                if data.get('success') and data.get('code') == 0:
-                    logger.info(f"MEXC: Captcha verification successful for {side} {symbol}")
+                if data.get('success'):
+                    logger.info(f"MEXC: Captcha verified successfully for {endpoint}")
                     return True
                 else:
-                    logger.warning(f"MEXC: Captcha verification failed: {data}")
+                    logger.error(f"MEXC: Captcha verification failed for {endpoint}: {data}")
                     return False
             else:
-                logger.error(f"MEXC: Captcha request failed with status {response.status_code}")
+                logger.error(f"MEXC: Captcha verification request failed with status {response.status_code}: {response.text}")
                 return False
-                
         except Exception as e:
-            logger.error(f"MEXC: Captcha verification error: {e}")
+            logger.error(f"MEXC: Captcha verification error for {endpoint}: {str(e)}")
             return False
     
+    def _extract_captcha_token_from_browser(self) -> str:
+        """
+        Extract captcha token from browser session using stored cookies or requests.
+        This is a placeholder for actual implementation which would interact with browser data.
+        """
+        try:
+            # Placeholder for extracting token from browser session or stored data
+            # In a real scenario, this would parse the mexc_requests JSON file or interact with Selenium
+            logger.info("MEXC: Attempting to extract captcha token from browser data")
+            # For now, return empty string as placeholder
+            return ""
+        except Exception as e:
+            logger.error(f"MEXC: Error extracting captcha token from browser: {str(e)}")
+            return ""
+    
     def generate_signature(self, method: str, path: str, params: Dict[str, Any], 
                           timestamp: int, nonce: int) -> str:
         """