+
−# -*- coding: UTF-8 -*-
+
−# Copyright (c) 2008 - 2011, Pascal Volk
+
−# See COPYING for distribution information.
+
−"""
+
−    VirtualMailManager.aliasdomain
+
−    ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
−
+
−    Virtual Mail Manager's AliasDomain class to manage alias domains.
+
−"""
+
−
+
−from VirtualMailManager.domain import Domain, check_domainname
+
−from VirtualMailManager.constants import \
+
−     ALIASDOMAIN_EXISTS, ALIASDOMAIN_ISDOMAIN, ALIASDOMAIN_NO_DOMDEST, \
+
−     NO_SUCH_ALIASDOMAIN, NO_SUCH_DOMAIN
+
−from VirtualMailManager.errors import AliasDomainError as ADErr
+
−
+
−
+
−_ = lambda msg: msg
+
−
+
−
+
−class AliasDomain(object):
+
−    """Class to manage e-mail alias domains."""
+
−    __slots__ = ('_gid', '_name', '_domain', '_dbh')
+
−
+
−    def __init__(self, dbh, domainname):
+
−        """Creates a new AliasDomain instance.
+
−
+
−        Arguments:
+
−
+
−        `dbh` : pyPgSQL.PgSQL.Connection
+
−          a database connection for the database access
+
−        `domainname` : basestring
+
−          the name of the AliasDomain"""
+
−        self._dbh = dbh
+
−        self._name = check_domainname(domainname)
+
−        self._gid = 0
+
−        self._domain = None
+
−        self._load()
+
−
+
−    def _load(self):
+
−        """Loads the AliasDomain's GID from the database and checks if the
+
−        domain name is marked as primary."""
+
−        dbc = self._dbh.cursor()
+
−        dbc.execute('SELECT gid, is_primary FROM domain_name WHERE '
+
−                    'domainname = %s', (self._name,))
+
−        result = dbc.fetchone()
+
−        dbc.close()
+
−        if result:
+
−            if result[1]:
+
−                raise ADErr(_(u"The domain '%s' is a primary domain.") %
+
−                            self._name, ALIASDOMAIN_ISDOMAIN)
+
−            self._gid = result[0]
+
−
+
−    def set_destination(self, dest_domain):
+
−        """Set the destination of a new AliasDomain or updates the
+
−        destination of an existing AliasDomain.
+
−
+
−        Argument:
+
−
+
−        `dest_domain` : VirtualMailManager.Domain.Domain
+
−          the AliasDomain's destination domain
+
−        """
+
−        assert isinstance(dest_domain, Domain)
+
−        self._domain = dest_domain
+
−
+
−    def save(self):
+
−        """Stores information about the new AliasDomain in the database."""
+
−        if self._gid > 0:
+
−            raise ADErr(_(u"The alias domain '%s' already exists.") %
+
−                        self._name, ALIASDOMAIN_EXISTS)
+
−        if not self._domain:
+
−            raise ADErr(_(u'No destination domain set for the alias domain.'),
+
−                        ALIASDOMAIN_NO_DOMDEST)
+
−        if self._domain.gid < 1:
+
−            raise ADErr(_(u"The target domain '%s' does not exist.") %
+
−                        self._domain.name, NO_SUCH_DOMAIN)
+
−        dbc = self._dbh.cursor()
+
−        dbc.execute('INSERT INTO domain_name (domainname, gid, is_primary) '
+
−                    'VALUES (%s, %s, FALSE)', (self._name, self._domain.gid))
+
−        self._dbh.commit()
+
−        dbc.close()
+
−        self._gid = self._domain.gid
+
−
+
−    def info(self):
+
−        """Returns a dict (keys: "alias" and "domain") with the names of the
+
−        AliasDomain and its primary domain."""
+
−        if self._gid < 1:
+
−            raise ADErr(_(u"The alias domain '%s' does not exist.") %
+
−                        self._name, NO_SUCH_ALIASDOMAIN)
+
−        dbc = self._dbh.cursor()
+
−        dbc.execute('SELECT domainname FROM domain_name WHERE gid = %s AND '
+
−                    'is_primary', (self._gid,))
+
−        domain = dbc.fetchone()
+
−        dbc.close()
+
−        if domain:
+
−            return {'alias': self._name, 'domain': domain[0]}
+
−        else:  # an almost unlikely case, isn't it?
+
−            raise ADErr(_(u'There is no primary domain for the alias domain '
+
−                          u"'%s'.") % self._name, NO_SUCH_DOMAIN)
+
−
+
−    def switch(self):
+
−        """Switch the destination of the AliasDomain to the new destination,
+
−        set with the method `set_destination()`.
+
−        """
+
−        if not self._domain:
+
−            raise ADErr(_(u'No destination domain set for the alias domain.'),
+
−                        ALIASDOMAIN_NO_DOMDEST)
+
−        if self._domain.gid < 1:
+
−            raise ADErr(_(u"The target domain '%s' does not exist.") %
+
−                        self._domain.name, NO_SUCH_DOMAIN)
+
−        if self._gid < 1:
+
−            raise ADErr(_(u"The alias domain '%s' does not exist.") %
+
−                        self._name, NO_SUCH_ALIASDOMAIN)
+
−        if self._gid == self._domain.gid:
+
−            raise ADErr(_(u"The alias domain '%(alias)s' is already assigned "
+
−                          u"to the domain '%(domain)s'.") %
+
−                        {'alias': self._name, 'domain': self._domain.name},
+
−                        ALIASDOMAIN_EXISTS)
+
−        dbc = self._dbh.cursor()
+
−        dbc.execute('UPDATE domain_name SET gid = %s WHERE gid = %s AND '
+
−                    'domainname = %s AND NOT is_primary', (self._domain.gid,
+
−                    self._gid, self._name))
+
−        self._dbh.commit()
+
−        dbc.close()
+
−        self._gid = self._domain.gid
+
−
+
−    def delete(self):
+
−        """Delete the AliasDomain's record form the database.
+
−
+
−        Raises an AliasDomainError if the AliasDomain doesn't exist.
+
−        """
+
−        if self._gid < 1:
+
−            raise ADErr(_(u"The alias domain '%s' does not exist.") %
+
−                        self._name, NO_SUCH_ALIASDOMAIN)
+
−        dbc = self._dbh.cursor()
+
−        dbc.execute('DELETE FROM domain_name WHERE domainname = %s AND NOT '
+
−                    'is_primary', (self._name,))
+
−        if dbc.rowcount > 0:
+
−            self._dbh.commit()
+
−            self._gid = 0
+
−        dbc.close()
+
−
+
−del _