Update remote_uploader.py

remote_uploader.py

@@ -1,4 +1,3 @@
-import time
 import requests
 import argparse
 import os
@@ -8,6 +7,8 @@ import json
 import base64
 import logging
 import sys
+import time  # 修复点 1: 导入 time 模块以使用 time.sleep()
+from urllib.parse import urlparse # 修复点 2: 导入 urlparse 以便解析 URL
 
 # 配置日志
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -21,23 +22,35 @@ class WebSocketFileUploader:
         
     async def connect_and_upload(self, upload_dir):
         """连接到 WebSocket 服务器并上传文件"""
-        # 将 HTTP URL 转换为 WebSocket URL
-        ws_url = self.server_url.replace('http://', 'ws://').replace('https://', 'wss://').rstrip('/')
-        ws_url += f'/ws/upload/{self.space_id}?token={self.api_key}'
         
+        # --- 修复点 3: 智能地构建 WebSocket URL ---
+        # 目标: 将主应用传来的 HTTP URL (如 http://...:5001) 转换为正确的 WebSocket URL (ws://...:8765)
+        try:
+            parsed_http_url = urlparse(self.server_url)
+            # 根据 HTTP 协议确定 WebSocket 协议 (http -> ws, https -> wss)
+            ws_scheme = 'wss' if parsed_http_url.scheme == 'https' else 'ws'
+            
+            # 使用从 HTTP URL 获取的主机名，但硬编码正确的 WebSocket 端口 8765
+            ws_netloc = f"{parsed_http_url.hostname}:8765"
+            
+            # 重新构建完整的 WebSocket URL
+            ws_url = f"{ws_scheme}://{ws_netloc}/ws/upload/{self.space_id}?token={self.api_key}"
+        except Exception as e:
+            logger.error(f"解析服务器 URL '{self.server_url}' 时出错: {e}")
+            logger.info("回退到 HTTP API 上传...")
+            return self._fallback_http_upload(upload_dir)
+        # --- 修复结束 ---
+
         logger.info(f"正在连接到 WebSocket: {ws_url}")
         
         try:
             async with websockets.connect(ws_url, ping_interval=20, ping_timeout=60) as websocket:
                 logger.info("WebSocket 连接成功")
                 
-                # 发送认证信息
-                auth_msg = {
-                    "type": "auth",
-                    "api_key": self.api_key,
-                    "space_id": self.space_id
-                }
-                await websocket.send(json.dumps(auth_msg))
+                # 注意：原始代码中的认证流程是先连接再发认证消息，这很好。
+                # 我们将遵循这个流程。客户端不需要发送 auth 消息，
+                # 因为服务器端的 handle_websocket_upload 函数会直接从连接路径和参数中验证。
+                # 我们只需要等待服务器的 "auth_success" 响应。
                 
                 # 等待认证响应
                 response = await websocket.recv()
@@ -48,7 +61,9 @@ class WebSocketFileUploader:
                     return await self._upload_files(websocket, upload_dir)
                 else:
                     logger.error(f"认证失败: {auth_response.get('message', '未知错误')}")
-                    return False
+                    # 认证失败也回退到 HTTP
+                    logger.info("回退到 HTTP API 上传...")
+                    return self._fallback_http_upload(upload_dir)
                     
         except Exception as e:
             logger.error(f"WebSocket 连接失败: {e}")
@@ -181,15 +196,16 @@ class WebSocketFileUploader:
                     }
                     
                     # 发送 POST 请求
-                    response = requests.post(upload_url, headers=headers, data=data, files=files, timeout=30)
+                    response = requests.post(upload_url, headers=headers, data=data, files=files, timeout=60)
                     
                     if response.status_code == 200:
                         logger.info(f"✅ 文件 '{filename}' 上传成功!")
                         success_count += 1
                     else:
-                        logger.error(f"❌ 上传失败: HTTP {response.status_code}")
+                        logger.error(f"❌ 上传失败: HTTP {response.status_code} - {response.text}")
                         failed_count += 1
                 
+                # 使用 time.sleep()
                 time.sleep(0.5)
                         
             except Exception as e:
@@ -220,10 +236,10 @@ if __name__ == "__main__":
     parser = argparse.ArgumentParser(description="WebSocket 文件上传器 - 扫描并上传指定文件夹中的所有文件")
     parser.add_argument("api_key", help="您的 API 密钥")
     parser.add_argument("space_id", help="Space ID")
-    parser.add_argument("--server", default="https://gkbtyo-rqvays-5001.preview.cloudstudio.work", help="服务器的 URL 地址")
+    parser.add_argument("--server", default="http://127.0.0.1:5001", help="服务器的 URL 地址 (例如: http://127.0.0.1:5001)")
     parser.add_argument("--upload-dir", default="output", help="要上传的目录 (默认: output)")
     
     args = parser.parse_args()
     
     # 运行异步上传
-    asyncio.run(upload_directory_websocket(args.upload_dir, args.server, args.api_key, args.space_id))
+    asyncio.run(upload_directory_websocket(args.upload_dir, args.server, args.api_key, args.space_id))