model: add missing __repr__ methods

1 files changed, 10 insertions, 0 deletions

diff --git a/rumba/model.py b/rumba/model.py
index a043384..d1d9156 100644
--- a/rumba/model.py
+++ b/rumba/model.py
@@ -50,6 +50,9 @@ class Interface:
         self.name = name
         self.ip = ip
 
+    def __repr__(self):
+        return self.name
+
 # Represents a link in the physical graph
 #
 # @name [string] Link name
@@ -58,6 +61,9 @@ class Link:
     def __init__(self, name):
         self.name = name
 
+    def __repr__(self):
+        return self.name
+
 # Represents a point-to-point link in the physical graph
 #
 # @name [string] DIF name
@@ -76,6 +82,10 @@ class P2PLink(Link):
             int_b = Interface()
         self.int_b = int_b
 
+    def __repr__(self):
+        return '%s:%s--%s:%s' % (self.node_a.name, self.int_a,
+                                 self.node_b.name, self.int_b)
+
 # Base class for DIFs
 #
 # @name [string] DIF name