tryton/modules/account_invoice_correction/invoice.py

# 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 ModelView, fields
from trytond.pool import Pool, PoolMeta
from trytond.pyson import Eval
from trytond.transaction import Transaction
from trytond.wizard import Button, StateAction, StateView, Wizard


class Invoice(metaclass=PoolMeta):
    __name__ = 'account.invoice'

    @classmethod
    def __setup__(cls):
        super().__setup__()
        cls._buttons.update({
                'correct': {
                    'invisible': Eval('state') != 'posted',
                    'depends': ['state', 'type'],
                    },
                })

    @classmethod
    @ModelView.button_action(
        'account_invoice_correction.wizard_invoice_correct')
    def correct(cls, invoices):
        pass


class InvoiceCorrectStart(ModelView):
    __name__ = 'account.invoice.correct.start'

    invoice = fields.Many2One('account.invoice', 'Invoice', readonly=True)
    lines = fields.Many2Many('account.invoice.line', None, None,
        'Invoice Lines',
        domain=[
            ('invoice', '=', Eval('invoice', -1)),
            ('type', '=', 'line'),
            ])


class InvoiceCorrect(Wizard):
    __name__ = 'account.invoice.correct'

    start = StateView('account.invoice.correct.start',
        'account_invoice_correction.correct_start_view_form', [
            Button('Cancel', 'end', 'tryton-cancel'),
            Button('Correct', 'correct', 'tryton-ok', default=True),
            ])
    correct = StateAction('account_invoice.act_invoice_form')

    def default_start(self, fields):
        context = Transaction().context
        return {
            'invoice': context['active_id'],
            'lines': [],
            }

    def do_correct(self, action):
        pool = Pool()
        Invoice = self.model
        Line = pool.get('account.invoice.line')

        invoice = self.record
        with Transaction().set_context(_account_invoice_correction=True):
            correction, = Invoice.copy([invoice], default={
                    'lines': [],
                    })
            # Copy each line one by one to get negative and positive lines
            # following each other
            for line in self.start.lines:
                Line.copy([line], default={
                        'invoice': correction.id,
                        'quantity': -line.quantity,
                        })
                Line.copy([line], default={
                        'invoice': correction.id,
                        })
        correction.update_taxes()
        data = {'res_id': [correction.id]}
        return action, data