Use uv by default when pip installing (#215)

Author: basnijholt

README.md

@@ -18,6 +18,7 @@ This often leads to confusion and inefficiency, as developers juggle between mul
 - **📝 Unified Dependency File**: Use either `requirements.yaml` or `pyproject.toml` to manage both Conda and Pip dependencies in one place.
 - **⚙️ Build System Integration**: Integrates with Setuptools and Hatchling for automatic dependency handling during `pip install ./your-package`.
 - **💻 One-Command Installation**: `unidep install` handles Conda, Pip, and local dependencies effortlessly.
+- **⚡️ Fast Pip Operations**: Leverages `uv` (if installed) for faster pip installations.
 - **🏢 Monorepo-Friendly**: Render (multiple) `requirements.yaml` or `pyproject.toml` files into one Conda `environment.yaml` file and maintain fully consistent global *and* per sub package `conda-lock` files.
 - **🌍 Platform-Specific Support**: Specify dependencies for different operating systems or architectures.
 - **🔧 `pip-compile` Integration**: Generate fully pinned `requirements.txt` files from `requirements.yaml` or `pyproject.toml` files using `pip-compile`.
@@ -557,6 +558,7 @@ usage: unidep install [-h] [-v] [-e] [--skip-local] [--skip-pip]
                       [--conda-env-name CONDA_ENV_NAME | --conda-env-prefix CONDA_ENV_PREFIX]
                       [--dry-run] [--ignore-pin IGNORE_PIN]
                       [--overwrite-pin OVERWRITE_PIN] [-f CONDA_LOCK_FILE]
+                      [--no-uv]
                       files [files ...]
 
 Automatically install all dependencies from one or more `requirements.yaml` or
@@ -591,7 +593,8 @@ options:
                         specifying a different package to skip. For example,
                         use `--skip-dependency pandas` to skip installing
                         pandas.
-  --no-dependencies     Skip installing dependencies from `requirements.yaml`
+  --no-dependencies, --no-deps
+                        Skip installing dependencies from `requirements.yaml`
                         or `pyproject.toml` file(s) and only install local
                         package(s). Useful after installing a `conda-lock.yml`
                         file because then all dependencies have already been
@@ -621,6 +624,8 @@ options:
                         the new environment. Assumes that the lock file
                         contains all dependencies. Must be used with `--conda-
                         env-name` or `--conda-env-prefix`.
+  --no-uv               Disables the use of `uv` for pip install. By default,
+                        `uv` is used if it is available in the PATH.
 ```
 
 <!-- OUTPUT:END -->
@@ -645,6 +650,7 @@ usage: unidep install [-h] [-v] [-e] [--skip-local] [--skip-pip]
                       [--conda-env-name CONDA_ENV_NAME | --conda-env-prefix CONDA_ENV_PREFIX]
                       [--dry-run] [--ignore-pin IGNORE_PIN]
                       [--overwrite-pin OVERWRITE_PIN] [-f CONDA_LOCK_FILE]
+                      [--no-uv]
                       files [files ...]
 
 Automatically install all dependencies from one or more `requirements.yaml` or
@@ -679,7 +685,8 @@ options:
                         specifying a different package to skip. For example,
                         use `--skip-dependency pandas` to skip installing
                         pandas.
-  --no-dependencies     Skip installing dependencies from `requirements.yaml`
+  --no-dependencies, --no-deps
+                        Skip installing dependencies from `requirements.yaml`
                         or `pyproject.toml` file(s) and only install local
                         package(s). Useful after installing a `conda-lock.yml`
                         file because then all dependencies have already been
@@ -709,6 +716,8 @@ options:
                         the new environment. Assumes that the lock file
                         contains all dependencies. Must be used with `--conda-
                         env-name` or `--conda-env-prefix`.
+  --no-uv               Disables the use of `uv` for pip install. By default,
+                        `uv` is used if it is available in the PATH.
 ```
 
 <!-- OUTPUT:END -->

example/README.md

@@ -114,7 +114,7 @@ $ unidep install --dry-run -e ./setup_py_project
 
 📝 Found local dependencies: {'setup_py_project': ['hatch_project', 'setuptools_project']}
 
-📦 Installing project with `/opt/hostedtoolcache/Python/3.12.8/x64/bin/python -m pip install --no-dependencies -e /home/runner/work/unidep/unidep/example/hatch_project -e /home/runner/work/unidep/unidep/example/setuptools_project -e ./setup_py_project`
+📦 Installing project with `/opt/hostedtoolcache/Python/3.12.8/x64/bin/python -m pip install --no-deps -e /home/runner/work/unidep/unidep/example/hatch_project -e /home/runner/work/unidep/unidep/example/setuptools_project -e ./setup_py_project`
 
 ```
 
@@ -161,7 +161,7 @@ $ unidep install-all -e --dry-run
 
 📝 Found local dependencies: {'pyproject_toml_project': ['hatch_project'], 'setup_py_project': ['hatch_project', 'setuptools_project'], 'setuptools_project': ['hatch_project']}
 
-📦 Installing project with `/opt/hostedtoolcache/Python/3.12.8/x64/bin/python -m pip install --no-dependencies -e ./hatch2_project -e ./hatch_project -e ./pyproject_toml_project -e ./setup_py_project -e ./setuptools_project`
+📦 Installing project with `/opt/hostedtoolcache/Python/3.12.8/x64/bin/python -m pip install --no-deps -e ./hatch2_project -e ./hatch_project -e ./pyproject_toml_project -e ./setup_py_project -e ./setuptools_project`
 
 ```
 

tests/test_cli.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 import os
+import platform
 import re
 import shutil
 import subprocess
@@ -127,10 +128,26 @@ def test_install_all_command(capsys: pytest.CaptureFixture) -> None:
     assert "Installing pip dependencies" in captured.out
     projects = [REPO_ROOT / "example" / p for p in EXAMPLE_PROJECTS]
     pkgs = " ".join([f"-e {p}" for p in sorted(projects)])
-    assert f"pip install --no-dependencies {pkgs}`" in captured.out
+    assert f"pip install --no-deps {pkgs}`" in captured.out
 
 
-def test_unidep_install_all_dry_run() -> None:
+def mock_uv_env(tmp_path: Path) -> dict[str, str]:
+    """Create a mock uv executable and return env with it in the PATH."""
+    mock_uv_path = tmp_path / ("uv.bat" if platform.system() == "Windows" else "uv")
+    if platform.system() == "Windows":
+        mock_uv_path.write_text("@echo off\necho Mock uv called %*")
+    else:
+        mock_uv_path.write_text("#!/bin/sh\necho 'Mock uv called' \"$@\"")
+    mock_uv_path.chmod(0o755)  # Make it executable
+
+    # Add tmp_path to the PATH environment variable
+    env = os.environ.copy()
+    env["PATH"] = f"{tmp_path}{os.pathsep}{env['PATH']}"
+    return env
+
+
+@pytest.mark.parametrize("with_uv", [True, False])
+def test_unidep_install_all_dry_run(tmp_path: Path, with_uv: bool) -> None:  # noqa: FBT001
     # Path to the requirements file
     requirements_path = REPO_ROOT / "example"
 
@@ -146,11 +163,13 @@ def test_unidep_install_all_dry_run() -> None:
             "--editable",
             "--directory",
             str(requirements_path),
+            *(["--no-uv"] if not with_uv else []),
         ],
         check=True,
         capture_output=True,
         text=True,
         encoding="utf-8",
+        env=mock_uv_env(tmp_path) if with_uv else None,
     )
 
     # Check the output
@@ -165,7 +184,10 @@ def test_unidep_install_all_dry_run() -> None:
     projects = [REPO_ROOT / "example" / p for p in EXAMPLE_PROJECTS]
     pkgs = " ".join([f"-e {p}" for p in sorted(projects)])
     assert "📦 Installing project with `" in result.stdout
-    assert f" -m pip install --no-dependencies {pkgs}" in result.stdout
+    if with_uv:
+        assert "uv pip install --python" in result.stdout
+    else:
+        assert f" -m pip install --no-deps {pkgs}" in result.stdout
 
 
 def test_unidep_conda() -> None:
@@ -256,6 +278,7 @@ def test_doubly_nested_project_folder_installable(
             "--dry-run",
             "--editable",
             "--no-dependencies",
+            "--no-uv",
             str(project4 / "requirements.yaml"),
         ],
         check=True,
@@ -269,7 +292,7 @@ def test_doubly_nested_project_folder_installable(
     p3 = str(tmp_path / "example" / "setuptools_project")
     p4 = str(tmp_path / "example" / "extra_projects" / "project4")
     pkgs = " ".join([f"-e {p}" for p in sorted((p1, p2, p3, p4))])
-    assert f"pip install --no-dependencies {pkgs}`" in result.stdout
+    assert f"pip install --no-deps {pkgs}`" in result.stdout
 
     p5 = str(tmp_path / "example" / "pyproject_toml_project")
     p6 = str(tmp_path / "example" / "hatch2_project")
@@ -281,6 +304,7 @@ def test_doubly_nested_project_folder_installable(
             "--dry-run",
             "--editable",
             "--no-dependencies",
+            "--no-uv",
             "--directory",
             str(example_folder),
             "--depth",
@@ -292,7 +316,7 @@ def test_doubly_nested_project_folder_installable(
         encoding="utf-8",
     )
     pkgs = " ".join([f"-e {p}" for p in sorted((p1, p2, p3, p4, p5, p6))])
-    assert f"pip install --no-dependencies {pkgs}`" in result.stdout
+    assert f"pip install --no-deps {pkgs}`" in result.stdout
 
     # Test depth 1 (should not install project4)
     result = subprocess.run(
@@ -302,6 +326,7 @@ def test_doubly_nested_project_folder_installable(
             "--dry-run",
             "--editable",
             "--no-dependencies",
+            "--no-uv",
             "--directory",
             str(example_folder),
             "--depth",
@@ -313,7 +338,7 @@ def test_doubly_nested_project_folder_installable(
         encoding="utf-8",
     )
     pkgs = " ".join([f"-e {p}" for p in sorted((p1, p2, p3, p5, p6))])
-    assert f"pip install --no-dependencies {pkgs}`" in result.stdout
+    assert f"pip install --no-deps {pkgs}`" in result.stdout
 
 
 def test_pip_compile_command(tmp_path: Path, capsys: pytest.CaptureFixture) -> None:

unidep/_cli.py

@@ -177,6 +177,7 @@ def _add_common_args(  # noqa: PLR0912, C901
     if "no-dependencies" in options:
         sub_parser.add_argument(
             "--no-dependencies",
+            "--no-deps",
             action="store_true",
             help=f"Skip installing dependencies from {_DEP_FILES}"
             " file(s) and only install local package(s). Useful after"
@@ -245,6 +246,13 @@ def _add_common_args(  # noqa: PLR0912, C901
             " environment. Assumes that the lock file contains all dependencies."
             " Must be used with `--conda-env-name` or `--conda-env-prefix`.",
         )
+    if "no-uv" in options:
+        sub_parser.add_argument(
+            "--no-uv",
+            action="store_true",
+            help="Disables the use of `uv` for pip install. By default, `uv` is used"
+            " if it is available in the PATH.",
+        )
 
 
 def _add_extra_flags(
@@ -373,6 +381,7 @@ def _parse_args() -> argparse.Namespace:
             "ignore-pin",
             "skip-dependency",
             "overwrite-pin",
+            "no-uv",
             "verbose",
         },
     )
@@ -416,6 +425,7 @@ def _parse_args() -> argparse.Namespace:
             "ignore-pin",
             "skip-dependency",
             "overwrite-pin",
+            "no-uv",
             "verbose",
         },
     )
@@ -833,15 +843,27 @@ def _python_executable(
     return str(python_executable)
 
 
+def _use_uv(no_uv: bool) -> bool:  # noqa: FBT001
+    """Check if the user wants to use the `uv` package."""
+    if no_uv:
+        return False
+    return shutil.which("uv") is not None
+
+
 def _pip_install_local(
     *folders: str | Path,
     editable: bool,
     dry_run: bool,
     python_executable: str,
     conda_run: list[str],
+    no_uv: bool,
     flags: list[str] | None = None,
 ) -> None:  # pragma: no cover
-    pip_command = [*conda_run, python_executable, "-m", "pip", "install"]
+    if _use_uv(no_uv):
+        pip_command = ["uv", "pip", "install", "--python", python_executable]
+    else:
+        pip_command = [*conda_run, python_executable, "-m", "pip", "install"]
+
     if flags:
         pip_command.extend(flags)
 
@@ -879,6 +901,7 @@ def _install_command(  # noqa: PLR0912, PLR0915
     ignore_pins: list[str] | None = None,
     overwrite_pins: list[str] | None = None,
     skip_dependencies: list[str] | None = None,
+    no_uv: bool = True,
     verbose: bool = False,
 ) -> None:
     """Install the dependencies of a single `requirements.yaml` or `pyproject.toml` file."""  # noqa: E501
@@ -952,15 +975,29 @@ def _install_command(  # noqa: PLR0912, PLR0915
         conda_env_prefix,
     )
     if env_spec.pip and not skip_pip:
-        conda_run = _maybe_conda_run(conda_executable, conda_env_name, conda_env_prefix)
-        pip_command = [
-            *conda_run,
-            python_executable,
-            "-m",
-            "pip",
-            "install",
-            *env_spec.pip,
-        ]
+        if _use_uv(no_uv):
+            pip_command = [
+                "uv",
+                "pip",
+                "install",
+                "--python",
+                python_executable,
+                *env_spec.pip,
+            ]
+        else:
+            conda_run = _maybe_conda_run(
+                conda_executable,
+                conda_env_name,
+                conda_env_prefix,
+            )
+            pip_command = [
+                *conda_run,
+                python_executable,
+                "-m",
+                "pip",
+                "install",
+                *env_spec.pip,
+            ]
         print(f"📦 Installing pip dependencies with `{' '.join(pip_command)}`\n")
         if not dry_run:  # pragma: no cover
             subprocess.run(pip_command, check=True)
@@ -992,7 +1029,7 @@ def _install_command(  # noqa: PLR0912, PLR0915
             if dep.resolve() not in installable_set
         ]
         if installable:
-            pip_flags = ["--no-dependencies"]  # we just ran pip/conda install, so skip
+            pip_flags = ["--no-deps"]  # we just ran pip/conda install, so skip
             if verbose:
                 pip_flags.append("--verbose")
             conda_run = _maybe_conda_run(
@@ -1006,6 +1043,7 @@ def _install_command(  # noqa: PLR0912, PLR0915
                 dry_run=dry_run,
                 python_executable=python_executable,
                 flags=pip_flags,
+                no_uv=no_uv,
                 conda_run=conda_run,
             )
 
@@ -1030,6 +1068,7 @@ def _install_all_command(
     ignore_pins: list[str] | None = None,
     overwrite_pins: list[str] | None = None,
     skip_dependencies: list[str] | None = None,
+    no_uv: bool = True,
     verbose: bool = False,
 ) -> None:  # pragma: no cover
     found_files = find_requirements_files(
@@ -1055,6 +1094,7 @@ def _install_all_command(
         ignore_pins=ignore_pins,
         overwrite_pins=overwrite_pins,
         skip_dependencies=skip_dependencies,
+        no_uv=no_uv,
         verbose=verbose,
     )
 
@@ -1471,6 +1511,7 @@ def main() -> None:
             ignore_pins=args.ignore_pin,
             skip_dependencies=args.skip_dependency,
             overwrite_pins=args.overwrite_pin,
+            no_uv=args.no_uv,
             verbose=args.verbose,
         )
     elif args.command == "install-all":
@@ -1492,6 +1533,7 @@ def main() -> None:
             ignore_pins=args.ignore_pin,
             skip_dependencies=args.skip_dependency,
             overwrite_pins=args.overwrite_pin,
+            no_uv=args.no_uv,
             verbose=args.verbose,
         )
     elif args.command == "conda-lock":  # pragma: no cover

unidep/_dependencies_parsing.py

@@ -624,7 +624,7 @@ def _extract_local_dependencies(  # noqa: PLR0912
                         f"⚠️ Installing a local dependency (`{abs_local.name}`) which"
                         " is not managed by unidep, this will skip all of its"
                         " dependencies, i.e., it will call `pip install` with"
-                        "  `--no-dependencies`. To properly manage this dependency,"
+                        "  `--no-deps`. To properly manage this dependency,"
                         " add a `requirements.yaml` or `pyproject.toml` file with"
                         " `[tool.unidep]` in its directory.",
                     )