Merge pull request #15 from WarpCloud/graph_type_hash

- add __hash__ and __eq__ to GraphElement and Path

pystellardb/graph_types.py

@@ -8,6 +8,7 @@
 import json
 import logging
 import binascii
+from typing import cast
 
 _logger = logging.getLogger(__name__)
 
@@ -46,6 +47,21 @@ def setRowKeyHexString(self, rowkey):
 
     def getRowKeyHexString(self):
         return self._rowKeyHexString
+
+    def __hash__(self):
+        if self._rowKeyHexString:
+            return hash(self._rowKeyHexString)
+        else:
+            return hash(self._fields)
+
+    def __eq__(self, value):
+        if isinstance(value, GraphElement):
+            if self._rowKeyHexString:
+                return self._rowKeyHexString == value._rowKeyHexString
+            else:
+                return self._fields == value._fields
+        else:
+            return False
 
 
 class Vertex(GraphElement):
@@ -292,6 +308,22 @@ def length(self):
 
     def __str__(self):
         return str([str(entry) for entry in self._elems])
+
+    def __hash__(self):
+        rks = []
+        for elem in self._elems:
+            rks.append(cast(GraphElement, elem).getRowKeyHexString())
+
+        return hash(tuple(rks))
+
+    def __eq__(self, value):
+        if isinstance(value, Path):
+            if self.length() == value.length():
+                return self.getElements() == value.getElements()
+            else:
+                return False
+        else:
+            return False
 
     @staticmethod
     def parsePathFromJson(schema, json_str):