Class PatchData

vulnerabilities/importer.py:265–303

@dataclasses.dataclass(eq=True)
@functools.total_ordering
class PatchData:
    patch_url: Optional[str] = ""
    patch_text: Optional[str] = ""
    patch_checksum: Optional[str] = dataclasses.field(init=False, default=None)

    def __post_init__(self):
        if not self.patch_url and not self.patch_text:
            raise ValueError("A patch must include either patch_url or patch_text")

        if self.patch_text:
            self.patch_checksum = compute_patch_checksum(self.patch_text)

    def __lt__(self, other):
        if not isinstance(other, PatchData):
            return NotImplemented
        return self._cmp_key() < other._cmp_key()

    def _cmp_key(self):
        return (
            self.patch_url or "",
            self.patch_text or "",
            self.patch_checksum or "",
        )

    def to_dict(self) -> dict:
        """Return a normalized dictionary representation of the commit."""
        return {
            "patch_url": self.patch_url,
            "patch_text": self.patch_text,
            "patch_checksum": self.patch_checksum,
        }

    @classmethod
    def from_dict(cls, data: dict):
        """Create a PatchData instance from a dictionary."""
        return cls(
            patch_url=data.get("patch_url"),
            patch_text=data.get("patch_text"),
        )


class UnMergeablePackageError(Exception):