fix: add expected lower-than comparators for OrganizationalEntity a…

…nd `VulnerabilityCredits` (#248) Signed-off-by: Jan Kowalleck <jan.kowalleck@gmail.com>
CycloneDX · Jun 15, 2022 · 0046ee1 · 0046ee1
1 parent ddd0144
commit 0046ee1
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/cyclonedx/model/__init__.py b/cyclonedx/model/__init__.py
@@ -1068,6 +1068,11 @@ def __eq__(self, other: object) -> bool:
             return hash(other) == hash(self)
         return False
 
+    def __lt__(self, other: Any) -> bool:
+        if isinstance(other, OrganizationalEntity):
+            return hash(self) < hash(other)
+        return NotImplemented
+
     def __hash__(self) -> int:
         return hash((self.name, tuple(self.url), tuple(self.contact)))
 

diff --git a/cyclonedx/model/vulnerability.py b/cyclonedx/model/vulnerability.py
@@ -755,6 +755,11 @@ def __eq__(self, other: object) -> bool:
             return hash(other) == hash(self)
         return False
 
+    def __lt__(self, other: Any) -> bool:
+        if isinstance(other, VulnerabilityCredits):
+            return hash(self) < hash(other)
+        return NotImplemented
+
     def __hash__(self) -> int:
         return hash((tuple(self.organizations), tuple(self.individuals)))