stock-rma/rma_account/models/invoice.py

# © 2017 Eficent Business and IT Consulting Services S.L.
# License LGPL-3.0 or later (https://www.gnu.org/licenses/lgpl.html)

from odoo import api, fields, models
from odoo.tools.float_utils import float_compare


class AccountInvoice(models.Model):
    _inherit = "account.invoice"

    @api.depends('invoice_line_ids.rma_line_ids')
    def _compute_rma_count(self):
        for inv in self:
            rmas = self.mapped('invoice_line_ids.rma_line_ids')
            inv.rma_count = len(rmas)

    def _prepare_invoice_line_from_rma_line(self, line):
        qty = line.qty_to_refund
        if float_compare(
                qty, 0.0, precision_rounding=line.uom_id.rounding) <= 0:
            qty = 0.0
        # Todo fill taxes from somewhere
        invoice_line = self.env['account.invoice.line']
        data = {
            'purchase_line_id': line.id,
            'name': line.name + ': '+line.name,
            'origin': line.origin,
            'uom_id': line.uom_id.id,
            'product_id': line.product_id.id,
            'account_id': invoice_line.with_context(
                {'journal_id': self.journal_id.id,
                 'type': 'in_invoice'})._default_account(),
            'price_unit': line.company_id.currency_id.with_context(
                date=self.date_invoice).compute(
                line.price_unit, self.currency_id, round=False),
            'quantity': qty,
            'discount': 0.0,
            'rma_line_ids': [(4, line.id)],
        }
        return data

    @api.onchange('add_rma_line_id')
    def on_change_add_rma_line_id(self):
        if not self.add_rma_line_id:
            return {}
        if not self.partner_id:
            self.partner_id = self.add_rma_line_id.partner_id.id

        new_line = self.env['account.invoice.line']
        if self.add_rma_line_id not in (
                self.invoice_line_ids.mapped('rma_line_id')):
            data = self._prepare_invoice_line_from_rma_line(
                self.add_rma_line_id)
            new_line = new_line.new(data)
            new_line._set_additional_fields(self)
        self.invoice_line_ids += new_line
        self.add_rma_line_id = False
        return {}

    rma_count = fields.Integer(
        compute=_compute_rma_count, string='# of RMA')

    add_rma_line_id = fields.Many2one(
        comodel_name='rma.order.line',
        string="Add from RMA line",
        ondelete="set null",
        help="Create a refund in based on an existing rma_line")

    @api.multi
    def action_view_rma_supplier(self):
        action = self.env.ref('rma.action_rma_supplier_lines')
        result = action.read()[0]
        rma_ids = self.mapped('invoice_line_ids.rma_line_ids').ids
        if rma_ids:
            # choose the view_mode accordingly
            if len(rma_ids) > 1:
                result['domain'] = [('id', 'in', rma_ids)]
            else:
                res = self.env.ref('rma.view_rma_line_supplier_form', False)
                result['views'] = [(res and res.id or False, 'form')]
                result['res_id'] = rma_ids[0]
        return result

    @api.multi
    def action_view_rma_customer(self):
        action = self.env.ref('rma.action_rma_customer_lines')
        result = action.read()[0]
        rma_ids = self.mapped('invoice_line_ids.rma_line_ids').ids
        if rma_ids:
            # choose the view_mode accordingly
            if len(rma_ids) > 1:
                result['domain'] = [('id', 'in', rma_ids)]
            else:
                res = self.env.ref('rma.view_rma_line_form', False)
                result['views'] = [(res and res.id or False, 'form')]
                result['res_id'] = rma_ids[0]
        return result


class AccountInvoiceLine(models.Model):
    _inherit = "account.invoice.line"

    @api.model
    def name_search(self, name, args=None, operator='ilike', limit=100):
        """Allows to search by Invoice number. This has to be done this way,
        as Odoo adds extra args to name_search on _name_search method that
        will make impossible to get the desired result."""
        if not args:
            args = []
        lines = self.search(
            [('invoice_id.number', operator, name)] + args, limit=limit,
        )
        res = lines.name_get()
        if limit:
            limit_rest = limit - len(lines)
        else:
            # limit can be 0 or None representing infinite
            limit_rest = limit
        if limit_rest or not limit:
            args += [('id', 'not in', lines.ids)]
            res += super(AccountInvoiceLine, self).name_search(
                name, args=args, operator=operator, limit=limit_rest,
            )
        return res

    @api.multi
    def name_get(self):
        res = []
        if self.env.context.get('rma'):
            for inv in self:
                if inv.invoice_id.reference:
                    res.append(
                        (inv.id,
                         "INV:%s | REF:%s | ORIG:%s | PART:%s | QTY:%s" % (
                             inv.invoice_id.number or '',
                             inv.origin or '',
                             inv.invoice_id.reference or "",
                             inv.product_id.name, inv.quantity)))
                elif inv.invoice_id.number:
                    res.append(
                        (inv.id,
                         "INV:%s | ORIG:%s | PART:%s | QTY:%s" % (
                             inv.invoice_id.number or '',
                             inv.origin or '',
                             inv.product_id.name, inv.quantity)))
                else:
                    res.append(super(AccountInvoiceLine, inv).name_get()[0])
            return res
        else:
            return super(AccountInvoiceLine, self).name_get()

    @api.multi
    def _compute_rma_count(self):
        for invl in self:
            rma_lines = invl.mapped('rma_line_ids')
            invl.rma_line_count = len(rma_lines)

    rma_line_count = fields.Integer(
        compute=_compute_rma_count, string='# of RMA')
    rma_line_ids = fields.One2many(
        comodel_name='rma.order.line', inverse_name='invoice_line_id',
        string="RMA", readonly=True,
        help="This will contain the RMA lines for the invoice line")

    rma_line_id = fields.Many2one(
        comodel_name='rma.order.line',
        string="RMA line refund",
        ondelete="set null",
        help="This will contain the rma line that originated the refund line")