initial commit

pythonapp/Libs/pip_api.py

@@ -0,0 +1,138 @@
+import subprocess
+from typing import List, Optional
+import re
+import subprocess
+import shlex
+from typing import List, Optional, Dict, Any, Union
+from dataclasses import dataclass
+
+@dataclass
+class PipResult:
+    """Результат выполнения команды pip."""
+    exit_code: int
+    stdout: str
+    stderr: str
+    command: str
+    success: bool
+
+
+class pip_api:
+    @staticmethod
+    def get_pkg_versions(package: str, index_url: Optional[str] = None) -> List[str]:
+        # Формируем базовую команду
+        command = [
+            'python3', '-m', 'pip',
+            'install',
+            '--use-deprecated=legacy-resolver',  # Для совместимости со старыми репозиториями
+            '--dry-run',
+            '--no-deps',
+            f'{package}==0.0.0.0'  # Специально несуществующая версия
+        ]
+
+        if index_url:
+            command.extend(['--index-url', index_url])
+
+        # Запускаем процесс
+        result = subprocess.run(
+            command,
+            capture_output=True,
+            text=True,
+            timeout=30  # Таймаут для избежания зависаний
+        )
+
+        #print(result.stdout)
+        #print(result.stderr)
+        # Парсим вывод для получения версий
+        if result.stderr:
+            match = re.search(r'from versions: (.+?)\)', result.stderr)
+            if match:
+                versions = match.group(1).split(', ')
+                return [v.strip() for v in versions if v.strip()]
+
+        return []
+
+    @staticmethod
+    def run_pip_install(
+        pip_path: str,
+        packages: List[str],
+        index_url: Optional[str] = None,
+        extra_index_urls: Optional[List[str]] = None,
+        dry_run: bool = False
+    ) -> PipResult:
+        """
+            Выполняет установку пакетов через pip с указанными параметрами.
+
+            Args:
+                pip_path: Путь к исполняемому файлу pip
+                packages: Список пакетов для установки
+                index_url: Основной URL репозитория пакетов
+                extra_index_urls: Дополнительные URLs репозиториев пакетов
+                dry_run: Если True, команда только выводится, но не выполняется
+
+            Returns:
+                PipResult: Объект с результатами выполнения команды
+
+            Raises:
+                FileNotFoundError: Если pip_path не существует
+                ValueError: Если packages пуст
+            """
+        # Проверка входных параметров
+        if not packages:
+            raise ValueError("Список пакетов не может быть пустым")
+
+        # Формирование команды
+        command = [pip_path, "install"]
+
+        if dry_run:
+            command.append('--dry-run')
+
+        # Добавление основного index-url
+        if index_url:
+            command.extend(["--index-url", index_url])
+
+        # Добавление дополнительных index-urls
+        if extra_index_urls:
+            for url in extra_index_urls:
+                command.extend(["--extra-index-url", url])
+
+        # Добавление пакетов
+        command.extend(packages)
+
+        # Преобразование команды в строку для вывода
+        command_str = " ".join(shlex.quote(arg) for arg in command)
+
+        # Выполнение команды
+        try:
+            result = subprocess.run(
+                command,
+                capture_output=True,
+                text=True,
+                check=False  # Не вызываем исключение при ненулевом коде возврата
+            )
+
+            return PipResult(
+                exit_code=result.returncode,
+                stdout=result.stdout,
+                stderr=result.stderr,
+                command=command_str,
+                success=result.returncode == 0
+            )
+
+        except FileNotFoundError:
+            raise FileNotFoundError(f"Файл pip не найден по пути: {pip_path}")
+        except Exception as e:
+            return PipResult(
+                exit_code=-1,
+                stdout="",
+                stderr=str(e),
+                command=command_str,
+                success=False
+            )
+
+
+
+
+
+if __name__ == "__main__":
+    versions = pip_api.get_pkg_versions("torch", "https://download.pytorch.org/whl/cu121")
+    print(versions)