--- a/VirtualMailManager/Transport.py Mon Nov 07 03:22:15 2011 +0000
+++ /dev/null Thu Jan 01 00:00:00 1970 +0000
@@ -1,90 +0,0 @@
-# -*- coding: UTF-8 -*-
-# Copyright (c) 2008 - 2010, Pascal Volk
-# See COPYING for distribution information.
-
-"""Virtual Mail Manager's Transport class to manage the transport for
-domains and accounts."""
-
-from __main__ import ERR
-from Exceptions import VMMTransportException
-
-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.
-
- Either tid or transport must be specified.
-
- Keyword arguments:
- dbh -- a pyPgSQL.PgSQL.connection
- tid -- the id of a transport (long)
- transport -- the value of the transport (str)
- """
- self._dbh = dbh
- if tid is None and transport is None:
- raise VMMTransportException(
- _('Either tid or transport must be specified.'),
- ERR.TRANSPORT_INIT)
- elif tid is not None:
- try:
- self.__id = long(tid)
- except ValueError:
- raise VMMTransportException(_('tid must be an int/long.'),
- ERR.TRANSPORT_INIT)
- self._loadByID()
- else:
- 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)
- result = dbc.fetchone()
- dbc.close()
- if result is not None:
- self.__transport = result[0]
- else:
- raise VMMTransportException(_('Unknown tid specified.'),
- ERR.UNKNOWN_TRANSPORT_ID)
-
- def _loadByName(self):
- dbc = self._dbh.cursor()
- dbc.execute('SELECT tid FROM transport WHERE transport = %s',
- self.__transport)
- result = dbc.fetchone()
- dbc.close()
- if result is not None:
- self.__id = result[0]
- else:
- self._save()
-
- def _save(self):
- dbc = self._dbh.cursor()
- dbc.execute("SELECT nextval('transport_id')")
- self.__id = dbc.fetchone()[0]
- dbc.execute('INSERT INTO transport (tid, transport) VALUES (%s, %s)',
- self.__id, self.__transport)
- self._dbh.commit()
- dbc.close()
-
- def getID(self):
- """Returns the unique ID of the transport."""
- return self.__id
-
- def getTransport(self):
- """Returns the value of transport, ex: 'dovecot:'"""
- return self.__transport