pylock: handwoven constructors

We can remove them when we drop support for Python 3.9

Author: sbidoul

src/pip/_internal/models/pylock.py

@@ -256,16 +256,33 @@ class PylockUnsupportedVersionError(PylockValidationError):
     pass
 
 
-@dataclass(kw_only=True)
+@dataclass
 class PackageVcs:
     type: str
-    url: Optional[str] = None
-    path: Optional[str] = None
-    requested_revision: Optional[str] = None
+    url: Optional[str]  # = None
+    path: Optional[str]  # = None
+    requested_revision: Optional[str]  # = None
     commit_id: str
     subdirectory: Optional[str] = None
 
-    def __post_init__(self) -> None:
+    def __init__(
+        self,
+        *,
+        type: str,
+        commit_id: str,
+        url: Optional[str] = None,
+        path: Optional[str] = None,
+        requested_revision: Optional[str] = None,
+        subdirectory: Optional[str] = None,
+    ) -> None:
+        # In Python 3.10+ make dataclass kw_only=True and remove __init__
+        self.type = type
+        self.url = url
+        self.path = path
+        self.requested_revision = requested_revision
+        self.commit_id = commit_id
+        self.subdirectory = subdirectory
+        # __post_init__ in Python 3.10+
         _validate_path_url(self.path, self.url)
 
     @classmethod
@@ -280,12 +297,24 @@ def from_dict(cls, d: Dict[str, Any]) -> "Self":
         )
 
 
-@dataclass(kw_only=True)
+@dataclass
 class PackageDirectory:
     path: str
     editable: Optional[bool] = None
     subdirectory: Optional[str] = None
 
+    def __init__(
+        self,
+        *,
+        path: str,
+        editable: Optional[bool] = None,
+        subdirectory: Optional[str] = None,
+    ) -> None:
+        # In Python 3.10+ make dataclass kw_only=True and remove __init__
+        self.path = path
+        self.editable = editable
+        self.subdirectory = subdirectory
+
     @classmethod
     def from_dict(cls, d: Dict[str, Any]) -> "Self":
         return cls(
@@ -295,16 +324,33 @@ def from_dict(cls, d: Dict[str, Any]) -> "Self":
         )
 
 
-@dataclass(kw_only=True)
+@dataclass
 class PackageArchive:
-    url: Optional[str] = None
-    path: Optional[str] = None
-    size: Optional[int] = None
-    upload_time: Optional[datetime] = None
+    url: Optional[str]  # = None
+    path: Optional[str]  # = None
+    size: Optional[int]  # = None
+    upload_time: Optional[datetime]  # = None
     hashes: Dict[str, str]
     subdirectory: Optional[str] = None
 
-    def __post_init__(self) -> None:
+    def __init__(
+        self,
+        *,
+        hashes: Dict[str, str],
+        url: Optional[str] = None,
+        path: Optional[str] = None,
+        size: Optional[int] = None,
+        upload_time: Optional[datetime] = None,
+        subdirectory: Optional[str] = None,
+    ) -> None:
+        # In Python 3.10+ make dataclass kw_only=True and remove __init__
+        self.url = url
+        self.path = path
+        self.size = size
+        self.upload_time = upload_time
+        self.hashes = hashes
+        self.subdirectory = subdirectory
+        # __post_init__ in Python 3.10+
         _validate_path_url(self.path, self.url)
         _validate_hashes(self.hashes)
 
@@ -320,16 +366,33 @@ def from_dict(cls, d: Dict[str, Any]) -> "Self":
         )
 
 
-@dataclass(kw_only=True)
+@dataclass
 class PackageSdist:
     name: str
-    upload_time: Optional[datetime] = None
-    url: Optional[str] = None
-    path: Optional[str] = None
-    size: Optional[int] = None
+    upload_time: Optional[datetime]  # = None
+    url: Optional[str]  # = None
+    path: Optional[str]  # = None
+    size: Optional[int]  # = None
     hashes: Dict[str, str]
 
-    def __post_init__(self) -> None:
+    def __init__(
+        self,
+        *,
+        name: str,
+        hashes: Dict[str, str],
+        upload_time: Optional[datetime] = None,
+        url: Optional[str] = None,
+        path: Optional[str] = None,
+        size: Optional[int] = None,
+    ) -> None:
+        # In Python 3.10+ make dataclass kw_only=True and remove __init__
+        self.name = name
+        self.upload_time = upload_time
+        self.url = url
+        self.path = path
+        self.size = size
+        self.hashes = hashes
+        # __post_init__ in Python 3.10+
         _validate_path_url(self.path, self.url)
         _validate_hashes(self.hashes)
 
@@ -345,16 +408,33 @@ def from_dict(cls, d: Dict[str, Any]) -> "Self":
         )
 
 
-@dataclass(kw_only=True)
+@dataclass
 class PackageWheel:
     name: str
-    upload_time: Optional[datetime] = None
-    url: Optional[str] = None
-    path: Optional[str] = None
-    size: Optional[int] = None
+    upload_time: Optional[datetime]  # = None
+    url: Optional[str]  # = None
+    path: Optional[str]  # = None
+    size: Optional[int]  # = None
     hashes: Dict[str, str]
 
-    def __post_init__(self) -> None:
+    def __init__(
+        self,
+        *,
+        name: str,
+        hashes: Dict[str, str],
+        upload_time: Optional[datetime] = None,
+        url: Optional[str] = None,
+        path: Optional[str] = None,
+        size: Optional[int] = None,
+    ) -> None:
+        # In Python 3.10+ make dataclass kw_only=True and remove __init__
+        self.name = name
+        self.upload_time = upload_time
+        self.url = url
+        self.path = path
+        self.size = size
+        self.hashes = hashes
+        # __post_init__ in Python 3.10+
         _validate_path_url(self.path, self.url)
         _validate_hashes(self.hashes)
 
@@ -371,7 +451,7 @@ def from_dict(cls, d: Dict[str, Any]) -> "Self":
         return wheel
 
 
-@dataclass(kw_only=True)
+@dataclass
 class Package:
     name: str
     version: Optional[Version] = None
@@ -387,7 +467,38 @@ class Package:
     attestation_identities: Optional[List[Dict[str, Any]]] = None
     tool: Optional[Dict[str, Any]] = None
 
-    def __post_init__(self) -> None:
+    def __init__(
+        self,
+        *,
+        name: str,
+        version: Optional[Version] = None,
+        marker: Optional[Marker] = None,
+        requires_python: Optional[SpecifierSet] = None,
+        dependencies: Optional[List[Dict[str, Any]]] = None,
+        vcs: Optional[PackageVcs] = None,
+        directory: Optional[PackageDirectory] = None,
+        archive: Optional[PackageArchive] = None,
+        index: Optional[str] = None,
+        sdist: Optional[PackageSdist] = None,
+        wheels: Optional[List[PackageWheel]] = None,
+        attestation_identities: Optional[List[Dict[str, Any]]] = None,
+        tool: Optional[Dict[str, Any]] = None,
+    ) -> None:
+        # In Python 3.10+ make dataclass kw_only=True and remove __init__
+        self.name = name
+        self.version = version
+        self.marker = marker
+        self.requires_python = requires_python
+        self.dependencies = dependencies
+        self.vcs = vcs
+        self.directory = directory
+        self.archive = archive
+        self.index = index
+        self.sdist = sdist
+        self.wheels = wheels
+        self.attestation_identities = attestation_identities
+        self.tool = tool
+        # __post_init__ in Python 3.10+
         if self.sdist or self.wheels:
             if any([self.vcs, self.directory, self.archive]):
                 raise PylockValidationError(
@@ -422,19 +533,42 @@ def from_dict(cls, d: Dict[str, Any]) -> "Self":
         return package
 
 
-@dataclass(kw_only=True)
+@dataclass
 class Pylock:
     lock_version: Version
-    environments: Optional[List[Marker]] = None
-    requires_python: Optional[SpecifierSet] = None
-    extras: List[str] = dataclasses.field(default_factory=list)
-    dependency_groups: List[str] = dataclasses.field(default_factory=list)
-    default_groups: List[str] = dataclasses.field(default_factory=list)
+    environments: Optional[List[Marker]]  # = None
+    requires_python: Optional[SpecifierSet]  # = None
+    extras: List[str]  # = dataclasses.field(default_factory=list)
+    dependency_groups: List[str]  # = dataclasses.field(default_factory=list)
+    default_groups: List[str]  # = dataclasses.field(default_factory=list)
     created_by: str
     packages: List[Package]
     tool: Optional[Dict[str, Any]] = None
 
-    def __post_init__(self) -> None:
+    def __init__(
+        self,
+        *,
+        lock_version: Version,
+        created_by: str,
+        packages: List[Package],
+        environments: Optional[List[Marker]] = None,
+        requires_python: Optional[SpecifierSet] = None,
+        extras: Optional[List[str]] = None,
+        dependency_groups: Optional[List[str]] = None,
+        default_groups: Optional[List[str]] = None,
+        tool: Optional[Dict[str, Any]] = None,
+    ) -> None:
+        # In Python 3.10+ make dataclass kw_only=True and remove __init__
+        self.lock_version = lock_version
+        self.environments = environments
+        self.requires_python = requires_python
+        self.extras = extras or []
+        self.dependency_groups = dependency_groups or []
+        self.default_groups = default_groups or []
+        self.created_by = created_by
+        self.packages = packages
+        self.tool = tool
+        # __post_init__ in Python 3.10+
         if self.lock_version < Version("1") or self.lock_version >= Version("2"):
             raise PylockUnsupportedVersionError(
                 f"pylock version {self.lock_version} is not supported"