+
−# -*- coding: UTF-8 -*-
+
−# Copyright (c) 2011, Pascal Volk
+
−# See COPYING for distribution information.
+
−"""
+
−    VirtualMailManager.quotalimit
+
−    ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
−
+
−    Virtual Mail Manager's QuotaLimit class to manage quota limits
+
−    for domains and accounts.
+
−"""
+
−
+
−from VirtualMailManager.pycompat import all
+
−
+
−_ = lambda msg: msg
+
−
+
−
+
−class QuotaLimit(object):
+
−    """Class to handle quota limit specific data."""
+
−    __slots__ = ('_dbh', '_qid', '_bytes', '_messages')
+
−    _kwargs = ('qid', 'bytes', 'messages')
+
−
+
−    def __init__(self, dbh, **kwargs):
+
−        """Create a new QuotaLimit instance.
+
−
+
−        Either the `qid` keyword or the `bytes` and `messages` keywords
+
−        must be specified.
+
−
+
−        Arguments:
+
−
+
−        `dbh` : pyPgSQL.PgSQL.Connection || psycopg2._psycopg.connection
+
−          A database connection for the database access.
+
−
+
−        Keyword arguments:
+
−
+
−        `qid` : int
+
−          The id of a quota limit
+
−        `bytes` : long
+
−          The quota limit in bytes.
+
−        `messages` : int
+
−          The quota limit in number of messages
+
−        """
+
−        self._dbh = dbh
+
−        self._qid = 0
+
−        self._bytes = 0
+
−        self._messages = 0
+
−
+
−        for key in kwargs.iterkeys():
+
−            if key not in self.__class__._kwargs:
+
−                raise ValueError('unrecognized keyword: %r' % key)
+
−        qid = kwargs.get('qid')
+
−        if qid is not None:
+
−            assert isinstance(qid, (int, long))
+
−            self._load_by_qid(qid)
+
−        else:
+
−            bytes_, msgs = kwargs.get('bytes'), kwargs.get('messages')
+
−            assert all(isinstance(i, (int, long)) for i in (bytes_, msgs))
+
−            if bytes_ < 0:
+
−                self._bytes = -bytes_
+
−            else:
+
−                self._bytes = bytes_
+
−            if msgs < 0:
+
−                self._messages = -msgs
+
−            else:
+
−                self._messages = msgs
+
−            self._load_by_limit()
+
−
+
−    @property
+
−    def bytes(self):
+
−        """Quota limit in bytes."""
+
−        return self._bytes
+
−
+
−    @property
+
−    def messages(self):
+
−        """Quota limit in number of messages."""
+
−        return self._messages
+
−
+
−    @property
+
−    def qid(self):
+
−        """The quota limit's unique ID."""
+
−        return self._qid
+
−
+
−    def __eq__(self, other):
+
−        if isinstance(other, self.__class__):
+
−            return self._qid == other._qid
+
−        return NotImplemented
+
−
+
−    def __ne__(self, other):
+
−        if isinstance(other, self.__class__):
+
−            return self._qid != other._qid
+
−        return NotImplemented
+
−
+
−    def _load_by_limit(self):
+
−        """Load the quota limit by limit values from the database."""
+
−        dbc = self._dbh.cursor()
+
−        dbc.execute('SELECT qid FROM quotalimit WHERE bytes = %s AND '
+
−                    'messages = %s', (self._bytes, self._messages))
+
−        res = dbc.fetchone()
+
−        dbc.close()
+
−        if res:
+
−            self._qid = res[0]
+
−        else:
+
−            self._save()
+
−
+
−    def _load_by_qid(self, qid):
+
−        """Load the quota limit by its unique ID from the database."""
+
−        dbc = self._dbh.cursor()
+
−        dbc.execute('SELECT bytes, messages FROM quotalimit WHERE qid = %s',
+
−                    (qid,))
+
−        res = dbc.fetchone()
+
−        dbc.close()
+
−        if not res:
+
−            raise ValueError('Unknown quota limit id specified: %r' % qid)
+
−        self._qid = qid
+
−        self._bytes, self._messages = res
+
−
+
−    def _save(self):
+
−        """Store a new quota limit in the database."""
+
−        dbc = self._dbh.cursor()
+
−        dbc.execute("SELECT nextval('quotalimit_id')")
+
−        self._qid = dbc.fetchone()[0]
+
−        dbc.execute('INSERT INTO quotalimit (qid, bytes, messages) VALUES '
+
−                    '(%s, %s, %s)', (self._qid, self._bytes, self._messages))
+
−        self._dbh.commit()
+
−        dbc.close()
+
−
+
−del _