# -*- coding: UTF-8 -*-

# See COPYING for distribution information.

"""Virtual Mail Manager's Relocated class to manage relocated users."""

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 self._isNew and VMM.VirtualMailManager.accountExists(self._dbh,

                self._addr):

            raise VMMRE(_(u"There is already an account with address »%s«.") %\

                    self._addr, ERR.ACCOUNT_EXISTS)

        if self._isNew and 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)