VirtualMailManager/Transport.py
changeset 110 cb8b2f6a5fca
parent 102 485d3f7d6981
child 128 cf8116625866
--- a/VirtualMailManager/Transport.py	Fri Aug 14 06:16:01 2009 +0000
+++ b/VirtualMailManager/Transport.py	Fri Aug 14 06:40:55 2009 +0000
@@ -10,8 +10,9 @@
 from Exceptions import VMMTransportException
 import constants.ERROR as ERR
 
-class Transport:
-    """A wrapper class thats provide access to the transport table"""
+class Transport(object):
+    """A wrapper class that provides access to the transport table"""
+    __slots__ = ('__id', '__transport', '_dbh')
     def __init__(self, dbh, tid=None, transport=None):
         """Creates a new Transport instance.
 
@@ -38,6 +39,19 @@
             self.__transport = transport
             self._loadByName()
 
+    def __eq__(self, other):
+        if isinstance(other, self.__class__):
+            return self.__id == other.getID()
+        return NotImplemented
+
+    def __ne__(self, other):
+        if isinstance(other, self.__class__):
+            return self.__id != other.getID()
+        return NotImplemented
+
+    def __str__(self):
+        return self.__transport
+
     def _loadByID(self):
         dbc = self._dbh.cursor()
         dbc.execute('SELECT transport FROM transport WHERE tid = %s', self.__id)