openbench-cli 0.0.5__py3-none-win_amd64.whl

ds_cli_bin/__init__.py

@@ -0,0 +1,326 @@
+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 = "openbench-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 = [
+        "-i",
+        "https://pypi.org/simple",
+    ]
+
+
+def _cache_path():
+    cache_name = os.getenv("DS_CLI_UPDATE_CACHE", "update-check.json")
+    return Path.home() / ".ob" / 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() == "ob.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() / ".ob" / 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("[ob] Running update command:")
+    print("  " + format_command(command))
+    rc = subprocess.call(command)
+    if rc == 0:
+        print("[ob] Update completed. Please run your command again.")
+    else:
+        print("[ob] 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("[ob] 已启动自动升级进程，当前命令将退出。")
+    print("[ob] 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("[ob] 正在升级，请稍候...")
+    print("[ob] 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("[ob] 升级完成，请重新执行命令。")
+        print("[ob] Update completed. Please run your command again.")
+    else:
+        print(
+            f"[ob] 升级失败，请手动执行：\n"
+            f"  {_format_command(command)}\n"
+            f"[ob] 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("[ob] 是否现在自动升级？输入 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"[ob] 当前工具版本过低，请升级。\n"
+                f"[ob] 当前版本：{current}，最新版本：{latest}。\n"
+                f"[ob] The current tool version is too old. Please upgrade. "
+                f"Current version: {current}; latest version: {latest}.\n"
+                f"[ob] 升级命令 / 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:
+        # 版本校验失败（网络不可达/超时/镜像异常等）：不阻断命令，WARN 后继续执行（"失败放行"）。
+        # "版本过旧"（latest > current）分支不受影响，仍走原流程（提示升级 / 拦截）。
+        command = _upgrade_command()
+        print(
+            f"[ob] 未校验到最新版本，已跳过并继续执行（版本校验失败不影响命令运行）。\n"
+            f"[ob] Skipping the version check and continuing. Reason: {exc}\n"
+            f"[ob] 如需手动升级 / To upgrade manually:\n"
+            f"  {_format_command(command)}",
+            file=sys.stderr,
+            flush=True,
+        )
+        return 0
+
+    return 0
+
+
+def main():
+    # ob -V / ob --version：直接打印已安装包的版本（= PyPI 上的 openbench-cli 版本），
+    # 在升级检查之前处理，避免触发"请升级"流程。
+    if len(sys.argv) >= 2 and sys.argv[1] in ("-V", "--version"):
+        try:
+            print(version(PACKAGE_NAME))
+        except Exception as exc:
+            print(f"[ob] 无法读取版本信息: {exc}", file=sys.stderr)
+            return 3
+        return 0
+
+    update_status = _ensure_latest_version()
+    if update_status != 0:
+        return update_status
+
+    exe_name = "ob.exe" if sys.platform.startswith("win") else "ob"
+    exe = files(__package__).joinpath("bin", exe_name)
+    result = subprocess.run([str(exe), *sys.argv[1:]])
+    return result.returncode

ds_cli_bin/bin/_internal/attrs-26.1.0.dist-info/INSTALLER

@@ -0,0 +1 @@
+pip