first commit

modules/commission_waiting/commission.py

@@ -0,0 +1,97 @@
+# 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 trytond.model import fields
+from trytond.pool import Pool, PoolMeta
+from trytond.pyson import Bool, Eval
+from trytond.transaction import Transaction
+
+
+class Agent(metaclass=PoolMeta):
+    __name__ = 'commission.agent'
+    waiting_account = fields.Many2One('account.account', 'Waiting Account',
+        domain=[
+            ('type', '!=', None),
+            ('closed', '!=', True),
+            ('company', '=', Eval('company', -1)),
+            ],
+        help="The account the agent's waiting commission amounts are posted "
+        "to.")
+
+
+class Commission(metaclass=PoolMeta):
+    __name__ = 'commission'
+    waiting_move = fields.Many2One(
+        'account.move', "Waiting Move", readonly=True)
+
+    @classmethod
+    def __setup__(cls):
+        super().__setup__()
+        cls.amount.states['readonly'] |= Bool(Eval('waiting_move'))
+
+    @classmethod
+    def copy(cls, commissions, default=None):
+        if default is None:
+            default = {}
+        else:
+            default = default.copy()
+        default.setdefault('waiting_move', None)
+        return super().copy(commissions, default=default)
+
+    @classmethod
+    def create_waiting_move(cls, commissions):
+        pool = Pool()
+        Move = pool.get('account.move')
+        Commission = pool.get('commission')
+        moves = []
+        for commission in commissions:
+            move = commission.get_move()
+            if move:
+                moves.append(move)
+            commission.waiting_move = move
+
+        Move.save(moves)
+        Commission.save(commissions)
+
+    def get_move(self, date=None):
+        pool = Pool()
+        Move = pool.get('account.move')
+        Line = pool.get('account.move.line')
+        Date = pool.get('ir.date')
+        Period = pool.get('account.period')
+        Currency = pool.get('currency.currency')
+
+        if not self.agent.waiting_account:
+            return
+        if self.waiting_move:
+            return self.waiting_move
+
+        if date is None:
+            with Transaction().set_context(company=self.agent.company.id):
+                date = Date.today()
+        period = Period.find(self.agent.company, date=date)
+
+        move = Move(journal=self.get_journal(), origin=self,
+            date=date, period=period, company=self.agent.company)
+        amount = Currency.compute(self.currency, self.amount,
+            self.agent.company.currency)
+        line = Line()
+        with Transaction().set_context(date=date):
+            if self.type_ == 'in':
+                line.credit = amount if amount > 0 else 0
+                line.debit = amount if amount < 0 else 0
+                line.account = self.product.account_revenue_used
+            else:
+                line.debit = amount if amount > 0 else 0
+                line.credit = amount if amount < 0 else 0
+                line.account = self.product.account_expense_used
+        if line.account.party_required:
+            line.party = self.agent.party
+        # XXX second currency?
+        counterpart = Line()
+        counterpart.debit, counterpart.credit = line.credit, line.debit
+        counterpart.account = self.agent.waiting_account
+        if counterpart.account.party_required:
+            counterpart.party = self.agent.party
+        move.lines = (line, counterpart)
+        return move