127 lines
3.3 KiB
Python
127 lines
3.3 KiB
Python
#!/usr/bin/env python3
|
|
|
|
from __future__ import annotations
|
|
|
|
import hashlib
|
|
import subprocess
|
|
import sys
|
|
import traceback
|
|
from pathlib import Path
|
|
|
|
sys.stderr = sys.stdout
|
|
|
|
|
|
def _skill_root_from(script_dir: Path) -> Path:
|
|
return script_dir.parent
|
|
|
|
|
|
def _venv_dir(script_dir: Path) -> Path:
|
|
return _skill_root_from(script_dir) / ".venv"
|
|
|
|
|
|
def _venv_python(venv_dir: Path) -> Path:
|
|
if sys.platform == "win32":
|
|
return venv_dir / "Scripts" / "python.exe"
|
|
return venv_dir / "bin" / "python"
|
|
|
|
|
|
def _stamp_file(venv_dir: Path) -> Path:
|
|
return venv_dir / ".req_hash"
|
|
|
|
|
|
def _file_hash(path: Path) -> str:
|
|
return hashlib.sha256(path.read_bytes()).hexdigest()
|
|
|
|
|
|
def _deps_up_to_date(requirements_file: Path, venv_dir: Path) -> bool:
|
|
stamp = _stamp_file(venv_dir)
|
|
if not stamp.is_file():
|
|
return False
|
|
return stamp.read_text().strip() == _file_hash(requirements_file)
|
|
|
|
|
|
def _write_stamp(requirements_file: Path, venv_dir: Path) -> None:
|
|
_stamp_file(venv_dir).write_text(_file_hash(requirements_file))
|
|
|
|
|
|
def _ensure_venv(venv_dir: Path, venv_python: Path) -> int:
|
|
if venv_python.is_file():
|
|
return 0
|
|
|
|
sys.stdout.write(f"未检测到技能虚拟环境，正在创建: {venv_dir}\n")
|
|
command = [
|
|
sys.executable,
|
|
"-m",
|
|
"venv",
|
|
str(venv_dir),
|
|
]
|
|
|
|
try:
|
|
subprocess.run(command, check=True, stdout=sys.stdout, stderr=sys.stdout)
|
|
except subprocess.CalledProcessError as exc:
|
|
sys.stdout.write(f"创建虚拟环境失败，退出码: {exc.returncode}\n")
|
|
return exc.returncode or 1
|
|
|
|
return 0
|
|
|
|
def main() -> int:
|
|
script_dir = Path(__file__).resolve().parent
|
|
requirements_file = script_dir / "requirements.txt"
|
|
venv_dir = _venv_dir(script_dir)
|
|
venv_python = _venv_python(venv_dir)
|
|
|
|
if not requirements_file.is_file():
|
|
sys.stdout.write(f"未找到依赖文件: {requirements_file}\n")
|
|
return 1
|
|
|
|
ensure_result = _ensure_venv(venv_dir, venv_python)
|
|
if ensure_result != 0:
|
|
return ensure_result
|
|
|
|
if _deps_up_to_date(requirements_file, venv_dir):
|
|
sys.stdout.write("依赖已是最新，跳过安装\n")
|
|
return 0
|
|
|
|
command = [
|
|
str(venv_python),
|
|
"-m",
|
|
"pip",
|
|
"install",
|
|
"--upgrade",
|
|
"pip",
|
|
]
|
|
|
|
try:
|
|
subprocess.run(command, check=True, stdout=sys.stdout, stderr=sys.stdout)
|
|
except subprocess.CalledProcessError as exc:
|
|
sys.stdout.write(f"升级 pip 失败，退出码: {exc.returncode}\n")
|
|
return exc.returncode or 1
|
|
|
|
command = [
|
|
str(venv_python),
|
|
"-m",
|
|
"pip",
|
|
"install",
|
|
"-r",
|
|
str(requirements_file),
|
|
]
|
|
|
|
try:
|
|
subprocess.run(command, check=True, stdout=sys.stdout, stderr=sys.stdout)
|
|
except subprocess.CalledProcessError as exc:
|
|
sys.stdout.write(f"安装依赖失败，退出码: {exc.returncode}\n")
|
|
return exc.returncode or 1
|
|
|
|
_write_stamp(requirements_file, venv_dir)
|
|
sys.stdout.write(f"依赖安装完成，当前技能虚拟环境: {venv_dir}\n")
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
raise SystemExit(main())
|
|
except SystemExit:
|
|
raise
|
|
except Exception:
|
|
traceback.print_exc(file=sys.stdout)
|
|
raise SystemExit(1) |