update

GrayHat12 · Jun 23, 2024 · 94be7fa · 94be7fa
1 parent 0646ee8
commit 94be7fa
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/python/pymhash/__init__.pyi b/python/pymhash/__init__.pyi
@@ -39,4 +39,7 @@ class ImageHash:
 
     def __str__(self) -> str: ...
 
-    def __hash__(self) -> int: ...
+    def __hash__(self) -> int: ...
+
+    @property
+    def hashes(self) -> List[OrientationHash]: ...
diff --git a/src/hash/image.rs b/src/hash/image.rs
@@ -12,6 +12,7 @@ use super::orientation::OrientationHash;
 #[pyclass]
 #[derive(Clone)]
 pub struct ImageHash {
+    #[pyo3(get)]
     pub hashes: Vec<OrientationHash>,
 }
 

diff --git a/test.py b/test.py
@@ -10,6 +10,14 @@
 print(p.metadata.hash.to_str())
 print(p.metadata.hash.to_str() == str(ImageHash.from_str(p.metadata.hash.to_str())))
 print(p.metadata.hash == ImageHash.from_str(p.metadata.hash.to_str()))
+hashes_str = [str(hash) for hash in p.metadata.hash.hashes]
+print(hashes_str)
+
+n = ImageHash()
+for v in hashes_str:
+    n.add_hash(OrientationHash.from_str(v))
+
+print(n == p.metadata.hash)
 
 assert isinstance(a, OrientationHash)
 assert isinstance(a.to_str(), str)