move.py

# 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.pyson import Eval
from trytond.pool import PoolMeta


class RescheduleLines(metaclass=PoolMeta):
    "Reschedule Lines"
    __name__ = 'account.move.line.reschedule'

    @classmethod
    def __setup__(cls):
        super().__setup__()
        for button in cls.start.buttons:
            if button.state == 'preview':
                button.validate = None

    def default_start(self, fields):
        values = super().default_start(fields)
        terms = []
        for record in self.records:
            terms.append({
                    'date': record.maturity_date,
                    'amount': record.amount,
                    'currency': values.get('currency'),
                    'description': record.description,
                    'payment_type': (record.payment_type.id
                        if record.payment_type else None),
                    'bank_account': (record.bank_account.id
                        if record.bank_account else None),
                    })
        values['terms'] = terms
        return values

    def default_preview(self, fields):
        values = super().default_preview(fields)

        move_descriptions = list({r.move.description for r in self.records
            if r.move and r.move.description})
        if move_descriptions:
            values['description'] = move_descriptions[0]

        line_descriptions = list({r.description for r in self.records
            if r.description})
        payment_types = list({r.payment_type.id for r in self.records
            if r.payment_type})
        bank_accounts = list({r.bank_account.id for r in self.records
            if r.bank_account})
        for value in values.get('terms', []):
            if line_descriptions:
                value['description'] = line_descriptions[0]
            if payment_types:
                value['payment_type'] = payment_types[0]
            if bank_accounts:
                value['bank_account'] = bank_accounts[0]
        return values

    @classmethod
    def get_reschedule_move(
            cls, amount, balance, journal, terms, account, date=None,
            **line_values):
        move, balance_line = super().get_reschedule_move(amount, balance,
            journal, terms, account, date, **line_values)
        for line in move.lines:
            for term in terms:
                if term.date == line.maturity_date:
                    line.description = term.description
                    line.payment_type = term.payment_type
                    if getattr(term, 'bank_account', None):
                        line.bank_account = term.bank_account
                    else:
                        line.bank_account = None
                    if not line.bank_account:
                        line.on_change_payment_type()

        return move, balance_line


class RescheduleLinesStart(metaclass=PoolMeta):
    "Reschedule Lines"
    __name__ = 'account.move.line.reschedule.start'

    terms = fields.One2Many(
        'account.move.line.reschedule.term', None, "Terms",
        domain=[
            ('currency', '=', Eval('currency', -1)),
            ], readonly=True)


class RescheduleLinesTerm(metaclass=PoolMeta):
    "Reschedule Lines"
    __name__ = 'account.move.line.reschedule.term'

    description = fields.Char("Description")
    payment_type = fields.Many2One('account.payment.type', 'Payment Type')
    bank_account = fields.Many2One('bank.account', 'Bank Account')