stock-rma/rma_purchase/wizards/rma_add_purchase.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.exceptions import ValidationError


class RmaAddPurchase(models.TransientModel):
    _name = "rma_add_purchase"
    _description = "Wizard to add rma lines"

    @api.model
    def default_get(self, fields_list):
        res = super(RmaAddPurchase, self).default_get(fields_list)
        rma_obj = self.env["rma.order"]
        rma_id = self.env.context["active_ids"] or []
        active_model = self.env.context["active_model"]
        if not rma_id:
            return res
        assert active_model == "rma.order", "Bad context propagation"

        rma = rma_obj.browse(rma_id)
        res["rma_id"] = rma.id
        res["partner_id"] = rma.partner_id.id
        res["purchase_id"] = False
        res["purchase_line_ids"] = False
        return res

    rma_id = fields.Many2one(
        comodel_name="rma.order", string="RMA Order", readonly=True
    )
    partner_id = fields.Many2one(
        comodel_name="res.partner", string="Partner", readonly=True
    )
    purchase_id = fields.Many2one(comodel_name="purchase.order", string="Order")
    purchase_line_ids = fields.Many2many(
        comodel_name="purchase.order.line",
        relation="rma_add_purchase_add_line_rel",
        column1="rma_add_purchase_id",
        column2="purchase_line_id",
        readonly=False,
        string="Purchase Order Lines",
    )

    def _prepare_rma_line_from_po_line(self, line):
        if self.env.context.get("customer"):
            operation = (
                line.product_id.rma_customer_operation_id
                or line.product_id.categ_id.rma_customer_operation_id
            )
        else:
            operation = (
                line.product_id.rma_supplier_operation_id
                or line.product_id.categ_id.rma_supplier_operation_id
            )
        if not operation:
            operation = self.env["rma.operation"].search(
                [("type", "=", self.rma_id.type)], limit=1
            )
            if not operation:
                raise ValidationError(_("Please define an operation first"))
        if not operation.in_route_id or not operation.out_route_id:
            route = self.env["stock.location.route"].search(
                [("rma_selectable", "=", True)], limit=1
            )
            if not route:
                raise ValidationError(_("Please define a rma route."))
        if not operation.in_warehouse_id or not operation.out_warehouse_id:
            warehouse = self.env["stock.warehouse"].search(
                [
                    ("company_id", "=", self.rma_id.company_id.id),
                    ("lot_rma_id", "!=", False),
                ],
                limit=1,
            )
            if not warehouse:
                raise ValidationError(
                    _("Please define a warehouse with a " "default rma location.")
                )
        data = {
            "partner_id": self.partner_id.id,
            "purchase_order_line_id": line.id,
            "product_id": line.product_id.id,
            "origin": line.order_id.name,
            "uom_id": line.product_uom.id,
            "operation_id": operation.id,
            "product_qty": line.product_qty,
            "price_unit": line.currency_id._convert(
                line.price_unit,
                line.currency_id,
                self.env.user.company_id,
                fields.Date.today(),
                round=False,
            ),
            "rma_id": self.rma_id.id,
            "in_route_id": operation.in_route_id.id or route,
            "out_route_id": operation.out_route_id.id or route,
            "receipt_policy": operation.receipt_policy,
            "location_id": (
                operation.location_id.id
                or operation.in_warehouse_id.lot_rma_id.id
                or warehouse.lot_rma_id.id
            ),
            "refund_policy": operation.refund_policy,
            "delivery_policy": operation.delivery_policy,
            "in_warehouse_id": operation.in_warehouse_id.id or warehouse.id,
            "out_warehouse_id": operation.out_warehouse_id.id or warehouse.id,
        }
        return data

    @api.model
    def _get_rma_data(self):
        data = {
            "date_rma": fields.Datetime.now(),
        }
        return data

    @api.model
    def _get_existing_purchase_lines(self):
        existing_purchase_lines = []
        for rma_line in self.rma_id.rma_line_ids:
            existing_purchase_lines.append(rma_line.purchase_order_line_id)
        return existing_purchase_lines

    def add_lines(self):
        rma_line_obj = self.env["rma.order.line"]
        existing_purchase_lines = self._get_existing_purchase_lines()
        for line in self.purchase_line_ids:
            # Load a PO line only once
            if line not in existing_purchase_lines:
                data = self._prepare_rma_line_from_po_line(line)
                rma_line_obj.create(data)
        rma = self.rma_id
        data_rma = self._get_rma_data()
        rma.write(data_rma)
        return {"type": "ir.actions.act_window_close"}