Add support for uv.toml (#538)

Author: nathanjmcdougall

src/usethis/_config_file.py

@@ -8,6 +8,7 @@
 from usethis._integrations.file.pyproject_toml.io_ import PyprojectTOMLManager
 from usethis._integrations.file.setup_cfg.io_ import SetupCFGManager
 from usethis._integrations.file.toml.io_ import TOMLFileManager
+from usethis._integrations.uv.toml import UVTOMLManager
 
 if TYPE_CHECKING:
     from collections.abc import Iterator
@@ -26,6 +27,7 @@ def files_manager() -> Iterator[None]:
         PytestINIManager(),
         RuffTOMLManager(),
         ToxINIManager(),
+        UVTOMLManager(),
     ):
         yield
 

src/usethis/_integrations/uv/deps.py

@@ -10,6 +10,7 @@
 from usethis._integrations.file.pyproject_toml.io_ import PyprojectTOMLManager
 from usethis._integrations.uv.call import call_uv_subprocess
 from usethis._integrations.uv.errors import UVDepGroupError, UVSubprocessFailedError
+from usethis._integrations.uv.toml import UVTOMLManager
 
 
 class Dependency(BaseModel):
@@ -86,14 +87,20 @@ def register_default_group(group: str) -> None:
 
 
 def add_default_groups(groups: list[str]) -> None:
-    PyprojectTOMLManager().extend_list(
-        keys=["tool", "uv", "default-groups"], values=groups
-    )
+    if UVTOMLManager().path.exists():
+        UVTOMLManager().extend_list(keys=["default-groups"], values=groups)
+    else:
+        PyprojectTOMLManager().extend_list(
+            keys=["tool", "uv", "default-groups"], values=groups
+        )
 
 
 def get_default_groups() -> list[str]:
     try:
-        default_groups = PyprojectTOMLManager()[["tool", "uv", "default-groups"]]
+        if UVTOMLManager().path.exists():
+            default_groups = UVTOMLManager()[["default-groups"]]
+        else:
+            default_groups = PyprojectTOMLManager()[["tool", "uv", "default-groups"]]
         if not isinstance(default_groups, list):
             default_groups = []
     except KeyError:

src/usethis/_integrations/uv/toml.py

@@ -0,0 +1,11 @@
+from pathlib import Path
+
+from usethis._integrations.file.toml.io_ import TOMLFileManager
+
+
+class UVTOMLManager(TOMLFileManager):
+    """Class to manage the uv.toml file."""
+
+    @property
+    def relative_path(self) -> Path:
+        return Path("uv.toml")

tests/usethis/_core/test_core_tool.py

@@ -35,6 +35,7 @@
     get_deps_from_group,
     is_dep_satisfied_in,
 )
+from usethis._integrations.uv.toml import UVTOMLManager
 from usethis._test import change_cwd
 from usethis._tool import ALL_TOOLS, PyprojectTOMLTool, PytestTool, RuffTool
 
@@ -1991,6 +1992,19 @@ def test_registers_test_group(self, tmp_path: Path):
                 ]
                 assert "test" in default_groups
 
+        @pytest.mark.usefixtures("_vary_network_conn")
+        def test_registers_test_group_uv_toml(self, tmp_path: Path):
+            with change_cwd(tmp_path), files_manager():
+                # Arrange
+                (tmp_path / "uv.toml").touch()
+
+                # Act
+                use_pytest()
+
+                # Assert
+                default_groups = UVTOMLManager()[["default-groups"]]
+                assert "test" in default_groups
+
         @pytest.mark.usefixtures("_vary_network_conn")
         def test_ruff_integration(self, uv_init_dir: Path):
             with change_cwd(uv_init_dir), files_manager():

tests/usethis/_integrations/uv/test_deps.py

@@ -7,9 +7,11 @@
 import usethis._integrations.uv
 import usethis._integrations.uv.deps
 from usethis._config import usethis_config
+from usethis._config_file import files_manager
 from usethis._integrations.file.pyproject_toml.io_ import PyprojectTOMLManager
 from usethis._integrations.uv.deps import (
     Dependency,
+    add_default_groups,
     add_deps_to_group,
     get_default_groups,
     get_dep_groups,
@@ -20,6 +22,7 @@
     remove_deps_from_group,
 )
 from usethis._integrations.uv.errors import UVDepGroupError, UVSubprocessFailedError
+from usethis._integrations.uv.toml import UVTOMLManager
 from usethis._test import change_cwd
 
 
@@ -651,6 +654,25 @@ def test_dev_not_added_if_missing(self, tmp_path: Path):
             assert set(default_groups) == {"test"}
 
 
+class TestAddDefaultGroups:
+    def test_uv_toml(self, tmp_path: Path):
+        # Arrange
+        (tmp_path / "uv.toml").touch()
+
+        # Act
+        with change_cwd(tmp_path), files_manager():
+            add_default_groups(["test"])
+
+        # Assert
+        with change_cwd(tmp_path), UVTOMLManager():
+            assert (
+                (tmp_path / "uv.toml").read_text()
+                == """\
+default-groups = ["test"]
+"""
+            )
+
+
 class TestGetDefaultGroups:
     def test_empty_pyproject_toml(self, tmp_path: Path):
         # Arrange
@@ -676,3 +698,36 @@ def test_invalid_default_groups(self, tmp_path: Path):
 
             # Assert
             assert result == []
+
+    def test_uv_toml(self, tmp_path: Path):
+        # Arrange
+        (tmp_path / "uv.toml").write_text("""\
+default-groups = ["test"]
+""")
+        # Even if the pyproject.toml disagrees!
+        (tmp_path / "pyproject.toml").write_text("""\
+[tool.uv]
+default-groups = ["doc"]
+""")
+
+        with change_cwd(tmp_path), files_manager():
+            # Act
+            result = get_default_groups()
+
+            # Assert
+            assert result == ["test"]
+
+    def test_uv_toml_empty(self, tmp_path: Path):
+        # Arrange
+        (tmp_path / "uv.toml").touch()
+        (tmp_path / "pyproject.toml").write_text("""\
+[tool.uv]
+default-groups = ["doc"]
+""")
+
+        with change_cwd(tmp_path), files_manager():
+            # Act
+            result = get_default_groups()
+
+            # Assert
+            assert result == []