cnhkmcp 1.1.6__py3-none-any.whl → 1.1.7__py3-none-any.whl

cnhkmcp/__init__.py

@@ -50,7 +50,7 @@ from .untracked.forum_functions import (
     read_full_forum_post
 )
 
-__version__ = "1.1.6"
+__version__ = "1.1.7"
 __author__ = "CNHK"
 __email__ = "cnhk@example.com"
 

cnhkmcp/untracked/platform_functions.py

@@ -81,8 +81,8 @@ class BrainApiClient:
         print(f"[{level}] {message}", file=sys.stderr)
     
     async def authenticate(self, email: str, password: str) -> Dict[str, Any]:
-        """Authenticate with WorldQuant BRAIN platform using Basic Authentication."""
-        self.log("🔐 Starting Basic Authentication process...", "INFO")
+        """Authenticate with WorldQuant BRAIN platform with biometric support."""
+        self.log("🔐 Starting Authentication process...", "INFO")
         
         try:
             # Store credentials for potential re-authentication
@@ -106,7 +106,7 @@ class BrainApiClient:
             
             # Check for successful authentication (status code 201)
             if response.status_code == 201:
-                self.log("✅ Basic Authentication successful", "SUCCESS")
+                self.log("✅ Authentication successful", "SUCCESS")
                 
                 # Check if JWT token was automatically stored by session
                 jwt_token = self.session.cookies.get('t')
@@ -120,10 +120,26 @@ class BrainApiClient:
                     'user': {'email': email},
                     'status': 'authenticated',
                     'permissions': ['read', 'write'],
-                    'message': 'Basic Authentication successful',
+                    'message': 'Authentication successful',
                     'status_code': response.status_code,
                     'has_jwt': jwt_token is not None
                 }
+            
+            # Check if biometric authentication is required (401 with persona)
+            elif response.status_code == 401:
+                www_auth = response.headers.get("WWW-Authenticate")
+                location = response.headers.get("Location")
+                
+                if www_auth == "persona" and location:
+                    self.log("🔴 Biometric authentication required", "INFO")
+                    
+                    # Handle biometric authentication
+                    from urllib.parse import urljoin
+                    biometric_url = urljoin(response.url, location)
+                    
+                    return await self._handle_biometric_auth(biometric_url, email)
+                else:
+                    raise Exception("Incorrect email or password")
             else:
                 raise Exception(f"Authentication failed with status code: {response.status_code}")
                     
@@ -134,6 +150,97 @@ class BrainApiClient:
             self.log(f"❌ Authentication failed: {str(e)}", "ERROR")
             raise
     
+    async def _handle_biometric_auth(self, biometric_url: str, email: str) -> Dict[str, Any]:
+        """Handle biometric authentication using browser automation."""
+        self.log("🌐 Starting biometric authentication...", "INFO")
+        
+        try:
+            # Import selenium for browser automation
+            from selenium import webdriver
+            from selenium.webdriver.chrome.options import Options
+            import time
+            
+            # Setup Chrome options
+            options = Options()
+            options.add_argument('--no-sandbox')
+            options.add_argument('--disable-dev-shm-usage')
+            
+            driver = None
+            try:
+                # Open browser with timeout
+                driver = webdriver.Chrome(options=options)
+                # Set a short timeout so it doesn't wait forever
+                driver.set_page_load_timeout(80)  # Only wait 5 seconds
+                
+                self.log("🌐 Opening browser for biometric authentication...", "INFO")
+                
+                # Try to open the URL but handle timeout
+                try:
+                    driver.get(biometric_url)
+                    self.log("✅ Browser page loaded successfully", "SUCCESS")
+                except Exception as timeout_error:
+                    self.log(f"⚠️ Page load timeout (expected): {str(timeout_error)[:50]}...", "WARNING")
+                    self.log("✅ Browser window is open for biometric authentication", "INFO")
+                
+                # Print instructions
+                print("\n" + "="*60, file=sys.stderr)
+                print("🔒 BIOMETRIC AUTHENTICATION REQUIRED", file=sys.stderr)
+                print("="*60, file=sys.stderr)
+                print("🌐 Browser window is open with biometric authentication page", file=sys.stderr)
+                print("🔧 Complete the biometric authentication in the browser", file=sys.stderr)
+                print("⏳ The system will automatically check when you're done...", file=sys.stderr)
+                print("="*60, file=sys.stderr)
+                
+                # Keep checking until authentication is complete
+                max_attempts = 60  # 5 minutes maximum (60 * 5 seconds)
+                attempt = 0
+                
+                while attempt < max_attempts:
+                    time.sleep(5)  # Check every 5 seconds
+                    attempt += 1
+                    
+                    # Check if authentication completed
+                    check_response = self.session.post(biometric_url)
+                    self.log(f"🔄 Checking authentication status (attempt {attempt}/{max_attempts}): {check_response.status_code}", "INFO")
+                    
+                    if check_response.status_code == 201:
+                        self.log("✅ Biometric authentication successful!", "SUCCESS")
+                        
+                        # Close browser
+                        driver.quit()
+                        
+                        # Check JWT token
+                        jwt_token = self.session.cookies.get('t')
+                        if jwt_token:
+                            self.log("✅ JWT token received", "SUCCESS")
+                        
+                        # Return success response
+                        return {
+                            'user': {'email': email},
+                            'status': 'authenticated',
+                            'permissions': ['read', 'write'],
+                            'message': 'Biometric authentication successful',
+                            'status_code': check_response.status_code,
+                            'has_jwt': jwt_token is not None
+                        }
+                
+                # If we get here, authentication timed out
+                if driver:
+                    driver.quit()
+                raise Exception("Biometric authentication timed out")
+                
+            except Exception as driver_error:
+                if driver:
+                    try:
+                        driver.quit()
+                    except:
+                        pass
+                raise Exception(f"Browser automation error: {driver_error}")
+                
+        except Exception as e:
+            self.log(f"❌ Biometric authentication failed: {str(e)}", "ERROR")
+            raise
+    
     async def is_authenticated(self) -> bool:
         """Check if currently authenticated using JWT token."""
         try:

{cnhkmcp-1.1.6.dist-info → cnhkmcp-1.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cnhkmcp
-Version: 1.1.6
+Version: 1.1.7
 Summary: A comprehensive Model Context Protocol (MCP) server for quantitative trading platform integration
 Home-page: https://github.com/cnhk/cnhkmcp
 Author: CNHK

cnhkmcp-1.1.7.dist-info/RECORD

@@ -0,0 +1,10 @@
+cnhkmcp/__init__.py,sha256=zA4Je1BkR8_9NorDUU5n_QEozwqBquxVGsUrGk3yHjc,2758
+cnhkmcp/untracked/forum_functions.py,sha256=78wzvN_UYWwbWU40q8_FJNSFPJnND6W9ZRey6MSSiEk,45516
+cnhkmcp/untracked/platform_functions.py,sha256=sDW-tOhWUT41YuO-A3ugimkrNEj0ShMEUYp4EFVQpeA,77441
+cnhkmcp/untracked/user_config.json,sha256=_INn1X1qIsITrmEno-BRlQOAGm9wnNCw-6B333DEvnk,695
+cnhkmcp-1.1.7.dist-info/licenses/LICENSE,sha256=QLxO2eNMnJQEdI_R1UV2AOD-IvuA8zVrkHWA4D9gtoc,1081
+cnhkmcp-1.1.7.dist-info/METADATA,sha256=sNnIQ4NvGvk6qScb5HBmfhUiP9-OiDG1hsGQLn-RazA,5171
+cnhkmcp-1.1.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+cnhkmcp-1.1.7.dist-info/entry_points.txt,sha256=lTQieVyIvjhSMK4fT-XwnccY-JBC1H4vVQ3V9dDM-Pc,70
+cnhkmcp-1.1.7.dist-info/top_level.txt,sha256=x--ibUcSgOS9Z_RWK2Qc-vfs7DaXQN-WMaaxEETJ1Bw,8
+cnhkmcp-1.1.7.dist-info/RECORD,,

cnhkmcp-1.1.6.dist-info/RECORD

@@ -1,10 +0,0 @@
-cnhkmcp/__init__.py,sha256=nmpcykQd12FFDQYnKS368uGeZXoKT3sMHH7SFulZGLA,2758
-cnhkmcp/untracked/forum_functions.py,sha256=78wzvN_UYWwbWU40q8_FJNSFPJnND6W9ZRey6MSSiEk,45516
-cnhkmcp/untracked/platform_functions.py,sha256=IpqBv8q7xo8iYwPgreNjcsk_k0k9_4zV1i4afPFssBI,72253
-cnhkmcp/untracked/user_config.json,sha256=_INn1X1qIsITrmEno-BRlQOAGm9wnNCw-6B333DEvnk,695
-cnhkmcp-1.1.6.dist-info/licenses/LICENSE,sha256=QLxO2eNMnJQEdI_R1UV2AOD-IvuA8zVrkHWA4D9gtoc,1081
-cnhkmcp-1.1.6.dist-info/METADATA,sha256=4NPDEazaUY0jyaGsFEm1g9SHwcLAHe7HGX3w8NZ-eek,5171
-cnhkmcp-1.1.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-cnhkmcp-1.1.6.dist-info/entry_points.txt,sha256=lTQieVyIvjhSMK4fT-XwnccY-JBC1H4vVQ3V9dDM-Pc,70
-cnhkmcp-1.1.6.dist-info/top_level.txt,sha256=x--ibUcSgOS9Z_RWK2Qc-vfs7DaXQN-WMaaxEETJ1Bw,8
-cnhkmcp-1.1.6.dist-info/RECORD,,