add __hash__ to FerveoVariant

nucypher · Jul 14, 2023 · 06321d7 · 06321d7
1 parent cea467e
commit 06321d7
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/ferveo-python/ferveo/__init__.pyi b/ferveo-python/ferveo/__init__.pyi
@@ -201,6 +201,9 @@ class FerveoVariant:
     def __eq__(self, other: object) -> bool:
         ...
 
+    def __hash__(self) -> int:
+        ...
+
 
 def encrypt(message: bytes, aad: bytes, dkg_public_key: DkgPublicKey) -> Ciphertext:
     ...

diff --git a/ferveo/src/bindings_python.rs b/ferveo/src/bindings_python.rs
@@ -295,6 +295,14 @@ impl FerveoVariant {
     fn __richcmp__(&self, other: &Self, op: CompareOp) -> PyResult<bool> {
         richcmp(self, other, op)
     }
+
+    fn __hash__(&self) -> PyResult<isize> {
+        let bytes = self
+            .0
+            .to_bytes()
+            .map_err(|err| FerveoPythonError::Other(err.to_string()))?;
+        hash("FerveoVariant", &bytes)
+    }
 }
 
 impl fmt::Display for FerveoVariant {