dma-cli 0.1.0__tar.gz → 0.1.2__tar.gz

{dma_cli-0.1.0 → dma_cli-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dma-cli
-Version: 0.1.0
+Version: 0.1.2
 Summary: DMA 任务提交命令行工具
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
@@ -9,11 +9,7 @@ Requires-Dist: pyyaml
 
 # DMA CLI —— 任务提交工具
 
-深度挖掘（DMA，`https://szc.test.dp.tech`）网页外的命令行提交工具，用于提交单个或批量任务。
-与压测无关，只借用压测脚本里已验证的接口形态，不改动任何压测内容。
-
-> **术语**：本工具的 `image`（镜像）即网页上的「软件目录」，底层接口为 `/softwares/deep-mining`，
-> snapshot 用 `softwareId`/`softwareVersionId`。容器镜像接口（image/public|private）不参与 job 提交，但 `image list` 合并展示供参考。
+深度挖掘算法（DMA）网页外的命令行提交工具，用于提交单个或批量任务。
 
 ## 安装
 
@@ -22,14 +18,6 @@ Requires-Dist: pyyaml
 pip install dma-cli
 dma --help
 
-# 从源码安装
-cd tools/dma-cli
-pip install .
-
-# 方式二：本地开发模式（可直接修改代码调试）
-cd tools/dma-cli
-pip install -e .
-
 # 安装后 dma 命令即可全局使用
 dma --help
 ```
@@ -64,12 +52,11 @@ cp config.example.yaml config.yaml
 base_url: https://szc.test.dp.tech
 auth:
   email: ""          # 验证码登录邮箱（留空则必须填 token）
-  captcha: "952754"  # 固定验证码（测试环境）
+  captcha: ""        # 留空，执行 dma login 时交互输入验证码
   token: ""          # 直接配 token（优先于 email 登录）
   uexid: ""          # 留空从 JWT 自动解析为 fe_<userId>
   user_id: ""        # 留空从 JWT 自动解析
 defaults:
-  biz_platform: 2
   name_prefix: "cli-job"
   max_run_time: 15
   max_run_time_unit: d
@@ -108,6 +95,11 @@ dma login
 dma login --email you@example.com
 ```
 
+`auth.captcha` 留空时，登录会提示 `请输入邮箱验证码:`，输入本次验证码后提交。
+验证码不会保存到配置文件或登录缓存。若配置中已填写 `auth.captcha`，则使用该值；
+已有配置中的旧验证码请清空，以启用交互输入。输入为空或取消输入时，不发送登录请求。
+此命令只校验输入的验证码，不会自动发送验证码邮件。
+
 解析 JWT 缓存 `token` / `user_id` / `uexid` 到 `~/.dma_cli/credentials.json`。也可跳过登录，直接在 `config.yaml` 填 `token`。
 
 ### whoami —— 查看当前状态
@@ -120,12 +112,19 @@ dma whoami
 
 ### project —— 项目管理
 
+先查询当前账号可访问的项目，根据项目名称找到自己的 `projectId`，再设为当前项目：
+
 ```bash
-dma project list        # 列 projectId | name
-dma project use 3948    # 设为当前项目
-dma project current     # 显示当前项目
+dma project list             # 列出 projectId | name
+dma project use <你的项目ID>   # 替换为上一步查到的 projectId
+dma project current          # 确认当前项目 ID 和名称
 ```
 
+`<你的项目ID>` 是占位符，执行前请替换为列表中你要使用的数字 ID，不要输入尖括号。
+当前项目会保存在本机，后续提交任务时省略 `--project_id` 或设置为 `0` 即使用该项目；
+JSON 配置中的 `"project_id": 0` 含义相同。`0` 表示使用当前项目，不是真实项目 ID，首次提交前需先完成项目选择。
+如需单次提交到其他项目，可传入 `--project_id <你的项目ID>`，不会改变已保存的当前项目。
+
 ### sku —— 机型查询
 
 ```bash
@@ -158,7 +157,7 @@ dma job submit -i job.example.json
 dma job submit \
   --machine_type c2_m2_cpu \
   --image_address ubuntu:22.04-py3.10-terminal-20260916 \
-  --project_id 3283 \
+  --project_id 0 \
   --command "python calc.py" \
   --job_name my-job
 ```
@@ -175,7 +174,6 @@ dma job submit \
 | `--machine_type` | `-t` | 机器类型名（自动匹配） |
 | `--command` | `-c` | 执行命令 |
 | `--max_run_time` | — | 最大运行时间（分钟） |
-| `--nnode` | — | 节点数 |
 | `--count` | — | 批量提交任务数 |
 | `--concurrency` | — | 批量并发数 |
 | `--batch_file` | — | 批量任务 JSON 数组文件 |
@@ -191,7 +189,8 @@ dma job submit \
 dma job submit \
   --machine_type c2_m2_cpu \
   --image_address ubuntu:22.04-py3.10-terminal-20260916 \
-  --count 10 \
+  --command "python calc.py" \
+  --count 5 \
   --concurrency 5
 ```
 
@@ -204,7 +203,7 @@ dma job submit \
 #   {"job_name": "job-002", "machine_type": "c4_m8_cpu", ...}
 # ]
 
-dma job submit --batch_file batch.json --concurrency 3
+dma job submit --batch_file batch.example.json --concurrency 3
 ```
 
 ### 方式三：--batch_input_dir（目录扫描）
@@ -212,9 +211,10 @@ dma job submit --batch_file batch.json --concurrency 3
 ```bash
 # 每个子目录作为一个任务的输入
 dma job submit \
-  --batch_input_dir ./inputs \
+  --batch_input_dir ./input \
   --machine_type c2_m2_cpu \
   --image_address ubuntu:22.04-py3.10-terminal-20260916 \
+  --command "python calc.py" \
   --concurrency 5
 ```
 
@@ -228,9 +228,10 @@ cp config.example.yaml config.yaml
 # 2. 查看状态
 dma whoami
 
-# 3. 切换项目
+# 3. 查询并选择自己的项目（将占位符替换为列表中的 projectId，不保留尖括号）
 dma project list
-dma project use 3948
+dma project use <你的项目ID>
+dma project current
 
 # 4. 查询机型和镜像
 dma sku list --kind cpu
@@ -240,30 +241,5 @@ dma image list
 dma job submit -i job.example.json
 
 # 6. 批量提交
-dma job submit --batch_file batch.json
-```
-
-## 文件结构
-
+dma job submit --batch_file batch.example.json
 ```
-tools/dma-cli/
-├── README.md              # 本文档
-├── pyproject.toml         # 打包配置
-├── requirements.txt       # requests, pyyaml
-├── config.example.yaml    # 配置模板
-├── job.example.json       # 任务配置示例
-├── batch.example.json     # 批量任务示例
-└── dma_cli/               # Python 包目录
-    ├── __init__.py
-    ├── cli.py             # 命令行入口
-    ├── client.py          # API 客户端
-    ├── state.py           # 状态管理
-    ├── resources.py       # 资源查询
-    ├── submit.py          # 任务提交
-    └── calc.py            # 默认任务脚本
-```
-
-## 不在范围
-
-- 不做多用户 / 不改压测脚本
-- 不做分布式压测

{dma_cli-0.1.0 → dma_cli-0.1.2}/README.md

@@ -1,10 +1,6 @@
 # DMA CLI —— 任务提交工具
 
-深度挖掘（DMA，`https://szc.test.dp.tech`）网页外的命令行提交工具，用于提交单个或批量任务。
-与压测无关，只借用压测脚本里已验证的接口形态，不改动任何压测内容。
-
-> **术语**：本工具的 `image`（镜像）即网页上的「软件目录」，底层接口为 `/softwares/deep-mining`，
-> snapshot 用 `softwareId`/`softwareVersionId`。容器镜像接口（image/public|private）不参与 job 提交，但 `image list` 合并展示供参考。
+深度挖掘算法（DMA）网页外的命令行提交工具，用于提交单个或批量任务。
 
 ## 安装
 
@@ -13,14 +9,6 @@
 pip install dma-cli
 dma --help
 
-# 从源码安装
-cd tools/dma-cli
-pip install .
-
-# 方式二：本地开发模式（可直接修改代码调试）
-cd tools/dma-cli
-pip install -e .
-
 # 安装后 dma 命令即可全局使用
 dma --help
 ```
@@ -55,12 +43,11 @@ cp config.example.yaml config.yaml
 base_url: https://szc.test.dp.tech
 auth:
   email: ""          # 验证码登录邮箱（留空则必须填 token）
-  captcha: "952754"  # 固定验证码（测试环境）
+  captcha: ""        # 留空，执行 dma login 时交互输入验证码
   token: ""          # 直接配 token（优先于 email 登录）
   uexid: ""          # 留空从 JWT 自动解析为 fe_<userId>
   user_id: ""        # 留空从 JWT 自动解析
 defaults:
-  biz_platform: 2
   name_prefix: "cli-job"
   max_run_time: 15
   max_run_time_unit: d
@@ -99,6 +86,11 @@ dma login
 dma login --email you@example.com
 ```
 
+`auth.captcha` 留空时，登录会提示 `请输入邮箱验证码:`，输入本次验证码后提交。
+验证码不会保存到配置文件或登录缓存。若配置中已填写 `auth.captcha`，则使用该值；
+已有配置中的旧验证码请清空，以启用交互输入。输入为空或取消输入时，不发送登录请求。
+此命令只校验输入的验证码，不会自动发送验证码邮件。
+
 解析 JWT 缓存 `token` / `user_id` / `uexid` 到 `~/.dma_cli/credentials.json`。也可跳过登录，直接在 `config.yaml` 填 `token`。
 
 ### whoami —— 查看当前状态
@@ -111,12 +103,19 @@ dma whoami
 
 ### project —— 项目管理
 
+先查询当前账号可访问的项目，根据项目名称找到自己的 `projectId`，再设为当前项目：
+
 ```bash
-dma project list        # 列 projectId | name
-dma project use 3948    # 设为当前项目
-dma project current     # 显示当前项目
+dma project list             # 列出 projectId | name
+dma project use <你的项目ID>   # 替换为上一步查到的 projectId
+dma project current          # 确认当前项目 ID 和名称
 ```
 
+`<你的项目ID>` 是占位符，执行前请替换为列表中你要使用的数字 ID，不要输入尖括号。
+当前项目会保存在本机，后续提交任务时省略 `--project_id` 或设置为 `0` 即使用该项目；
+JSON 配置中的 `"project_id": 0` 含义相同。`0` 表示使用当前项目，不是真实项目 ID，首次提交前需先完成项目选择。
+如需单次提交到其他项目，可传入 `--project_id <你的项目ID>`，不会改变已保存的当前项目。
+
 ### sku —— 机型查询
 
 ```bash
@@ -149,7 +148,7 @@ dma job submit -i job.example.json
 dma job submit \
   --machine_type c2_m2_cpu \
   --image_address ubuntu:22.04-py3.10-terminal-20260916 \
-  --project_id 3283 \
+  --project_id 0 \
   --command "python calc.py" \
   --job_name my-job
 ```
@@ -166,7 +165,6 @@ dma job submit \
 | `--machine_type` | `-t` | 机器类型名（自动匹配） |
 | `--command` | `-c` | 执行命令 |
 | `--max_run_time` | — | 最大运行时间（分钟） |
-| `--nnode` | — | 节点数 |
 | `--count` | — | 批量提交任务数 |
 | `--concurrency` | — | 批量并发数 |
 | `--batch_file` | — | 批量任务 JSON 数组文件 |
@@ -182,7 +180,8 @@ dma job submit \
 dma job submit \
   --machine_type c2_m2_cpu \
   --image_address ubuntu:22.04-py3.10-terminal-20260916 \
-  --count 10 \
+  --command "python calc.py" \
+  --count 5 \
   --concurrency 5
 ```
 
@@ -195,7 +194,7 @@ dma job submit \
 #   {"job_name": "job-002", "machine_type": "c4_m8_cpu", ...}
 # ]
 
-dma job submit --batch_file batch.json --concurrency 3
+dma job submit --batch_file batch.example.json --concurrency 3
 ```
 
 ### 方式三：--batch_input_dir（目录扫描）
@@ -203,9 +202,10 @@ dma job submit --batch_file batch.json --concurrency 3
 ```bash
 # 每个子目录作为一个任务的输入
 dma job submit \
-  --batch_input_dir ./inputs \
+  --batch_input_dir ./input \
   --machine_type c2_m2_cpu \
   --image_address ubuntu:22.04-py3.10-terminal-20260916 \
+  --command "python calc.py" \
   --concurrency 5
 ```
 
@@ -219,9 +219,10 @@ cp config.example.yaml config.yaml
 # 2. 查看状态
 dma whoami
 
-# 3. 切换项目
+# 3. 查询并选择自己的项目（将占位符替换为列表中的 projectId，不保留尖括号）
 dma project list
-dma project use 3948
+dma project use <你的项目ID>
+dma project current
 
 # 4. 查询机型和镜像
 dma sku list --kind cpu
@@ -231,30 +232,5 @@ dma image list
 dma job submit -i job.example.json
 
 # 6. 批量提交
-dma job submit --batch_file batch.json
-```
-
-## 文件结构
-
+dma job submit --batch_file batch.example.json
 ```
-tools/dma-cli/
-├── README.md              # 本文档
-├── pyproject.toml         # 打包配置
-├── requirements.txt       # requests, pyyaml
-├── config.example.yaml    # 配置模板
-├── job.example.json       # 任务配置示例
-├── batch.example.json     # 批量任务示例
-└── dma_cli/               # Python 包目录
-    ├── __init__.py
-    ├── cli.py             # 命令行入口
-    ├── client.py          # API 客户端
-    ├── state.py           # 状态管理
-    ├── resources.py       # 资源查询
-    ├── submit.py          # 任务提交
-    └── calc.py            # 默认任务脚本
-```
-
-## 不在范围
-
-- 不做多用户 / 不改压测脚本
-- 不做分布式压测

{dma_cli-0.1.0 → dma_cli-0.1.2}/config.example.yaml

@@ -4,13 +4,12 @@ base_url: https://szc.test.dp.tech
 
 auth:
   email: ""          # 验证码登录邮箱（留空则必须填 token）
-  captcha: "952754"  # 固定验证码（测试环境）
+  captcha: ""        # 留空，执行 dma login 时交互输入验证码
   token: ""          # 直接配 token（优先于 email 登录，留空走 email）
   uexid: ""          # 留空则从 JWT 自动解析为 fe_<userId>
   user_id: ""        # 留空则从 JWT 自动解析
 
 defaults:
-  biz_platform: 2
   name_prefix: "cli-job"
   cmd: "python calc.py"
   max_run_time: 15

{dma_cli-0.1.0 → dma_cli-0.1.2}/dma_cli/__init__.py

@@ -1,3 +1,3 @@
 """DMA CLI —— 任务提交命令行工具"""
 
-__version__ = "0.1.0"
+__version__ = "0.1.2"

{dma_cli-0.1.0 → dma_cli-0.1.2}/dma_cli/client.py

@@ -24,11 +24,12 @@ _PKG_DIR = Path(__file__).parent
 CONFIG_FILE = Path.cwd() / "config.yaml" if (Path.cwd() / "config.yaml").exists() else _PKG_DIR / "config.yaml"
 EXAMPLE_CONFIG = _PKG_DIR / "config.example.yaml"
 
+BIZ_PLATFORM = 2
+
 DEFAULT_CONFIG = {
     "base_url": "https://szc.test.dp.tech",
-    "auth": {"email": "", "captcha": "952754", "token": "", "uexid": "", "user_id": ""},
+    "auth": {"email": "", "captcha": "", "token": "", "uexid": "", "user_id": ""},
     "defaults": {
-        "biz_platform": 2,
         "name_prefix": "cli-job",
         "cmd": "python calc.py",
         "max_run_time": 15,
@@ -153,7 +154,14 @@ class DmaClient:
         email = email or self.config["auth"].get("email")
         if not email:
             raise RuntimeError("未配置 auth.email，无法验证码登录（请在 config.yaml 填写或改用 token）")
-        captcha = self.config["auth"].get("captcha") or "952754"
+        captcha = str(self.config["auth"].get("captcha") or "").strip()
+        if not captcha:
+            try:
+                captcha = input("请输入邮箱验证码: ").strip()
+            except (EOFError, KeyboardInterrupt):
+                raise RuntimeError("已取消验证码输入，未发送登录请求") from None
+        if not captcha:
+            raise RuntimeError("验证码不能为空，请重新执行 dma login")
         url = self.base_url + LOGIN_PATH
         payload = {
             "channel": "gkx-talent-universe",

{dma_cli-0.1.0 → dma_cli-0.1.2}/dma_cli/resources.py

@@ -12,14 +12,10 @@ import sys
 
 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
 
-from .client import STATUS_MAP
+from .client import BIZ_PLATFORM, 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:
@@ -38,7 +34,7 @@ def _print_table(headers, rows) -> None:
 
 # ============ project ============
 def project_list(client) -> list:
-    params = {"queryType": "all", "pageSize": 1000, "bizPlatform": _biz_platform(client), "page": 1}
+    params = {"queryType": "all", "pageSize": 1000, "bizPlatform": BIZ_PLATFORM, "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 []
@@ -123,7 +119,7 @@ def handle_sku_use(client, args) -> None:
 
 # ============ image (软件目录) ============
 def software_list(client) -> list:
-    resp = client.get("/dmaapi/szc/v1/softwares/deep-mining", params={"bizPlatform": _biz_platform(client)})
+    resp = client.get("/dmaapi/szc/v1/softwares/deep-mining", params={"bizPlatform": BIZ_PLATFORM})
     return ((resp or {}).get("data") or {}).get("items") or []
 
 
@@ -155,7 +151,7 @@ def handle_image_list(client, args) -> None:
     all_images = []
 
     # 1. 查询软件目录（/softwares/deep-mining）
-    resp = client.get("/dmaapi/szc/v1/softwares/deep-mining", params={"bizPlatform": 2})
+    resp = client.get("/dmaapi/szc/v1/softwares/deep-mining", params={"bizPlatform": BIZ_PLATFORM})
     if resp.get("code") == 0:
         for item in resp.get("data", {}).get("items", []):
             for version in item.get("versions", []):
@@ -170,7 +166,7 @@ def handle_image_list(client, args) -> None:
 
     # 2. 查询公共镜像（/image/public）
     resp = client.get("/dmaapi/szc/v1/image/public/deep-mining",
-                     params={"page": 1, "pageSize": 1000, "bizPlatform": 2})
+                     params={"page": 1, "pageSize": 1000, "bizPlatform": BIZ_PLATFORM})
     if resp.get("code") == 0:
         for item in resp.get("data", {}).get("items", []):
             # url 字段可能是 img
@@ -184,7 +180,7 @@ def handle_image_list(client, args) -> None:
 
     # 3. 查询私有镜像（/image/private）
     resp = client.get("/dmaapi/szc/v1/image/private",
-                     params={"type": "image", "device": "container", "page": 1, "pageSize": 50, "bizPlatform": 2})
+                     params={"type": "image", "device": "container", "page": 1, "pageSize": 50, "bizPlatform": BIZ_PLATFORM})
     if resp.get("code") == 0:
         for item in resp.get("data", {}).get("items", []):
             url = item.get("url") or ""

{dma_cli-0.1.0 → dma_cli-0.1.2}/dma_cli/submit.py

@@ -24,7 +24,7 @@ import requests
 
 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
 
-from .client import DmaClient, STATUS_MAP
+from .client import BIZ_PLATFORM, DmaClient, STATUS_MAP
 
 # 内置默认任务脚本（无 --input_directory 时上传它）
 DEFAULT_INPUT = Path(__file__).parent / "calc.py"
@@ -57,7 +57,7 @@ def find_software_by_image_address(client: DmaClient, image_address: str) -> dic
     2. softwareVersion 字段精确匹配
     3. softwareName 模糊匹配
     """
-    resp = client.get("/dmaapi/szc/v1/softwares/deep-mining", params={"bizPlatform": 2})
+    resp = client.get("/dmaapi/szc/v1/softwares/deep-mining", params={"bizPlatform": BIZ_PLATFORM})
     if resp.get("code") != 0:
         raise RuntimeError(f"查询软件目录失败: code={resp.get('code')}")
 
@@ -407,7 +407,7 @@ def handle_submit(client, args):
             json_config = json_config["_batch_data"][0] if json_config["_batch_data"] else {}
 
         # 映射 JSON 字段到 overrides
-        if json_config.get("project_id"):
+        if json_config.get("project_id") is not None and json_config.get("project_id") != 0:
             overrides["project_id"] = json_config["project_id"]
         if json_config.get("job_name"):
             args.job_name = json_config["job_name"]
@@ -435,7 +435,7 @@ def handle_submit(client, args):
             overrides["dataset_path"] = json_config["dataset_path"]
 
     # CLI 参数覆盖 JSON 配置
-    if args.project_id:
+    if args.project_id is not None and args.project_id != 0:
         overrides["project_id"] = args.project_id
     if args.machine_type:
         overrides["machine_type"] = args.machine_type
@@ -542,8 +542,12 @@ def handle_submit(client, args):
             for key in ["project_id", "job_name", "machine_type", "image_address", "command",
                        "input_directory", "max_run_time", "max_reschedule_times", "nnode",
                        "log_file", "out_files", "result_path", "dataset_path"]:
-                if config.get(key):
-                    item_overrides[key] = config[key]
+                val = config.get(key)
+                if val is not None:
+                    # 如果 project_id 显式为 0，则应使用默认当前项目
+                    if key == "project_id" and val == 0:
+                        continue
+                    item_overrides[key] = val
 
             # 自动查找
             if item_overrides.get("machine_type"):
@@ -670,4 +674,4 @@ def handle_submit(client, args):
                 print("📊 轮询任务状态...")
                 poll_task_status(client, result['job_id'])
         else:
-            print(f"❌ 提交失败: {result['error']}")
+            print(f"❌ 提交失败: {result['error']}")

{dma_cli-0.1.0 → dma_cli-0.1.2}/dma_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dma-cli
-Version: 0.1.0
+Version: 0.1.2
 Summary: DMA 任务提交命令行工具
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
@@ -9,11 +9,7 @@ Requires-Dist: pyyaml
 
 # DMA CLI —— 任务提交工具
 
-深度挖掘（DMA，`https://szc.test.dp.tech`）网页外的命令行提交工具，用于提交单个或批量任务。
-与压测无关，只借用压测脚本里已验证的接口形态，不改动任何压测内容。
-
-> **术语**：本工具的 `image`（镜像）即网页上的「软件目录」，底层接口为 `/softwares/deep-mining`，
-> snapshot 用 `softwareId`/`softwareVersionId`。容器镜像接口（image/public|private）不参与 job 提交，但 `image list` 合并展示供参考。
+深度挖掘算法（DMA）网页外的命令行提交工具，用于提交单个或批量任务。
 
 ## 安装
 
@@ -22,14 +18,6 @@ Requires-Dist: pyyaml
 pip install dma-cli
 dma --help
 
-# 从源码安装
-cd tools/dma-cli
-pip install .
-
-# 方式二：本地开发模式（可直接修改代码调试）
-cd tools/dma-cli
-pip install -e .
-
 # 安装后 dma 命令即可全局使用
 dma --help
 ```
@@ -64,12 +52,11 @@ cp config.example.yaml config.yaml
 base_url: https://szc.test.dp.tech
 auth:
   email: ""          # 验证码登录邮箱（留空则必须填 token）
-  captcha: "952754"  # 固定验证码（测试环境）
+  captcha: ""        # 留空，执行 dma login 时交互输入验证码
   token: ""          # 直接配 token（优先于 email 登录）
   uexid: ""          # 留空从 JWT 自动解析为 fe_<userId>
   user_id: ""        # 留空从 JWT 自动解析
 defaults:
-  biz_platform: 2
   name_prefix: "cli-job"
   max_run_time: 15
   max_run_time_unit: d
@@ -108,6 +95,11 @@ dma login
 dma login --email you@example.com
 ```
 
+`auth.captcha` 留空时，登录会提示 `请输入邮箱验证码:`，输入本次验证码后提交。
+验证码不会保存到配置文件或登录缓存。若配置中已填写 `auth.captcha`，则使用该值；
+已有配置中的旧验证码请清空，以启用交互输入。输入为空或取消输入时，不发送登录请求。
+此命令只校验输入的验证码，不会自动发送验证码邮件。
+
 解析 JWT 缓存 `token` / `user_id` / `uexid` 到 `~/.dma_cli/credentials.json`。也可跳过登录，直接在 `config.yaml` 填 `token`。
 
 ### whoami —— 查看当前状态
@@ -120,12 +112,19 @@ dma whoami
 
 ### project —— 项目管理
 
+先查询当前账号可访问的项目，根据项目名称找到自己的 `projectId`，再设为当前项目：
+
 ```bash
-dma project list        # 列 projectId | name
-dma project use 3948    # 设为当前项目
-dma project current     # 显示当前项目
+dma project list             # 列出 projectId | name
+dma project use <你的项目ID>   # 替换为上一步查到的 projectId
+dma project current          # 确认当前项目 ID 和名称
 ```
 
+`<你的项目ID>` 是占位符，执行前请替换为列表中你要使用的数字 ID，不要输入尖括号。
+当前项目会保存在本机，后续提交任务时省略 `--project_id` 或设置为 `0` 即使用该项目；
+JSON 配置中的 `"project_id": 0` 含义相同。`0` 表示使用当前项目，不是真实项目 ID，首次提交前需先完成项目选择。
+如需单次提交到其他项目，可传入 `--project_id <你的项目ID>`，不会改变已保存的当前项目。
+
 ### sku —— 机型查询
 
 ```bash
@@ -158,7 +157,7 @@ dma job submit -i job.example.json
 dma job submit \
   --machine_type c2_m2_cpu \
   --image_address ubuntu:22.04-py3.10-terminal-20260916 \
-  --project_id 3283 \
+  --project_id 0 \
   --command "python calc.py" \
   --job_name my-job
 ```
@@ -175,7 +174,6 @@ dma job submit \
 | `--machine_type` | `-t` | 机器类型名（自动匹配） |
 | `--command` | `-c` | 执行命令 |
 | `--max_run_time` | — | 最大运行时间（分钟） |
-| `--nnode` | — | 节点数 |
 | `--count` | — | 批量提交任务数 |
 | `--concurrency` | — | 批量并发数 |
 | `--batch_file` | — | 批量任务 JSON 数组文件 |
@@ -191,7 +189,8 @@ dma job submit \
 dma job submit \
   --machine_type c2_m2_cpu \
   --image_address ubuntu:22.04-py3.10-terminal-20260916 \
-  --count 10 \
+  --command "python calc.py" \
+  --count 5 \
   --concurrency 5
 ```
 
@@ -204,7 +203,7 @@ dma job submit \
 #   {"job_name": "job-002", "machine_type": "c4_m8_cpu", ...}
 # ]
 
-dma job submit --batch_file batch.json --concurrency 3
+dma job submit --batch_file batch.example.json --concurrency 3
 ```
 
 ### 方式三：--batch_input_dir（目录扫描）
@@ -212,9 +211,10 @@ dma job submit --batch_file batch.json --concurrency 3
 ```bash
 # 每个子目录作为一个任务的输入
 dma job submit \
-  --batch_input_dir ./inputs \
+  --batch_input_dir ./input \
   --machine_type c2_m2_cpu \
   --image_address ubuntu:22.04-py3.10-terminal-20260916 \
+  --command "python calc.py" \
   --concurrency 5
 ```
 
@@ -228,9 +228,10 @@ cp config.example.yaml config.yaml
 # 2. 查看状态
 dma whoami
 
-# 3. 切换项目
+# 3. 查询并选择自己的项目（将占位符替换为列表中的 projectId，不保留尖括号）
 dma project list
-dma project use 3948
+dma project use <你的项目ID>
+dma project current
 
 # 4. 查询机型和镜像
 dma sku list --kind cpu
@@ -240,30 +241,5 @@ dma image list
 dma job submit -i job.example.json
 
 # 6. 批量提交
-dma job submit --batch_file batch.json
-```
-
-## 文件结构
-
+dma job submit --batch_file batch.example.json
 ```
-tools/dma-cli/
-├── README.md              # 本文档
-├── pyproject.toml         # 打包配置
-├── requirements.txt       # requests, pyyaml
-├── config.example.yaml    # 配置模板
-├── job.example.json       # 任务配置示例
-├── batch.example.json     # 批量任务示例
-└── dma_cli/               # Python 包目录
-    ├── __init__.py
-    ├── cli.py             # 命令行入口
-    ├── client.py          # API 客户端
-    ├── state.py           # 状态管理
-    ├── resources.py       # 资源查询
-    ├── submit.py          # 任务提交
-    └── calc.py            # 默认任务脚本
-```
-
-## 不在范围
-
-- 不做多用户 / 不改压测脚本
-- 不做分布式压测

{dma_cli-0.1.0 → dma_cli-0.1.2}/dma_cli.egg-info/SOURCES.txt

@@ -19,4 +19,6 @@ dma_cli.egg-info/requires.txt
 dma_cli.egg-info/top_level.txt
 input/calc.py
 input/case1/calc.py
-input/case2/calc.py
+input/case2/calc.py
+tests/test_biz_platform.py
+tests/test_login.py

{dma_cli-0.1.0 → dma_cli-0.1.2}/job.example.json

@@ -1,5 +1,5 @@
 {
-  "project_id": 3948,
+  "project_id": 0,
   "job_name": "cli-job-001",
   "image_address": "ubuntu:22.04-py3.10-terminal-20260916",
   "machine_type": "c2_m2_cpu",

{dma_cli-0.1.0 → dma_cli-0.1.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "dma-cli"
-version = "0.1.0"
+version = "0.1.2"
 description = "DMA 任务提交命令行工具"
 readme = "README.md"
 requires-python = ">=3.8"

dma_cli-0.1.2/tests/test_biz_platform.py

@@ -0,0 +1,27 @@
+import unittest
+from unittest.mock import Mock
+
+from dma_cli.resources import project_list, software_list
+from dma_cli.submit import find_software_by_image_address
+
+
+class BizPlatformTests(unittest.TestCase):
+    def test_requests_ignore_legacy_platform_config(self):
+        for operation, args in (
+            (project_list, ()),
+            (software_list, ()),
+            (find_software_by_image_address, ("test-image",)),
+        ):
+            with self.subTest(operation=operation.__name__):
+                client = Mock()
+                client.defaults = {"biz_platform": 99}
+                client.get.return_value = {"code": 0, "data": {"items": []}}
+                try:
+                    operation(client, *args)
+                except RuntimeError as error:
+                    self.assertIn("未找到镜像", str(error))
+                self.assertEqual(client.get.call_args.kwargs["params"]["bizPlatform"], 2)
+
+
+if __name__ == "__main__":
+    unittest.main()

dma_cli-0.1.2/tests/test_login.py

@@ -0,0 +1,55 @@
+import unittest
+from unittest.mock import Mock, patch
+
+from dma_cli.client import DmaClient
+
+
+class LoginCaptchaTests(unittest.TestCase):
+    def setUp(self):
+        save = patch("dma_cli.client.save_creds")
+        save.start()
+        self.addCleanup(save.stop)
+        self.client = DmaClient.__new__(DmaClient)
+        self.client.config = {
+            "base_url": "https://example.invalid",
+            "auth": {"email": "user@example.invalid", "captcha": ""},
+        }
+        self.client._http = Mock()
+        self.client._http.post.return_value.json.return_value = {
+            "code": 0, "data": {"token": "test-token"}
+        }
+
+    def test_prompts_for_missing_captcha_and_sends_input(self):
+        with patch("builtins.input", return_value=" 123456 ") as prompt, \
+                patch("dma_cli.client.save_creds"):
+            self.client.login()
+        prompt.assert_called_once()
+        payload = self.client._http.post.call_args.kwargs["json"]
+        self.assertEqual(payload["emailCode"], "123456")
+        self.assertEqual(payload["code"], "123456")
+        self.assertEqual(self.client.config["auth"]["captcha"], "")
+        self.assertNotIn("captcha", self.client.creds)
+
+    def test_configured_captcha_does_not_prompt(self):
+        self.client.config["auth"]["captcha"] = "654321"
+        with patch("builtins.input") as prompt, patch("dma_cli.client.save_creds"):
+            self.client.login()
+        prompt.assert_not_called()
+        self.assertEqual(self.client._http.post.call_args.kwargs["json"]["code"], "654321")
+
+    def test_empty_input_does_not_send_login_request(self):
+        with patch("builtins.input", return_value="  "):
+            with self.assertRaisesRegex(RuntimeError, "验证码不能为空"):
+                self.client.login()
+        self.client._http.post.assert_not_called()
+
+    def test_cancelled_input_does_not_send_login_request(self):
+        for error in (EOFError, KeyboardInterrupt):
+            with self.subTest(error=error), patch("builtins.input", side_effect=error):
+                with self.assertRaisesRegex(RuntimeError, "取消验证码输入"):
+                    self.client.login()
+        self.client._http.post.assert_not_called()
+
+
+if __name__ == "__main__":
+    unittest.main()