@agentunion/kite 1.3.2 → 1.5.0

package/core/dependency_checker.py

@@ -0,0 +1,250 @@
+"""
+依赖检查和自动安装模块
+
+在 Kite 启动前检查 requirements.txt 中的依赖是否满足，
+不满足则自动安装到用户级虚拟环境 ~/.kite/venv/
+"""
+
+import os
+import sys
+import subprocess
+from pathlib import Path
+from typing import List, Tuple, Optional
+
+
+def get_venv_path() -> Path:
+    """获取 Kite 虚拟环境路径"""
+    return Path.home() / ".kite" / "venv"
+
+
+def get_site_packages_path() -> Optional[Path]:
+    """获取虚拟环境的 site-packages 路径"""
+    venv_path = get_venv_path()
+
+    if os.name == "nt":  # Windows
+        site_packages = venv_path / "Lib" / "site-packages"
+    else:  # Linux / macOS
+        python_version = f"python{sys.version_info.major}.{sys.version_info.minor}"
+        site_packages = venv_path / "lib" / python_version / "site-packages"
+
+    return site_packages if site_packages.exists() else None
+
+
+def activate_venv():
+    """将虚拟环境的 site-packages 添加到 sys.path"""
+    site_packages = get_site_packages_path()
+
+    if site_packages and site_packages not in sys.path:
+        sys.path.insert(0, str(site_packages))
+        print(f"[依赖] 已激活虚拟环境: {site_packages}")
+
+
+def ensure_venv_exists() -> bool:
+    """确保虚拟环境存在，不存在则创建"""
+    venv_path = get_venv_path()
+
+    if venv_path.exists():
+        return True
+
+    print(f"[依赖] 虚拟环境不存在，正在创建: {venv_path}")
+
+    try:
+        venv_path.parent.mkdir(parents=True, exist_ok=True)
+        subprocess.run(
+            [sys.executable, "-m", "venv", str(venv_path)],
+            check=True,
+            capture_output=True,
+            text=True
+        )
+        print(f"[依赖] 虚拟环境创建成功")
+        return True
+    except subprocess.CalledProcessError as e:
+        print(f"[依赖] 虚拟环境创建失败: {e.stderr}")
+        return False
+
+
+def get_venv_pip() -> Path:
+    """获取虚拟环境的 pip 路径"""
+    venv_path = get_venv_path()
+
+    if os.name == "nt":  # Windows
+        return venv_path / "Scripts" / "pip.exe"
+    else:  # Linux / macOS
+        return venv_path / "bin" / "pip"
+
+
+def parse_requirements(requirements_file: Path) -> List[Tuple[str, str]]:
+    """
+    解析 requirements.txt
+
+    Returns:
+        List of (package_name, version_spec) tuples
+        例如: [("fastapi", ">=0.115.0"), ("uvicorn", ">=0.32.0")]
+    """
+    requirements = []
+
+    with open(requirements_file, "r", encoding="utf-8") as f:
+        for line in f:
+            line = line.strip()
+
+            # 跳过注释和空行
+            if not line or line.startswith("#"):
+                continue
+
+            # 解析包名和版本
+            if ">=" in line:
+                package, version = line.split(">=", 1)
+                requirements.append((package.strip(), f">={version.strip()}"))
+            elif "==" in line:
+                package, version = line.split("==", 1)
+                requirements.append((package.strip(), f"=={version.strip()}"))
+            else:
+                requirements.append((line.strip(), ""))
+
+    return requirements
+
+
+def check_package_installed(package_name: str, version_spec: str) -> bool:
+    """
+    检查包是否已安装且版本满足要求
+
+    Args:
+        package_name: 包名
+        version_spec: 版本要求，如 ">=0.115.0"
+
+    Returns:
+        True if installed and version matches, False otherwise
+    """
+    try:
+        import importlib.metadata
+
+        # 检查包是否存在
+        try:
+            installed_version = importlib.metadata.version(package_name)
+        except importlib.metadata.PackageNotFoundError:
+            return False
+
+        # 如果没有版本要求，只要安装了就行
+        if not version_spec:
+            return True
+
+        # 简单的版本比较（只支持 >= 和 ==）
+        if version_spec.startswith(">="):
+            required_version = version_spec[2:].strip()
+            return _version_gte(installed_version, required_version)
+        elif version_spec.startswith("=="):
+            required_version = version_spec[2:].strip()
+            return installed_version == required_version
+
+        return True
+
+    except Exception as e:
+        print(f"[依赖] 检查包 {package_name} 时出错: {e}")
+        return False
+
+
+def _version_gte(v1: str, v2: str) -> bool:
+    """简单的版本比较：v1 >= v2"""
+    try:
+        parts1 = [int(x) for x in v1.split(".")[:3]]
+        parts2 = [int(x) for x in v2.split(".")[:3]]
+
+        # 补齐到 3 位
+        while len(parts1) < 3:
+            parts1.append(0)
+        while len(parts2) < 3:
+            parts2.append(0)
+
+        return parts1 >= parts2
+    except:
+        return True  # 解析失败时假设满足
+
+
+def install_packages(packages: List[str]) -> bool:
+    """
+    使用虚拟环境的 pip 安装包
+
+    Args:
+        packages: 包列表，如 ["fastapi>=0.115.0", "uvicorn>=0.32.0"]
+
+    Returns:
+        True if success, False otherwise
+    """
+    pip_path = get_venv_pip()
+
+    if not pip_path.exists():
+        print(f"[依赖] pip 不存在: {pip_path}")
+        return False
+
+    print(f"[依赖] 正在安装 {len(packages)} 个包...")
+
+    try:
+        # 使用 pip install 安装所有缺失的包
+        cmd = [str(pip_path), "install"] + packages
+
+        result = subprocess.run(
+            cmd,
+            check=True,
+            capture_output=True,
+            text=True
+        )
+
+        print(f"[依赖] 安装成功")
+        return True
+
+    except subprocess.CalledProcessError as e:
+        print(f"[依赖] 安装失败: {e.stderr}")
+        return False
+
+
+def check_and_install_dependencies(requirements_file: Path) -> bool:
+    """
+    检查并安装依赖
+
+    Args:
+        requirements_file: requirements.txt 文件路径
+
+    Returns:
+        True if all dependencies satisfied, False otherwise
+    """
+    if not requirements_file.exists():
+        print(f"[依赖] requirements.txt 不存在: {requirements_file}")
+        return True  # 没有依赖文件就跳过
+
+    # 1. 确保虚拟环境存在
+    if not ensure_venv_exists():
+        print("[依赖] 无法创建虚拟环境，跳过依赖检查")
+        return False
+
+    # 2. 激活虚拟环境（添加到 sys.path）
+    activate_venv()
+
+    # 3. 解析 requirements.txt
+    requirements = parse_requirements(requirements_file)
+
+    if not requirements:
+        print("[依赖] requirements.txt 为空")
+        return True
+
+    # 4. 检查哪些包需要安装
+    missing_packages = []
+
+    for package_name, version_spec in requirements:
+        if not check_package_installed(package_name, version_spec):
+            spec = f"{package_name}{version_spec}" if version_spec else package_name
+            missing_packages.append(spec)
+            print(f"[依赖] 缺失: {spec}")
+
+    # 5. 如果有缺失的包，安装它们
+    if missing_packages:
+        print(f"[依赖] 发现 {len(missing_packages)} 个缺失的包，开始安装...")
+
+        if not install_packages(missing_packages):
+            print("[依赖] 安装失败，Kite 可能无法正常运行")
+            return False
+
+        print("[依赖] 所有依赖已安装")
+    else:
+        print("[依赖] 所有依赖已满足")
+
+    return True