Update app.py

app.py

@@ -102,42 +102,42 @@ class AuthManager:
         self.logger.info(f"\033[92mRefresh Token: {self.refresh_token}\033[0m")
         
     def fetch_apikey(self) -> str:
-    """获取API密钥。"""
-    if self.api_key:
-        return self.api_key
-
-    try:
-        url = "https://chat.notdiamond.ai/login"
-        headers = {
-            'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/128.0.0.0 Safari/537.36'
-        }
-        response = self.session.get(url, headers=headers)
-        response.raise_for_status()
-
-        # 匹配 <script> 标签中的 JS 文件路径
-        match = re.search(r'<script src="(/_next/static/chunks/app/layout-[^"]+\.js)"', response.text)
-        if not match:
-            self.logger.warning("未找到匹配的脚本标签")
-            return ""
-
-        js_url = f"https://chat.notdiamond.ai{match.group(1)}"
-        js_response = self.session.get(js_url, headers=headers)
-        js_response.raise_for_status()
-        
-        # 匹配 API key
-        api_key_match = re.search(r'\("https://spuckhogycrxcbomznwo\.supabase\.co","([^"]+)"\)', js_response.text)
-        if api_key_match:
-            self.api_key = api_key_match.group(1)
-            logger.info(f"Extracted API key: {self.api_key}")
+        """获取API密钥。"""
+        if self.api_key:
             return self.api_key
-        else:
-            self.logger.error("未能匹配API key")
+    
+        try:
+            url = "https://chat.notdiamond.ai/login"
+            headers = {
+                'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/128.0.0.0 Safari/537.36'
+            }
+            response = self.session.get(url, headers=headers)
+            response.raise_for_status()
+    
+            # 匹配 <script> 标签中的 JS 文件路径
+            match = re.search(r'<script src="(/_next/static/chunks/app/layout-[^"]+\.js)"', response.text)
+            if not match:
+                self.logger.warning("未找到匹配的脚本标签")
+                return ""
+    
+            js_url = f"https://chat.notdiamond.ai{match.group(1)}"
+            js_response = self.session.get(js_url, headers=headers)
+            js_response.raise_for_status()
+            
+            # 匹配 API key
+            api_key_match = re.search(r'\("https://spuckhogycrxcbomznwo\.supabase\.co","([^"]+)"\)', js_response.text)
+            if api_key_match:
+                self.api_key = api_key_match.group(1)
+                logger.info(f"Extracted API key: {self.api_key}")
+                return self.api_key
+            else:
+                self.logger.error("未能匹配API key")
+                return ""
+    
+        except requests.RequestException as e:
+            self.logger.error(f"请求JS文件时发生错误: {e}")
             return ""
 
-    except requests.RequestException as e:
-        self.logger.error(f"请求JS文件时发生错误: {e}")
-        return ""
-
     def login(self) -> None:
         """使用电子邮件和密码进行用户登录，并获取用户信息。"""
         api_key = self.fetch_apikey()