pylock: read tool sections

Author: sbidoul

src/pip/_internal/models/pylock.py

@@ -343,18 +343,18 @@ def from_dict(cls, d: Dict[str, Any]) -> "Self":
 @dataclass
 class Package:
     name: str
-    version: Optional[Version] = None
-    marker: Optional[Marker] = None
-    requires_python: Optional[SpecifierSet] = None
+    version: Optional[Version]
+    marker: Optional[Marker]
+    requires_python: Optional[SpecifierSet]
     # (not supported) dependencies
-    vcs: Optional[PackageVcs] = None
-    directory: Optional[PackageDirectory] = None
-    archive: Optional[PackageArchive] = None
+    vcs: Optional[PackageVcs]
+    directory: Optional[PackageDirectory]
+    archive: Optional[PackageArchive]
     # (not supported) index: Optional[str]
-    sdist: Optional[PackageSdist] = None
-    wheels: Optional[List[PackageWheel]] = None
+    sdist: Optional[PackageSdist]
+    wheels: Optional[List[PackageWheel]]
     # (not supported) attestation_identities: Optional[List[Dict[str, Any]]]
-    # (not supported) tool: Optional[Dict[str, Any]]
+    tool: Optional[Dict[str, Any]]
 
     def __post_init__(self) -> None:
         if self.sdist or self.wheels:
@@ -383,6 +383,7 @@ def from_dict(cls, d: Dict[str, Any]) -> "Self":
             archive=_get_object(d, PackageArchive, "archive"),
             sdist=_get_object(d, PackageSdist, "sdist"),
             wheels=_get_list_of_objects(d, PackageWheel, "wheels"),
+            tool=_get(d, dict, "tool"),
         )
         return package
 
@@ -397,7 +398,7 @@ class Pylock:
     default_groups: List[str]
     created_by: str
     packages: List[Package]
-    # (not supported) tool: Optional[Dict[str, Any]]
+    tool: Optional[Dict[str, Any]]
 
     def __post_init__(self) -> None:
         if self.lock_version < Version("1") or self.lock_version >= Version("2"):
@@ -426,4 +427,5 @@ def from_dict(cls, d: Dict[str, Any]) -> "Self":
             created_by=_get_required(d, str, "created-by"),
             requires_python=_get_as(d, str, SpecifierSet, "requires-python"),
             packages=_get_required_list_of_objects(d, Package, "packages"),
+            tool=_get(d, dict, "tool"),
         )

src/pip/_internal/utils/pylock.py

@@ -104,6 +104,7 @@ def _pylock_package_from_install_requirement(
         archive=package_archive,
         sdist=package_sdist,
         wheels=package_wheels,
+        tool=None,
     )
 
 
@@ -125,4 +126,5 @@ def pylock_from_install_requirements(
             ),
             key=lambda p: p.name,
         ),
+        tool=None,
     )

tests/unit/test_pylock.py

@@ -227,3 +227,26 @@ def test_pylock_extras_and_groups() -> None:
     assert pylock.extras == ["feat1", "feat2"]
     assert pylock.dependency_groups == ["dev", "docs"]
     assert pylock.default_groups == ["dev"]
+
+
+def test_pylock_tool() -> None:
+    data = {
+        "lock-version": "1.0",
+        "created-by": "pip",
+        "packages": [
+            {
+                "name": "example",
+                "sdist": {
+                    "name": "example-1.0.tar.gz",
+                    "path": "./example-1.0.tar.gz",
+                    "hashes": {"sha256": "f" * 40},
+                },
+                "tool": {"pip": {"foo": "bar"}},
+            }
+        ],
+        "tool": {"pip": {"version": "25.2"}},
+    }
+    pylock = Pylock.from_dict(data)
+    assert pylock.tool == {"pip": {"version": "25.2"}}
+    package = pylock.packages[0]
+    assert package.tool == {"pip": {"foo": "bar"}}