@hupan56/wlkj 3.3.1 → 3.3.2

package/templates/qoder/scripts/domain/kg/switch_project.py

@@ -1,158 +1,158 @@
-#!/usr/bin/env python
-"""wlkj 项目切换脚本：选项目 → 拉平台 token → 写 mcp_config.json。
-
-用法：
-    python switch_project.py                          # 交互式：列出项目 → 选 → 写配置
-    python switch_project.py --email a@b.c --pw xxx  # 非交互：直接指定
-    python switch_project.py --project-uuid <UUID>   # 非交互：直接指定项目
-
-需工作台后端在跑（默认 http://10.89.7.120 云）。
-"""
-import json
-import os
-import sys
-import urllib.request
-import urllib.error
-
-# 宿主根：脚本部署在 {host}/scripts/domain/kg/，4 级 dirname = {host}（.qoder）
-HOST_DIR = os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
-REPO_ROOT = os.path.dirname(HOST_DIR)  # qoderAll/
-
-
-def _find_config_path():
-    """按宿主探测 mcp_config.json：当前宿主根 → .qoder。"""
-    for p in (
-        os.path.join(HOST_DIR, "mcp_config.json"),        # 当前宿主根
-        os.path.join(REPO_ROOT, ".qoder", "mcp_config.json"),
-    ):
-        if os.path.isfile(p):
-            return p
-    return os.path.join(HOST_DIR, "mcp_config.json")  # 都没有则写当前宿主根
-
-
-CONFIG_PATH = _find_config_path()
-
-# 平台地址（默认 10010，可 env 覆盖）
-BASE_URL = os.environ.get("WLKJ_BASE_URL", "http://10.89.7.120")
-
-
-def bind(email, password, base_url=BASE_URL):
-    """调平台 /api/auth/bind，返回 {projects:[{id,name,token,mcp_config}], base_url}。"""
-    data = json.dumps({"email": email, "password": password, "base_url": base_url}).encode()
-    req = urllib.request.Request(
-        f"{base_url}/api/auth/bind",
-        data=data, method="POST",
-        headers={"Content-Type": "application/json"},
-    )
-    try:
-        resp = urllib.request.urlopen(req, timeout=10)
-        return json.loads(resp.read())
-    except urllib.error.HTTPError as e:
-        body = e.read().decode("utf-8", errors="replace")[:200]
-        print(f"ERROR: 平台返回 {e.code}: {body}", file=sys.stderr)
-        sys.exit(1)
-    except urllib.error.URLError as e:
-        print(f"ERROR: 连不上平台 {base_url}: {e}", file=sys.stderr)
-        print(f"提示: 确保工作台在跑（start_pg.bat），或设 WLKJ_BASE_URL 环境变量", file=sys.stderr)
-        sys.exit(1)
-
-
-def write_config(project, base_url):
-    """把选中项目的 mcp_config 写入 mcp_config.json（保留 return_endpoint）。"""
-    # 读现有配置（保留注释和非 mcpServers 字段）
-    existing = {}
-    if os.path.isfile(CONFIG_PATH):
-        try:
-            with open(CONFIG_PATH, encoding="utf-8") as f:
-                existing = json.load(f)
-        except Exception:
-            pass
-
-    # 用平台的 mcp_config 覆盖
-    mcp_cfg = project.get("mcp_config", {})
-    if not mcp_cfg:
-        # 平台没返回完整 config，手动拼
-        mcp_cfg = {
-            "mcpServers": {"wlinkj-knowledge": {
-                "kind": "http",
-                "url": f"{base_url}/mcp",
-                "token": project.get("token", ""),
-                "env": {"WLKJ_PROJECT_ID": project.get("id", "")},
-            }},
-            "return_endpoint": {
-                "url": f"{base_url}/api/projects/{project.get('id','')}/returns",
-                "method": "POST",
-                "headers": {"X-Engine-Token": project.get("token", "")},
-            },
-        }
-
-    existing.update(mcp_cfg)
-    # 确保 return_endpoint 的端口正确
-    if "return_endpoint" in existing:
-        existing["return_endpoint"]["url"] = f"{base_url}/api/projects/{project['id']}/returns"
-
-    os.makedirs(os.path.dirname(CONFIG_PATH), exist_ok=True)
-    with open(CONFIG_PATH, "w", encoding="utf-8") as f:
-        json.dump(existing, f, ensure_ascii=False, indent=2)
-
-    print(f"OK: 已绑定项目「{project.get('name', project.get('id',''))}」")
-    print(f"    token: {project.get('token','')[:12]}...")
-    print(f"    project_id: {project.get('id','')}")
-    print(f"    配置写入: {CONFIG_PATH}")
-    print(f"    平台地址: {base_url}")
-
-
-def main():
-    import argparse
-    parser = argparse.ArgumentParser(description="切换 wlkj 项目绑定")
-    parser.add_argument("--email", help="平台邮箱")
-    parser.add_argument("--pw", "--password", dest="password", help="平台密码")
-    parser.add_argument("--project-uuid", help="直接指定项目 UUID（跳过选择）")
-    parser.add_argument("--base-url", default=BASE_URL, help=f"平台地址（默认 {BASE_URL}）")
-    parser.add_argument("--list", action="store_true", help="只列出项目，不写配置")
-    args = parser.parse_args()
-
-    # 1. 收集邮箱密码
-    email = args.email or os.environ.get("WLKJ_EMAIL") or input("平台邮箱: ").strip()
-    password = args.password or os.environ.get("WLKJ_PASSWORD") or input("平台密码: ").strip()
-
-    # 2. 拉项目列表
-    print(f"连接平台 {args.base_url} ...")
-    result = bind(email, password, args.base_url)
-    projects = result.get("projects", [])
-    if not projects:
-        print("该用户没有任何项目，请先在平台创建项目。")
-        sys.exit(1)
-
-    # 3. 列出项目
-    print(f"\n找到 {len(projects)} 个项目：")
-    for i, p in enumerate(projects):
-        print(f"  {i+1}. {p.get('name','?')} (id={p.get('id','?')[:8]})")
-
-    if args.list:
-        return
-
-    # 4. 选择项目
-    chosen = None
-    if args.project_uuid:
-        for p in projects:
-            if p.get("id") == args.project_uuid:
-                chosen = p
-                break
-        if not chosen:
-            print(f"ERROR: 项目 UUID '{args.project_uuid}' 不在列表中", file=sys.stderr)
-            sys.exit(1)
-    else:
-        try:
-            idx = int(input(f"\n选哪个项目？(1-{len(projects)}): ")) - 1
-            chosen = projects[idx]
-        except (ValueError, IndexError):
-            print("ERROR: 无效选择", file=sys.stderr)
-            sys.exit(1)
-
-    # 5. 写配置
-    write_config(chosen, args.base_url)
-
-
-if __name__ == "__main__":
-    main()
+#!/usr/bin/env python
+"""wlkj 项目切换脚本：选项目 → 拉平台 token → 写 mcp_config.json。
+
+用法：
+    python switch_project.py                          # 交互式：列出项目 → 选 → 写配置
+    python switch_project.py --email a@b.c --pw xxx  # 非交互：直接指定
+    python switch_project.py --project-uuid <UUID>   # 非交互：直接指定项目
+
+需工作台后端在跑（默认 http://127.0.0.1:10010）。
+"""
+import json
+import os
+import sys
+import urllib.request
+import urllib.error
+
+# 宿主根：脚本部署在 {host}/scripts/domain/kg/，4 级 dirname = {host}（.qoder）
+HOST_DIR = os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
+REPO_ROOT = os.path.dirname(HOST_DIR)  # qoderAll/
+
+
+def _find_config_path():
+    """按宿主探测 mcp_config.json：当前宿主根 → .qoder。"""
+    for p in (
+        os.path.join(HOST_DIR, "mcp_config.json"),        # 当前宿主根
+        os.path.join(REPO_ROOT, ".qoder", "mcp_config.json"),
+    ):
+        if os.path.isfile(p):
+            return p
+    return os.path.join(HOST_DIR, "mcp_config.json")  # 都没有则写当前宿主根
+
+
+CONFIG_PATH = _find_config_path()
+
+# 平台地址（默认 10010，可 env 覆盖）
+BASE_URL = os.environ.get("WLKJ_BASE_URL", "http://127.0.0.1:10010")
+
+
+def bind(email, password, base_url=BASE_URL):
+    """调平台 /api/auth/bind，返回 {projects:[{id,name,token,mcp_config}], base_url}。"""
+    data = json.dumps({"email": email, "password": password, "base_url": base_url}).encode()
+    req = urllib.request.Request(
+        f"{base_url}/api/auth/bind",
+        data=data, method="POST",
+        headers={"Content-Type": "application/json"},
+    )
+    try:
+        resp = urllib.request.urlopen(req, timeout=10)
+        return json.loads(resp.read())
+    except urllib.error.HTTPError as e:
+        body = e.read().decode("utf-8", errors="replace")[:200]
+        print(f"ERROR: 平台返回 {e.code}: {body}", file=sys.stderr)
+        sys.exit(1)
+    except urllib.error.URLError as e:
+        print(f"ERROR: 连不上平台 {base_url}: {e}", file=sys.stderr)
+        print(f"提示: 确保工作台在跑（start_pg.bat），或设 WLKJ_BASE_URL 环境变量", file=sys.stderr)
+        sys.exit(1)
+
+
+def write_config(project, base_url):
+    """把选中项目的 mcp_config 写入 mcp_config.json（保留 return_endpoint）。"""
+    # 读现有配置（保留注释和非 mcpServers 字段）
+    existing = {}
+    if os.path.isfile(CONFIG_PATH):
+        try:
+            with open(CONFIG_PATH, encoding="utf-8") as f:
+                existing = json.load(f)
+        except Exception:
+            pass
+
+    # 用平台的 mcp_config 覆盖
+    mcp_cfg = project.get("mcp_config", {})
+    if not mcp_cfg:
+        # 平台没返回完整 config，手动拼
+        mcp_cfg = {
+            "mcpServers": {"wlinkj-knowledge": {
+                "kind": "http",
+                "url": f"{base_url}/mcp",
+                "token": project.get("token", ""),
+                "env": {"WLKJ_PROJECT_ID": project.get("id", "")},
+            }},
+            "return_endpoint": {
+                "url": f"{base_url}/api/projects/{project.get('id','')}/returns",
+                "method": "POST",
+                "headers": {"X-Engine-Token": project.get("token", "")},
+            },
+        }
+
+    existing.update(mcp_cfg)
+    # 确保 return_endpoint 的端口正确
+    if "return_endpoint" in existing:
+        existing["return_endpoint"]["url"] = f"{base_url}/api/projects/{project['id']}/returns"
+
+    os.makedirs(os.path.dirname(CONFIG_PATH), exist_ok=True)
+    with open(CONFIG_PATH, "w", encoding="utf-8") as f:
+        json.dump(existing, f, ensure_ascii=False, indent=2)
+
+    print(f"OK: 已绑定项目「{project.get('name', project.get('id',''))}」")
+    print(f"    token: {project.get('token','')[:12]}...")
+    print(f"    project_id: {project.get('id','')}")
+    print(f"    配置写入: {CONFIG_PATH}")
+    print(f"    平台地址: {base_url}")
+
+
+def main():
+    import argparse
+    parser = argparse.ArgumentParser(description="切换 wlkj 项目绑定")
+    parser.add_argument("--email", help="平台邮箱")
+    parser.add_argument("--pw", "--password", dest="password", help="平台密码")
+    parser.add_argument("--project-uuid", help="直接指定项目 UUID（跳过选择）")
+    parser.add_argument("--base-url", default=BASE_URL, help=f"平台地址（默认 {BASE_URL}）")
+    parser.add_argument("--list", action="store_true", help="只列出项目，不写配置")
+    args = parser.parse_args()
+
+    # 1. 收集邮箱密码
+    email = args.email or os.environ.get("WLKJ_EMAIL") or input("平台邮箱: ").strip()
+    password = args.password or os.environ.get("WLKJ_PASSWORD") or input("平台密码: ").strip()
+
+    # 2. 拉项目列表
+    print(f"连接平台 {args.base_url} ...")
+    result = bind(email, password, args.base_url)
+    projects = result.get("projects", [])
+    if not projects:
+        print("该用户没有任何项目，请先在平台创建项目。")
+        sys.exit(1)
+
+    # 3. 列出项目
+    print(f"\n找到 {len(projects)} 个项目：")
+    for i, p in enumerate(projects):
+        print(f"  {i+1}. {p.get('name','?')} (id={p.get('id','?')[:8]})")
+
+    if args.list:
+        return
+
+    # 4. 选择项目
+    chosen = None
+    if args.project_uuid:
+        for p in projects:
+            if p.get("id") == args.project_uuid:
+                chosen = p
+                break
+        if not chosen:
+            print(f"ERROR: 项目 UUID '{args.project_uuid}' 不在列表中", file=sys.stderr)
+            sys.exit(1)
+    else:
+        try:
+            idx = int(input(f"\n选哪个项目？(1-{len(projects)}): ")) - 1
+            chosen = projects[idx]
+        except (ValueError, IndexError):
+            print("ERROR: 无效选择", file=sys.stderr)
+            sys.exit(1)
+
+    # 5. 写配置
+    write_config(chosen, args.base_url)
+
+
+if __name__ == "__main__":
+    main()