class DIF and Node: add necessary methods for use in dict() and set()

1 files changed, 18 insertions, 0 deletions

diff --git a/rumba/model.py b/rumba/model.py
index 5c22185..45d0473 100644
--- a/rumba/model.py
+++ b/rumba/model.py
@@ -91,6 +91,15 @@ class DIF:
         s = "DIF %s" % self.name
         return s
 
+    def __hash__(self):
+        return hash(self.name)
+
+    def __eq__(self, other):
+        return other != None and self.name == other.name
+
+    def __neq__(self, other):
+        return not (self == other)
+
     def add_member(self, node):
         self.members.append(node)
 
@@ -199,6 +208,15 @@ class Node:
 
         return s
 
+    def __hash__(self):
+        return hash(self.name)
+
+    def __eq__(self, other):
+        return other != None and self.name == other.name
+
+    def __neq__(self, other):
+        return not (self == other)
+
     def add_dif(self, dif):
         self.difs.append(dif)
         dif.add_member(self)