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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dma-cli
-Version: 0.1.1
+Version: 0.1.2
 Summary: DMA 任务提交命令行工具
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
@@ -52,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
@@ -96,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 —— 查看当前状态
@@ -108,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
@@ -146,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
 ```
@@ -192,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（目录扫描）
@@ -200,7 +211,7 @@ 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" \
@@ -217,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
@@ -229,5 +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
 ```

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

@@ -43,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
@@ -87,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 —— 查看当前状态
@@ -99,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
@@ -137,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
 ```
@@ -183,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（目录扫描）
@@ -191,7 +202,7 @@ 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" \
@@ -208,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
@@ -220,5 +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
 ```

{dma_cli-0.1.1 → 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.1 → dma_cli-0.1.2}/dma_cli/__init__.py

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

{dma_cli-0.1.1 → 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.1 → 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.1 → 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')}")
 
@@ -674,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.1 → 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.1
+Version: 0.1.2
 Summary: DMA 任务提交命令行工具
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
@@ -52,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
@@ -96,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 —— 查看当前状态
@@ -108,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
@@ -146,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
 ```
@@ -192,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（目录扫描）
@@ -200,7 +211,7 @@ 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" \
@@ -217,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
@@ -229,5 +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
 ```

{dma_cli-0.1.1 → 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.1 → dma_cli-0.1.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "dma-cli"
-version = "0.1.1"
+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()