feat: Add UV utility functions module

Co-Authored-By: Aaron <AJ> Steers <aj@airbyte.io>

Author: devin-ai-integration[bot]

airbyte/_util/uv_util.py

@@ -0,0 +1,86 @@
+"""Utility functions for UV virtual environment management."""
+
+from __future__ import annotations
+
+import os
+import subprocess
+from pathlib import Path
+
+
+def run_uv_command(args: list[str], env: dict | None = None, check: bool = False) -> subprocess.CompletedProcess:
+    """Run a UV command and return the result.
+
+    Args:
+        args: Command arguments to pass to UV
+        env: Optional environment variables
+        check: Whether to raise an exception on non-zero return code
+
+    Returns:
+        CompletedProcess instance with command output
+    """
+    return subprocess.run(
+        ["uv", *args],
+        capture_output=True,
+        text=True,
+        env=env,
+        check=check,
+    )
+
+
+def create_venv(venv_path: Path, with_pip: bool = True) -> None:
+    """Create a virtual environment using UV.
+
+    Args:
+        venv_path: Path where the virtual environment should be created
+        with_pip: Whether to include pip and other seed packages
+    """
+    args = ["venv"]
+    if with_pip:
+        args.append("--seed")
+    args.append(str(venv_path))
+    run_uv_command(args, check=True)
+
+
+def install_package(venv_path: Path, pip_url: str) -> None:
+    """Install a package into a virtual environment using UV.
+
+    Args:
+        venv_path: Path to the virtual environment
+        pip_url: Package specification (name, URL, or path) to install
+    """
+    venv_env = get_venv_env(venv_path)
+    run_uv_command(
+        ["pip", "install", pip_url],
+        env=venv_env,
+        check=True,
+    )
+
+
+def get_venv_env(venv_path: Path) -> dict:
+    """Get environment variables for a virtual environment.
+
+    Args:
+        venv_path: Path to the virtual environment
+
+    Returns:
+        Dict of environment variables
+    """
+    env = os.environ.copy()
+    env["VIRTUAL_ENV"] = str(venv_path)
+    env["PATH"] = f"{venv_path}/bin:{os.environ['PATH']}"
+    return env
+
+
+def get_executable_path(venv_path: Path, executable: str) -> Path:
+    """Get the path to an executable in a virtual environment.
+
+    Args:
+        venv_path: Path to the virtual environment
+        executable: Name of the executable
+
+    Returns:
+        Path to the executable
+    """
+    bin_dir = "Scripts" if os.name == "nt" else "bin"
+    suffix = ".exe" if os.name == "nt" else ""
+    return venv_path / bin_dir / f"{executable}{suffix}"