From 23b46068be8b698c55ab854eb5c2de2a3a14d453 Mon Sep 17 00:00:00 2001 From: Vincenzo Maffione Date: Sun, 19 Mar 2017 10:22:36 +0100 Subject: class DIF and Node: add necessary methods for use in dict() and set() --- rumba/model.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) 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) -- cgit v1.2.3