init branch

rma_account/models/rma_order.py

@@ -0,0 +1,145 @@
+# -*- coding: utf-8 -*-
+# © 2017 Eficent Business and IT Consulting Services S.L.
+# License LGPL-3.0 or later (https://www.gnu.org/licenses/lgpl.html)
+from openerp import _, api, fields, models
+from openerp.addons import decimal_precision as dp
+from openerp.exceptions import UserError
+from dateutil.relativedelta import relativedelta
+from openerp.tools import DEFAULT_SERVER_DATE_FORMAT
+from datetime import datetime
+
+
+class RmaOrder(models.Model):
+    _inherit = "rma.order"
+
+    @api.multi
+    def _compute_invoice_refund_count(self):
+        self.ensure_one()
+        invoice_list = []
+        for line in self.rma_line_ids:
+            for refund in line.refund_line_ids:
+                invoice_list.append(refund.invoice_id.id)
+        self.invoice_refund_count = len(list(set(invoice_list)))
+
+    @api.multi
+    def _compute_invoice_count(self):
+        self.ensure_one()
+        invoice_list = []
+        for line in self.rma_line_ids:
+            if line.invoice_line_id and line.invoice_line_id.id:
+                invoice_list.append(line.invoice_line_id.invoice_id.id)
+        self.invoice_count = len(list(set(invoice_list)))
+
+    add_invoice_id = fields.Many2one('account.invoice', string='Add Invoice',
+                                     ondelete='set null', readonly=True,
+                                     states={'draft': [('readonly', False)]})
+    invoice_refund_count = fields.Integer(
+        compute=_compute_invoice_refund_count,
+        string='# of Refunds',
+        copy=False)
+    invoice_count = fields.Integer(compute=_compute_invoice_count,
+                                   string='# of Incoming Shipments',
+                                   copy=False)
+
+    def _prepare_rma_line_from_inv_line(self, line):
+        operation = line.product_id.rma_operation_id and \
+                    line.product_id.rma_operation_id.id or False
+        if not operation:
+            operation = line.product_id.categ_id.rma_operation_id and \
+                        line.product_id.categ_id.rma_operation_id.id or False
+        data = {
+            'invoice_line_id': line.id,
+            'product_id': line.product_id.id,
+            'name': line.name,
+            'origin': line.invoice_id.number,
+            'uom_id': line.uom_id.id,
+            'operation_id': operation,
+            'product_qty': line.quantity,
+            'price_unit': line.invoice_id.currency_id.compute(
+                line.price_unit, line.currency_id, round=False),
+            'rma_id': self._origin.id
+        }
+        return data
+
+    @api.onchange('add_invoice_id')
+    def on_change_invoice(self):
+        if not self.add_invoice_id:
+            return {}
+        if not self.partner_id:
+            self.partner_id = self.add_invoice_id.partner_id.id
+        new_lines = self.env['rma.order.line']
+        for line in self.add_invoice_id.invoice_line_ids:
+            # Load a PO line only once
+            if line in self.rma_line_ids.mapped('invoice_line_id'):
+                continue
+            data = self._prepare_rma_line_from_inv_line(line)
+            new_line = new_lines.new(data)
+            new_lines += new_line
+
+        self.rma_line_ids += new_lines
+        self.date_rma = fields.Datetime.now()
+        self.delivery_address_id = self.add_invoice_id.partner_id.id
+        self.invoice_address_id = self.add_invoice_id.partner_id.id
+        self.add_invoice_id = False
+        return {}
+
+    @api.model
+    def prepare_rma_line(self, origin_rma, rma_id, line):
+        line_values = super(RmaOrder, self).prepare_rma_line(
+            origin_rma, rma_id, line)
+        line_values['invoice_address_id'] = line.invoice_address_id.id
+        return line_values
+
+    @api.model
+    def _prepare_rma_data(self, partner, origin_rma):
+        res = super(RmaOrder, self)._prepare_rma_data(partner, origin_rma)
+        res['invoice_address_id'] = partner.id
+        return res
+
+    @api.multi
+    def action_view_invoice_refund(self):
+        """
+        This function returns an action that display existing vendor refund
+        bills of given purchase order id.
+        When only one found, show the vendor bill immediately.
+        """
+        action = self.env.ref('account.action_invoice_tree2')
+        result = action.read()[0]
+        invoice_list = []
+        for line in self.rma_line_ids:
+            for refund in line.refund_line_ids:
+                invoice_list.append(refund.invoice_id.id)
+        invoice_ids = list(set(invoice_list))
+        # choose the view_mode accordingly
+        if len(invoice_ids) != 1:
+            result['domain'] = "[('id', 'in', " + \
+                               str(invoice_ids) + ")]"
+        elif len(invoice_ids) == 1:
+            res = self.env.ref('account.invoice_supplier_form', False)
+            result['views'] = [(res and res.id or False, 'form')]
+            result['res_id'] = invoice_ids[0]
+        return result
+
+    @api.multi
+    def action_view_invoice(self):
+        if self.type == "supplier":
+            action = self.env.ref('account.action_invoice_tree2')
+        else:
+            action = self.env.ref('account.action_invoice_tree')
+        result = action.read()[0]
+        invoice_list = []
+        for line in self.rma_line_ids:
+            invoice_list.append(line.invoice_id.id)
+        invoice_ids = list(set(invoice_list))
+        # choose the view_mode accordingly
+        if len(invoice_ids) != 1:
+            result['domain'] = "[('id', 'in', " + \
+                               str(invoice_ids) + ")]"
+        elif len(invoice_ids) == 1:
+            if self.type == "supplier":
+                res = self.env.ref('account.invoice_supplier_form', False)
+            else:
+                res = self.env.ref('account.invoice_form', False)
+            result['views'] = [(res and res.id or False, 'form')]
+            result['res_id'] = invoice_ids[0]
+        return result