first commit

modules/account_deposit/party.py

@@ -0,0 +1,162 @@
+# This file is part of Tryton.  The COPYRIGHT file at the top level of
+# this repository contains the full copyright notices and license terms.
+from decimal import Decimal
+
+from sql import For, Literal, Null
+from sql.aggregate import Sum
+from sql.conditionals import Case, Coalesce
+
+from trytond import backend
+from trytond.i18n import gettext
+from trytond.model import fields
+from trytond.modules.currency.fields import Monetary
+from trytond.modules.party.exceptions import EraseError
+from trytond.pool import Pool, PoolMeta
+from trytond.tools import grouped_slice, reduce_ids, sqlite_apply_types
+from trytond.transaction import Transaction
+
+
+class Party(metaclass=PoolMeta):
+    __name__ = 'party.party'
+
+    deposit = fields.Function(Monetary(
+            "Deposit", currency='currency', digits='currency'),
+        'get_deposit', searcher='search_deposit')
+
+    @classmethod
+    def get_deposit(cls, parties, name):
+        pool = Pool()
+        MoveLine = pool.get('account.move.line')
+        Account = pool.get('account.account')
+        AccountType = pool.get('account.account.type')
+        User = pool.get('res.user')
+        cursor = Transaction().connection.cursor()
+
+        line = MoveLine.__table__()
+        account = Account.__table__()
+        account_type = AccountType.__table__()
+
+        values = {p.id: Decimal(0) for p in parties}
+
+        user = User(Transaction().user)
+        if not user.company:
+            return values
+        currency = user.company.currency
+
+        line_clause, _ = MoveLine.query_get(line)
+
+        for sub_parties in grouped_slice(parties):
+            party_clause = reduce_ids(line.party, [p.id for p in sub_parties])
+            query = (line
+                .join(account, condition=account.id == line.account)
+                .join(account_type, condition=account.type == account_type.id)
+                .select(line.party,
+                    # Use credit - debit to positive deposit amount
+                    Sum(Coalesce(line.credit, 0) - Coalesce(line.debit, 0)
+                        ).as_('debit'),
+                    where=account_type.deposit
+                    & party_clause
+                    & (line.reconciliation == Null)
+                    & (account.company == user.company.id)
+                    & line_clause,
+                    group_by=line.party))
+            if backend.name == 'sqlite':
+                sqlite_apply_types(query, [None, 'NUMERIC'])
+            cursor.execute(*query)
+            values.update((p, currency.round(a)) for p, a in cursor)
+        return values
+
+    @classmethod
+    def search_deposit(cls, name, clause):
+        pool = Pool()
+        MoveLine = pool.get('account.move.line')
+        Account = pool.get('account.account')
+        AccountType = pool.get('account.account.type')
+        User = pool.get('res.user')
+
+        line = MoveLine.__table__()
+        account = Account.__table__()
+        account_type = AccountType.__table__()
+
+        user = User(Transaction().user)
+        if not user.company:
+            return []
+
+        line_clause, _ = MoveLine.query_get(line)
+        Operator = fields.SQL_OPERATORS[clause[1]]
+
+        query = (line
+            .join(account, condition=account.id == line.account)
+            .join(account_type, condition=account.type == account_type.id)
+            .select(line.party,
+                where=account.active
+                & account_type.deposit
+                & (line.party != Null)
+                & (line.reconciliation == Null)
+                & (account.company == user.company.id)
+                & line_clause,
+                group_by=line.party,
+                having=Operator(
+                    # Use credit - debit to positive deposit amount
+                    Sum(Coalesce(line.credit, 0) - Coalesce(line.debit, 0)),
+                    Decimal(clause[2] or 0))))
+        return [('id', 'in', query)]
+
+    def get_deposit_balance(self, deposit_account, currency=None):
+        'Return the deposit account balance (debit - credit) for the party'
+        pool = Pool()
+        MoveLine = pool.get('account.move.line')
+        transaction = Transaction()
+        cursor = transaction.connection.cursor()
+
+        line = MoveLine.__table__()
+        if currency is None:
+            currency = deposit_account.currency
+        assert deposit_account.type.deposit
+
+        where = ((line.account == deposit_account.id)
+            & (line.party == self.id)
+            & (line.reconciliation == Null))
+        if transaction.database.has_select_for():
+            cursor.execute(*line.select(
+                    Literal(1),
+                    where=where,
+                    for_=For('UPDATE', nowait=True)))
+        else:
+            MoveLine.lock()
+
+        if currency == deposit_account.currency:
+            amount = Sum(Coalesce(line.debit, 0) - Coalesce(line.credit, 0))
+        else:
+            amount = Sum(Case(
+                    (line.second_currency == currency.id,
+                        line.amount_second_currency),
+                    else_=0))
+
+        query = line.select(amount.as_('deposit_balance'), where=where)
+        if backend.name == 'sqlite':
+            sqlite_apply_types(query, ['NUMERIC'])
+        cursor.execute(*query)
+        amount, = cursor.fetchone()
+        if amount is None:
+            amount = Decimal(0)
+        return currency.round(amount)
+
+    def check_deposit(self, deposit_account, sign=1):
+        '''Check if the deposit account balance (debit - credit) has the same
+        sign for the party'''
+        assert sign in (1, -1)
+        amount = self.get_deposit_balance(
+            deposit_account, currency=deposit_account.second_currency)
+        return not amount or ((amount < 0) == (sign < 0))
+
+
+class Erase(metaclass=PoolMeta):
+    __name__ = 'party.erase'
+
+    def check_erase_company(self, party, company):
+        if party.deposit:
+            raise EraseError(
+                gettext('account_deposit.msg_erase_party_deposit',
+                    party=party.rec_name,
+                    company=company.rec_name))