diff -r af813ede1e19 -r 14c0a092d7d2 VirtualMailManager/Relocated.py
--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/VirtualMailManager/Relocated.py Mon Sep 08 05:30:17 2008 +0000
@@ -0,0 +1,107 @@
+# -*- coding: UTF-8 -*-
+# Copyright 2008 VEB IT
+# See COPYING for distribution information.
+# $Id$
+
+"""Virtual Mail Manager's Relocated class to manage relocated users."""
+
+from constants.VERSION import VERSION
+
+__author__ = 'Pascal Volk
'
+__version__ = VERSION
+__revision__ = 'rev '+'$Rev$'.split()[1]
+__date__ = '$Date$'.split()[1]
+
+from Exceptions import VMMRelocatedException as VMMRE
+from Domain import Domain
+from EmailAddress import EmailAddress
+import constants.ERROR as ERR
+import VirtualMailManager as VMM
+
+class Relocated:
+ """Class to manage e-mail addresses of relocated users."""
+ def __init__(self, dbh, address, destination=None):
+ if isinstance(address, EmailAddress):
+ self._addr = address
+ else:
+ raise TypeError("Argument 'address' is not an EmailAddress")
+ if destination is None:
+ self._dest = None
+ elif isinstance(destination, EmailAddress):
+ self._dest = destination
+ else:
+ raise TypeError("Argument 'destination' is not an EmailAddress")
+ if address == destination:
+ raise VMMRE(_(u"Address and destination are identical."),
+ ERR.RELOCATED_ADDR_DEST_IDENTICAL)
+ self._dbh = dbh
+ self._gid = 0
+ self._isNew = False
+ self._setAddr()
+ self._exists()
+ if VMM.VirtualMailManager.accountExists(self._dbh, self._addr):
+ raise VMMRE(_(u"There is already an account with address »%s«.") %\
+ self._addr, ERR.ACCOUNT_EXISTS)
+ if VMM.VirtualMailManager.aliasExists(self._dbh, self._addr):
+ raise VMMRE(
+ _(u"There is already an alias with the address »%s«.") %\
+ self._addr, ERR.ALIAS_EXISTS)
+
+ def _exists(self):
+ dbc = self._dbh.cursor()
+ dbc.execute("SELECT gid FROM relocated WHERE gid = %s AND address = %s",
+ self._gid, self._addr._localpart)
+ gid = dbc.fetchone()
+ dbc.close()
+ if gid is None:
+ self._isNew = True
+
+ def _setAddr(self):
+ dom = Domain(self._dbh, self._addr._domainname)
+ self._gid = dom.getID()
+ if self._gid == 0:
+ raise VMMRE(_(u"The domain »%s« doesn't exist yet.") %\
+ self._addr._domainname, ERR.NO_SUCH_DOMAIN)
+
+ def save(self):
+ if self._dest is None:
+ raise VMMRE(_(u"No destination address for relocated user denoted."),
+ ERR.RELOCATED_MISSING_DEST)
+ if self._isNew:
+ dbc = self._dbh.cursor()
+ dbc.execute("INSERT INTO relocated VALUES (%s, %s, %s)",
+ self._gid, self._addr._localpart, str(self._dest))
+ self._dbh.commit()
+ dbc.close()
+ else:
+ raise VMMRE(
+ _(u"The relocated user »%s« already exists.") % self._addr,
+ ERR.RELOCATED_EXISTS)
+
+ def getInfo(self):
+ dbc = self._dbh.cursor()
+ dbc.execute('SELECT destination FROM relocated WHERE gid=%s\
+ AND address=%s',
+ self._gid, self._addr._localpart)
+ destination = dbc.fetchone()
+ dbc.close()
+ if destination is not None:
+ return destination[0]
+ else:
+ raise VMMRE(
+ _(u"The relocated user »%s« doesn't exists.") % self._addr,
+ ERR.NO_SUCH_RELOCATED)
+
+ def delete(self):
+ dbc = self._dbh.cursor()
+ dbc.execute("DELETE FROM relocated WHERE gid = %s AND address = %s",
+ self._gid, self._addr._localpart)
+ rowcount = dbc.rowcount
+ dbc.close()
+ if rowcount > 0:
+ self._dbh.commit()
+ else:
+ raise VMMRE(
+ _(u"The relocated user »%s« doesn't exists.") % self._addr,
+ ERR.NO_SUCH_RELOCATED)
+