pylock: read extras, dependency_groups, default_groups

Author: sbidoul

src/pip/_internal/models/pylock.py

@@ -69,7 +69,7 @@ def _toml_dict_factory(data: List[Tuple[str, Any]]) -> Dict[str, Any]:
     return {
         _toml_key(key): _toml_value(key, value)
         for key, value in data
-        if value is not None
+        if value is not None and value != []
     }
 
 
@@ -392,8 +392,9 @@ class Pylock:
     lock_version: Version
     environments: Optional[List[Marker]]
     requires_python: Optional[SpecifierSet]
-    # (not supported) extras: List[str] = []
-    # (not supported) dependency_groups: List[str] = []
+    extras: List[str]
+    dependency_groups: List[str]
+    default_groups: List[str]
     created_by: str
     packages: List[Package]
     # (not supported) tool: Optional[Dict[str, Any]]
@@ -419,6 +420,9 @@ def from_dict(cls, d: Dict[str, Any]) -> "Self":
         return cls(
             lock_version=_get_required_as(d, str, Version, "lock-version"),
             environments=_get_list_as(d, str, Marker, "environments"),
+            extras=_get_list(d, str, "extras") or [],
+            dependency_groups=_get_list(d, str, "dependency-groups") or [],
+            default_groups=_get_list(d, str, "default-groups") or [],
             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"),

src/pip/_internal/utils/pylock.py

@@ -114,6 +114,9 @@ def pylock_from_install_requirements(
         lock_version=Version("1.0"),
         environments=None,  # not supported
         requires_python=None,  # not supported
+        extras=[],  # not supported
+        dependency_groups=[],  # not supported
+        default_groups=[],  # not supported
         created_by="pip",
         packages=sorted(
             (

tests/unit/test_pylock.py

@@ -212,3 +212,18 @@ def test_pylock_invalid_environments() -> None:
         '    invalid_marker == "..."\n'
         "    ^"
     )
+
+
+def test_pylock_extras_and_groups() -> None:
+    data = {
+        "lock-version": "1.0",
+        "created-by": "pip",
+        "extras": ["feat1", "feat2"],
+        "dependency-groups": ["dev", "docs"],
+        "default-groups": ["dev"],
+        "packages": [],
+    }
+    pylock = Pylock.from_dict(data)
+    assert pylock.extras == ["feat1", "feat2"]
+    assert pylock.dependency_groups == ["dev", "docs"]
+    assert pylock.default_groups == ["dev"]