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

# See COPYING for distribution information.

"""

    VirtualMailManager.transport

    ~~~~~~~~~~~~~~~~~~~~~~~~~~~~

    Virtual Mail Manager's Transport class to manage the transport for

    domains and accounts.

"""

from VirtualMailManager.constants import UNKNOWN_TRANSPORT_ID

from VirtualMailManager.errors import TransportError

from VirtualMailManager.pycompat import any

_ = lambda msg: msg

class Transport(object):

    """A wrapper class that provides access to the transport table"""

    __slots__ = ('_tid', '_transport', '_dbh')

    def __init__(self, dbh, tid=None, transport=None):

        """Creates a new Transport instance.

        Either tid or transport must be specified. When both arguments

        are given, tid will be used.

        Keyword arguments:

        dbh -- a pyPgSQL.PgSQL.connection

        tid -- the id of a transport (int/long)

        transport -- the value of the transport (str)

        """

        self._dbh = dbh

        assert any((tid, transport))

        if tid:

            assert not isinstance(tid, bool) and isinstance(tid, (int, long))

            self._tid = tid

            self._load_by_id()

        else:

            assert isinstance(transport, basestring)

            self._transport = transport

            self._load_by_name()

    @property

    def tid(self):

        """The transport's unique ID."""

        return self._tid

    @property

    def transport(self):

        """The transport's value, ex: 'dovecot:'"""

        return self._transport

    def __eq__(self, other):

        if isinstance(other, self.__class__):

            return self._tid == other._tid

        return NotImplemented

    def __ne__(self, other):

        if isinstance(other, self.__class__):

            return self._tid != other._tid

        return NotImplemented

    def __str__(self):

        return self._transport

    def _load_by_id(self):

        """load a transport by its id from the database"""

        dbc = self._dbh.cursor()

        dbc.execute('SELECT transport FROM transport WHERE tid = %s',

                    (self._tid,))

        result = dbc.fetchone()

        dbc.close()

        if result:

            self._transport = result[0]

        else:

            raise TransportError(_(u'Unknown transport id specified.'),

                                 UNKNOWN_TRANSPORT_ID)

    def _load_by_name(self):

        """Load a transport by its transport name from the database."""

        dbc = self._dbh.cursor()

        dbc.execute('SELECT tid FROM transport WHERE transport = %s',

                    (self._transport,))

        result = dbc.fetchone()

        dbc.close()

        if result:

            self._tid = result[0]

        else:

            self._save()

    def _save(self):

        """Save the new transport in the database."""

        dbc = self._dbh.cursor()

        dbc.execute("SELECT nextval('transport_id')")

        self._tid = dbc.fetchone()[0]

        dbc.execute('INSERT INTO transport VALUES (%s, %s)', (self._tid,

                    self._transport))

        self._dbh.commit()

        dbc.close()

del _