dma-cli 0.1.0__py3-none-any.whl

dma_cli/resources.py

@@ -0,0 +1,251 @@
+"""查询 + 切换：project / sku / image(软件) / jobs / job rm。
+
+底层接口（HAR/jmx 实测，仅引用接口形态）：
+- 项目: GET /dmaapi/szc/v2/project/list?queryType=all&pageSize=1000&bizPlatform=2&page=1
+- 机型: GET /dmaapi/szc/v1/softwares/skus?kind=cpu|gpu&create_source=2
+- 软件: GET /dmaapi/szc/v1/softwares/deep-mining?bizPlatform=2   （image==软件目录）
+- 任务: GET /irpapi/szc/v1/job/list?page=&queryType=all&version=v2&pageSize=&status=&createSource=2[&projectId=]
+- 删除: POST /irpapi/szc/v1/job/del/<bohrId>
+"""
+import os
+import sys
+
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+
+from .client import STATUS_MAP
+from .state import set_current
+
+
+def _biz_platform(client) -> int:
+    return int(client.defaults.get("biz_platform", 2))
+
+
+def _print_table(headers, rows) -> None:
+    """简易对齐表格打印。"""
+    if not rows:
+        print("  （无数据）")
+        return
+    widths = [len(h) for h in headers]
+    for row in rows:
+        for i, cell in enumerate(row):
+            widths[i] = max(widths[i], len(str(cell)))
+    fmt = "  ".join(f"{{:<{w}}}" for w in widths)
+    print(fmt.format(*headers))
+    print(fmt.format(*["-" * w for w in widths]))
+    for row in rows:
+        print(fmt.format(*[str(c) for c in row]))
+
+
+# ============ project ============
+def project_list(client) -> list:
+    params = {"queryType": "all", "pageSize": 1000, "bizPlatform": _biz_platform(client), "page": 1}
+    resp = client.get("/dmaapi/szc/v2/project/list", params=params)
+    data = (resp or {}).get("data") or {}
+    return data.get("items") or data.get("list") or []
+
+
+def project_use(client, project_id) -> dict:
+    for it in project_list(client):
+        pid = it.get("id") or it.get("projectId")
+        if str(pid) == str(project_id):
+            value = {"projectId": int(pid), "name": it.get("name", "")}
+            set_current("project", value)
+            return value
+    raise RuntimeError(f"未找到 projectId={project_id}")
+
+
+def handle_project_list(client, args) -> None:
+    items = project_list(client)
+    rows = [[it.get("id") or it.get("projectId"), it.get("name", "")] for it in items]
+    print(f"共 {len(rows)} 个项目")
+    _print_table(["projectId", "name"], rows)
+
+
+def handle_project_use(client, args) -> None:
+    value = project_use(client, args.project_id)
+    print(f"✅ 当前项目: {value['projectId']} | {value['name']}")
+
+
+def handle_project_current(client, args) -> None:
+    p = client.current_project()
+    if p:
+        print(f"当前项目: {p['projectId']} | {p['name']}")
+    else:
+        print("未设置当前项目（用 `project use <projectId>` 设置）")
+
+
+# ============ sku ============
+def sku_list(client, kind="cpu") -> list:
+    resp = client.get("/dmaapi/szc/v1/softwares/skus", params={"kind": kind, "create_source": 2})
+    return ((resp or {}).get("data") or {}).get("items") or []
+
+
+def sku_use(client, sku_id) -> dict:
+    found = None
+    for kind in ("cpu", "gpu"):
+        for it in sku_list(client, kind):
+            if str(it.get("skuId")) == str(sku_id):
+                found = it
+                break
+        if found:
+            break
+    if not found:
+        raise RuntimeError(f"未找到 skuId={sku_id}")
+    value = {
+        "skuId": int(found["skuId"]),
+        "skuName": found.get("skuName", ""),
+        "cpu": found.get("cpu"),
+        "memory": found.get("memory"),
+        "gpuModel": found.get("gpuModel", ""),
+        "gpuNum": found.get("gpuNum", 0),
+    }
+    set_current("sku", value)
+    return value
+
+
+def handle_sku_list(client, args) -> None:
+    kind = args.kind or "cpu"
+    items = sku_list(client, kind)
+    rows = [
+        [it.get("skuId"), it.get("skuName", ""), it.get("cpu"), it.get("memory"),
+         it.get("gpuModel", ""), it.get("gpuNum", 0), "是" if it.get("hasStock") else "否"]
+        for it in items
+    ]
+    print(f"共 {len(rows)} 个机型 (kind={kind})")
+    _print_table(["skuId", "skuName", "cpu", "memory", "gpuModel", "gpuNum", "hasStock"], rows)
+
+
+def handle_sku_use(client, args) -> None:
+    value = sku_use(client, args.sku_id)
+    print(f"✅ 当前机型: {value['skuId']} | {value['skuName']} "
+          f"(cpu={value['cpu']} memory={value['memory']} gpu={value['gpuModel']}x{value['gpuNum']})")
+
+
+# ============ image (软件目录) ============
+def software_list(client) -> list:
+    resp = client.get("/dmaapi/szc/v1/softwares/deep-mining", params={"bizPlatform": _biz_platform(client)})
+    return ((resp or {}).get("data") or {}).get("items") or []
+
+
+def software_use(client, software_id, version_id=None) -> dict:
+    items = software_list(client)
+    sw = next((it for it in items if str(it.get("softwareId")) == str(software_id)), None)
+    if not sw:
+        raise RuntimeError(f"未找到 softwareId={software_id}")
+    versions = sw.get("versions") or []
+    if not versions:
+        raise RuntimeError(f"软件 {software_id} 无可用版本")
+    ver = None
+    if version_id is not None:
+        ver = next((v for v in versions if str(v.get("softwareVersionId")) == str(version_id)), None)
+    if ver is None:
+        ver = versions[0]  # 不给 version 取第一个
+    value = {
+        "softwareId": int(sw["softwareId"]),
+        "softwareName": sw.get("softwareName", ""),
+        "softwareVersionId": int(ver["softwareVersionId"]),
+        "softwareVersion": ver.get("softwareVersion", ""),
+    }
+    set_current("software", value)
+    return value
+
+
+def handle_image_list(client, args) -> None:
+    """列出镜像（公共+私有合并，含 url 列）。"""
+    all_images = []
+
+    # 1. 查询软件目录（/softwares/deep-mining）
+    resp = client.get("/dmaapi/szc/v1/softwares/deep-mining", params={"bizPlatform": 2})
+    if resp.get("code") == 0:
+        for item in resp.get("data", {}).get("items", []):
+            for version in item.get("versions", []):
+                url = version.get("url") or f"{item.get('softwareName')}:{version.get('softwareVersion')}"
+                all_images.append({
+                    "id": item["softwareId"],
+                    "name": item.get("softwareName", ""),
+                    "type": "软件目录",
+                    "url": url,
+                    "version_id": version.get("softwareVersionId"),
+                })
+
+    # 2. 查询公共镜像（/image/public）
+    resp = client.get("/dmaapi/szc/v1/image/public/deep-mining",
+                     params={"page": 1, "pageSize": 1000, "bizPlatform": 2})
+    if resp.get("code") == 0:
+        for item in resp.get("data", {}).get("items", []):
+            # url 字段可能是 img
+            url = item.get("img") or item.get("url") or ""
+            all_images.append({
+                "id": item.get("id"),
+                "name": item.get("name", ""),
+                "type": "公共镜像",
+                "url": url,
+            })
+
+    # 3. 查询私有镜像（/image/private）
+    resp = client.get("/dmaapi/szc/v1/image/private",
+                     params={"type": "image", "device": "container", "page": 1, "pageSize": 50, "bizPlatform": 2})
+    if resp.get("code") == 0:
+        for item in resp.get("data", {}).get("items", []):
+            url = item.get("url") or ""
+            all_images.append({
+                "id": item.get("id"),
+                "name": item.get("name", ""),
+                "type": "私有镜像",
+                "url": url,
+            })
+
+    # 合并输出
+    print(f"共 {len(all_images)} 个镜像（软件目录 + 公共 + 私有）")
+    rows = [[img["id"], img["name"], img["type"], img["url"]] for img in all_images]
+    _print_table(["id", "name", "type", "url"], rows)
+
+
+def handle_image_use(client, args) -> None:
+    value = software_use(client, args.software_id, args.version)
+    print(f"✅ 当前镜像(软件): {value['softwareId']} | {value['softwareName']} "
+          f"-> 版本 {value['softwareVersionId']} | {value['softwareVersion']}")
+
+
+# ============ jobs ============
+def jobs_list(client, status=-1, project_id=None, page_size=100) -> tuple:
+    all_items, page, total = [], 1, 0
+    while True:
+        params = {"page": page, "queryType": "all", "version": "v2",
+                  "pageSize": page_size, "status": status, "createSource": 2}
+        if project_id:
+            params["projectId"] = project_id
+        resp = client.get("/irpapi/szc/v1/job/list", params=params)
+        inner = ((resp or {}).get("data") or {}) if isinstance(resp, dict) else {}
+        items = inner.get("items") or []
+        total = inner.get("total", 0)
+        all_items.extend(items)
+        if len(all_items) >= total or len(items) < page_size:
+            break
+        page += 1
+    return all_items, total
+
+
+def job_delete(client, bohr_id) -> dict:
+    return client.post_empty(f"/irpapi/szc/v1/job/del/{bohr_id}")
+
+
+def handle_jobs(client, args) -> None:
+    items, total = jobs_list(client, status=args.status, project_id=args.project_id)
+    print(f"共 {total} 个任务（拉取 {len(items)} 个，显示前 50）")
+    rows = []
+    for j in items[:50]:
+        jid = j.get("bohrId") or j.get("id")
+        s = j.get("status", "?")
+        rows.append([jid, STATUS_MAP.get(s, f"?({s})"), j.get("jobName", ""), j.get("createTime", "")])
+    _print_table(["bohrId", "status", "jobName", "createTime"], rows)
+    if len(items) > 50:
+        print(f"... 还有 {len(items) - 50} 个")
+
+
+def handle_job_rm(client, args) -> None:
+    resp = job_delete(client, args.bohr_id)
+    if resp.get("code") == 0:
+        print(f"✅ 已删除任务 {args.bohr_id}")
+    else:
+        print(f"❌ 删除失败: code={resp.get('code')} msg={resp.get('msg')}")

dma_cli/state.py

@@ -0,0 +1,65 @@
+"""本地状态读写：~/.dma_cli/
+
+- session.json    : 当前选择（project / sku / software），由 `use` 命令写入，`submit` 读取
+- credentials.json: 登录后的 token / user_id / uexid 缓存（不入库）
+"""
+import json
+import os
+from pathlib import Path
+
+_SESSION_DIR = Path(os.path.expanduser("~")) / ".dma_cli"
+SESSION_FILE = _SESSION_DIR / "session.json"
+CRED_FILE = _SESSION_DIR / "credentials.json"
+
+# session.json 的合法键
+_SESSION_KEYS = ("project", "sku", "software")
+
+
+def _ensure_dir() -> None:
+    _SESSION_DIR.mkdir(parents=True, exist_ok=True)
+
+
+def load_session() -> dict:
+    """读取当前选择，缺键补 None。"""
+    try:
+        with open(SESSION_FILE, "r", encoding="utf-8") as f:
+            data = json.load(f)
+    except (FileNotFoundError, json.JSONDecodeError):
+        data = {}
+    for k in _SESSION_KEYS:
+        data.setdefault(k, None)
+    return data
+
+
+def save_session(session: dict) -> None:
+    _ensure_dir()
+    with open(SESSION_FILE, "w", encoding="utf-8") as f:
+        json.dump(session, f, ensure_ascii=False, indent=2)
+
+
+def set_current(kind: str, value: dict) -> dict:
+    """设置某一项当前选择并落盘，返回更新后的完整 session。"""
+    if kind not in _SESSION_KEYS:
+        raise ValueError(f"未知的选择类型: {kind}（合法: {_SESSION_KEYS}）")
+    session = load_session()
+    session[kind] = value
+    save_session(session)
+    return session
+
+
+def load_creds() -> dict:
+    try:
+        with open(CRED_FILE, "r", encoding="utf-8") as f:
+            return json.load(f)
+    except (FileNotFoundError, json.JSONDecodeError):
+        return {}
+
+
+def save_creds(creds: dict) -> None:
+    _ensure_dir()
+    with open(CRED_FILE, "w", encoding="utf-8") as f:
+        json.dump(creds, f, ensure_ascii=False, indent=2)
+
+
+def creds_path() -> Path:
+    return CRED_FILE