aispeech-ds-cli 0.1.0__py3-none-win_amd64.whl

aispeech_ds_cli-0.1.0.dist-info/METADATA

@@ -0,0 +1,10 @@
+Metadata-Version: 2.4
+Name: aispeech-ds-cli
+Version: 0.1.0
+Summary: Dataset external client CLI
+Classifier: Programming Language :: Python :: 3
+Classifier: Operating System :: Microsoft :: Windows
+Classifier: Operating System :: MacOS
+Classifier: Operating System :: POSIX :: Linux
+Requires-Python: >=3.9
+Requires-Dist: packaging>=23

aispeech_ds_cli-0.1.0.dist-info/RECORD

@@ -0,0 +1,8 @@
+ds_cli_bin/__init__.py,sha256=KfrBbyZ4r2fRExiaGVbZttfXwDgHZplxZJyhQZKK_Vs,9286
+ds_cli_bin/release_config.py,sha256=Fb9IvCCG5bNENbTi4MkRKSmzeNm5NjeQknYLcYpdsg4,99
+ds_cli_bin/bin/ds-cli.exe,sha256=D0EAWbmtA5swEwBlELjOACwMTlohXLwYVaKPMoPOb-0,10779040
+aispeech_ds_cli-0.1.0.dist-info/METADATA,sha256=67NlDXMW7uEe1KcFVflxP-wTNu__D9WtwceaGBQCkbE,344
+aispeech_ds_cli-0.1.0.dist-info/WHEEL,sha256=hVx9elvUDfBjRmbl8JwIcXXikst35RZTZK9nspfI_28,98
+aispeech_ds_cli-0.1.0.dist-info/entry_points.txt,sha256=OOhUKuncoBdhvN1AfVO6SPkmse792Vg7NsIKZPnGThY,43
+aispeech_ds_cli-0.1.0.dist-info/top_level.txt,sha256=-7n-fnvxB_XT3_c82PDe2nLak3vBb9rULEnp2m_4yPQ,11
+aispeech_ds_cli-0.1.0.dist-info/RECORD,,

aispeech_ds_cli-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (83.0.0)
+Root-Is-Purelib: false
+Tag: py3-none-win_amd64
+

aispeech_ds_cli-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+ds-cli = ds_cli_bin:main

aispeech_ds_cli-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+ds_cli_bin

ds_cli_bin/__init__.py

@@ -0,0 +1,312 @@
+import json
+import os
+import shlex
+import subprocess
+import sys
+import time
+from importlib.metadata import version
+from importlib.resources import files
+from pathlib import Path
+from urllib.request import urlopen
+
+
+PACKAGE_NAME = "aispeech-ds-cli"
+DEFAULT_CHECK_INTERVAL_SECONDS = 0
+try:
+    from .release_config import DEFAULT_PIP_INDEX_ARGS, DEFAULT_PYPI_JSON_URL
+except Exception:
+    DEFAULT_PYPI_JSON_URL = f"https://pypi.org/pypi/{PACKAGE_NAME}/json"
+    DEFAULT_PIP_INDEX_ARGS = []
+
+
+def _cache_path():
+    cache_name = os.getenv("DS_CLI_UPDATE_CACHE", "update-check.json")
+    return Path.home() / ".ds-cli" / cache_name
+
+
+def _pypi_json_url():
+    return os.getenv("DS_CLI_UPDATE_JSON_URL", DEFAULT_PYPI_JSON_URL)
+
+
+def _pip_index_args():
+    override = os.getenv("DS_CLI_PIP_INDEX_ARGS")
+    if override:
+        return override.split()
+    return list(DEFAULT_PIP_INDEX_ARGS)
+
+
+def _check_interval_seconds():
+    try:
+        return max(0, int(os.getenv("DS_CLI_UPDATE_CACHE_SECONDS", DEFAULT_CHECK_INTERVAL_SECONDS)))
+    except ValueError:
+        return DEFAULT_CHECK_INTERVAL_SECONDS
+
+
+def _load_cached_latest():
+    path = _cache_path()
+    if not path.exists():
+        return None
+
+    try:
+        data = json.loads(path.read_text(encoding="utf-8"))
+        checked_at = data.get("checked_at", 0)
+        if time.time() - checked_at > _check_interval_seconds():
+            return None
+        return data.get("latest")
+    except Exception:
+        return None
+
+
+def _save_cached_latest(latest):
+    path = _cache_path()
+    path.parent.mkdir(parents=True, exist_ok=True)
+    path.write_text(
+        json.dumps({"checked_at": time.time(), "latest": latest}),
+        encoding="utf-8",
+    )
+
+
+def _fetch_latest_version(version_cls):
+    with urlopen(_pypi_json_url(), timeout=2) as response:
+        data = json.loads(response.read().decode("utf-8"))
+
+    versions = []
+    for raw_version, files_info in data.get("releases", {}).items():
+        parsed = version_cls(raw_version)
+        if parsed.is_prerelease:
+            continue
+        if files_info and all(item.get("yanked", False) for item in files_info):
+            continue
+        versions.append(parsed)
+
+    if not versions:
+        return None
+
+    return str(max(versions))
+
+
+def _upgrade_command():
+    base = [_python_executable(), "-m", "pip", "install", "-U"]
+    base.extend(_pip_index_args())
+    base.append(PACKAGE_NAME)
+    return base
+
+
+def _python_executable():
+    executable = Path(sys.executable)
+    if os.name == "nt" and executable.name.lower() == "ds-cli.exe":
+        candidate = Path(sys.prefix) / "Scripts" / "python.exe"
+        if candidate.exists():
+            return str(candidate)
+    return sys.executable
+
+
+def _format_command(args):
+    if os.name == "nt":
+        return subprocess.list2cmdline(args)
+    return " ".join(shlex.quote(arg) for arg in args)
+
+
+def _run_upgrade_after_exit(command):
+    updater_path = Path.home() / ".ds-cli" / f"self-update-{os.getpid()}.py"
+    updater_path.parent.mkdir(parents=True, exist_ok=True)
+    updater_path.write_text(
+        """
+import ctypes
+import json
+import os
+import subprocess
+import sys
+import time
+
+
+def wait_for_parent(pid, timeout_seconds=60):
+    if os.name == "nt":
+        handle = ctypes.windll.kernel32.OpenProcess(0x00100000, False, pid)
+        if handle:
+            try:
+                ctypes.windll.kernel32.WaitForSingleObject(handle, int(timeout_seconds * 1000))
+            finally:
+                ctypes.windll.kernel32.CloseHandle(handle)
+            return
+
+    deadline = time.time() + timeout_seconds
+    while time.time() < deadline:
+        try:
+            os.kill(pid, 0)
+        except OSError:
+            return
+        time.sleep(0.2)
+
+
+def format_command(command):
+    if os.name == "nt":
+        return subprocess.list2cmdline(command)
+    return " ".join(command)
+
+
+def restore_prompt():
+    if os.name != "nt" or os.getenv("DS_CLI_RESTORE_PROMPT", "1") == "0":
+        return
+
+    try:
+        from ctypes import wintypes
+
+        STD_INPUT_HANDLE = -10
+        KEY_EVENT = 0x0001
+        VK_RETURN = 0x0D
+
+        class KEY_EVENT_RECORD(ctypes.Structure):
+            _fields_ = [
+                ("bKeyDown", wintypes.BOOL),
+                ("wRepeatCount", wintypes.WORD),
+                ("wVirtualKeyCode", wintypes.WORD),
+                ("wVirtualScanCode", wintypes.WORD),
+                ("uChar", wintypes.WCHAR),
+                ("dwControlKeyState", wintypes.DWORD),
+            ]
+
+        class INPUT_RECORD(ctypes.Structure):
+            _fields_ = [
+                ("EventType", wintypes.WORD),
+                ("KeyEvent", KEY_EVENT_RECORD),
+            ]
+
+        records = (INPUT_RECORD * 2)()
+        records[0].EventType = KEY_EVENT
+        records[0].KeyEvent = KEY_EVENT_RECORD(True, 1, VK_RETURN, 0, "\\r", 0)
+        records[1].EventType = KEY_EVENT
+        records[1].KeyEvent = KEY_EVENT_RECORD(False, 1, VK_RETURN, 0, "\\r", 0)
+
+        written = wintypes.DWORD()
+        handle = ctypes.windll.kernel32.GetStdHandle(STD_INPUT_HANDLE)
+        ctypes.windll.kernel32.WriteConsoleInputW(handle, records, 2, ctypes.byref(written))
+    except Exception:
+        pass
+
+
+def main():
+    parent_pid = int(sys.argv[1])
+    command = json.loads(sys.argv[2])
+    wait_for_parent(parent_pid)
+    time.sleep(0.5)
+    print("[ds-cli] Running update command:")
+    print("  " + format_command(command))
+    rc = subprocess.call(command)
+    if rc == 0:
+        print("[ds-cli] Update completed. Please run your command again.")
+    else:
+        print("[ds-cli] Update failed. Please run the update command manually.", file=sys.stderr)
+    sys.stdout.flush()
+    sys.stderr.flush()
+    restore_prompt()
+    try:
+        os.remove(__file__)
+    except OSError:
+        pass
+    return rc
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())
+""".lstrip(),
+        encoding="utf-8",
+    )
+
+    subprocess.Popen(
+        [_python_executable(), str(updater_path), str(os.getpid()), json.dumps(command)],
+        close_fds=False,
+    )
+    print("[ds-cli] 已启动自动升级进程，当前命令将退出。")
+    print("[ds-cli] Auto-update has started. The current command will exit.")
+    return 2
+
+
+def _can_update_in_current_process():
+    return os.name != "nt"
+
+
+def _run_upgrade():
+    command = _upgrade_command()
+    print("[ds-cli] 正在升级，请稍候...")
+    print("[ds-cli] Updating, please wait...")
+    if not _can_update_in_current_process():
+        return _run_upgrade_after_exit(command)
+
+    result = subprocess.run(command)
+    if result.returncode == 0:
+        print("[ds-cli] 升级完成，请重新执行命令。")
+        print("[ds-cli] Update completed. Please run your command again.")
+    else:
+        print(
+            f"[ds-cli] 升级失败，请手动执行：\n"
+            f"  {_format_command(command)}\n"
+            f"[ds-cli] Update failed. Please run manually:\n"
+            f"  {_format_command(command)}",
+            file=sys.stderr,
+        )
+    return result.returncode or 2
+
+
+def _confirm_upgrade():
+    if not sys.stdin.isatty():
+        return False
+    try:
+        answer = input("[ds-cli] 是否现在自动升级？输入 y 继续，其他输入退出 / Update now? Enter y to continue: ")
+    except (EOFError, KeyboardInterrupt):
+        return False
+    return answer.strip().lower() == "y"
+
+
+def _ensure_latest_version():
+    try:
+        from packaging.version import Version
+
+        current = Version(version(PACKAGE_NAME))
+
+        latest = _load_cached_latest()
+        if latest is None:
+            latest = _fetch_latest_version(Version)
+            if latest:
+                _save_cached_latest(latest)
+
+        if latest and Version(latest) > current:
+            command = _upgrade_command()
+            print(
+                f"[ds-cli] 当前工具版本过低，请升级。\n"
+                f"[ds-cli] 当前版本：{current}，最新版本：{latest}。\n"
+                f"[ds-cli] The current tool version is too old. Please upgrade. "
+                f"Current version: {current}; latest version: {latest}.\n"
+                f"[ds-cli] 升级命令 / Upgrade command:\n"
+                f"  {_format_command(command)}",
+                file=sys.stderr,
+                flush=True,
+            )
+            if os.getenv("DS_CLI_AUTO_UPDATE") == "1" or _confirm_upgrade():
+                return _run_upgrade()
+            return 2
+    except Exception as exc:
+        command = _upgrade_command()
+        print(
+            f"[ds-cli] 无法校验最新版本，命令不能继续执行。\n"
+            f"[ds-cli] Failed to verify latest version, so the command cannot continue.\n"
+            f"[ds-cli] 原因 / Reason: {exc}\n"
+            f"[ds-cli] 可先尝试升级 / Try updating first:\n"
+            f"  {_format_command(command)}",
+            file=sys.stderr,
+            flush=True,
+        )
+        return 3
+
+    return 0
+
+
+def main():
+    update_status = _ensure_latest_version()
+    if update_status != 0:
+        return update_status
+
+    exe_name = "ds-cli.exe" if sys.platform.startswith("win") else "ds-cli"
+    exe = files(__package__).joinpath("bin", exe_name)
+    result = subprocess.run([str(exe), *sys.argv[1:]])
+    return result.returncode

ds_cli_bin/release_config.py

@@ -0,0 +1,2 @@
+DEFAULT_PYPI_JSON_URL = "https://pypi.org/pypi/aispeech-ds-cli/json"
+DEFAULT_PIP_INDEX_ARGS = []