From dfdcaf240551184ef75bb8a0ee2b2d52bf20057d Mon Sep 17 00:00:00 2001
From: Andhitia Rama <andhitia.r@gmail.com>
Date: Thu, 16 Nov 2023 11:41:12 +0700
Subject: [PATCH] [UPD] ssi_rma

* Lot implementation
* Custom partner location implementation
---
 ssi_rma/menu.xml                          |  2 +-
 ssi_rma/models/rma_line_mixin.py          | 55 ++++++++++++++++++++++-
 ssi_rma/models/rma_mixin.py               |  1 +
 ssi_rma/models/rma_route_template.py      |  6 +++
 ssi_rma/models/stock_rule.py              |  6 ++-
 ssi_rma/views/rma_mixin_view.xml          | 11 ++++-
 ssi_rma/views/rma_route_template_view.xml |  1 +
 7 files changed, 77 insertions(+), 5 deletions(-)

diff --git a/ssi_rma/menu.xml b/ssi_rma/menu.xml
index d2913c3..e20c766 100644
--- a/ssi_rma/menu.xml
+++ b/ssi_rma/menu.xml
@@ -9,7 +9,7 @@
             id="menu_rma"
             name="RMA"
             parent="stock.menu_stock_root"
-            sequence="20"
+            sequence="110"
         />
         <menuitem
             id="menu_rma_config"
diff --git a/ssi_rma/models/rma_line_mixin.py b/ssi_rma/models/rma_line_mixin.py
index ea50060..d2a17fc 100644
--- a/ssi_rma/models/rma_line_mixin.py
+++ b/ssi_rma/models/rma_line_mixin.py
@@ -21,6 +21,12 @@ class RMALineMixin(models.AbstractModel):
         ondelete="cascade",
     )
     sequence = fields.Integer(string="Sequence", required=True, default=10)
+    allowed_lot_ids = fields.Many2many(
+        string="Allowed Lots",
+        comodel_name="stock.production.lot",
+        compute="_compute_allowed_lot_ids",
+        store=False,
+    )
     lot_id = fields.Many2one(comodel_name="stock.production.lot", string="Lot")
     stock_move_ids = fields.Many2many(
         comodel_name="stock.move",
@@ -55,6 +61,21 @@ class RMALineMixin(models.AbstractModel):
         string="Qty Delivered", compute="_compute_qty_delivered", store=True
     )
 
+    @api.depends(
+        "product_id",
+    )
+    def _compute_allowed_lot_ids(self):
+        for record in self:
+            result = []
+            if record.product_id:
+                Lot = self.env["stock.production.lot"]
+                result = Lot.search(
+                    [
+                        ("product_id", "=", record.product_id.id),
+                    ]
+                ).ids
+            record.allowed_lot_ids = result
+
     @api.depends(
         "stock_move_ids",
         "stock_move_ids.state",
@@ -137,6 +158,31 @@ def _compute_qty_delivered(self):
             ]
             record.qty_delivered = record._get_rma_move_qty(states, "out")
 
+    @api.onchange(
+        "product_id",
+    )
+    def onchange_lot_id(self):
+        self.lot_id = False
+
+    @api.onchange(
+        "product_id",
+        "uom_quantity",
+        "uom_id",
+        "pricelist_id",
+        "lot_id",
+    )
+    def onchange_price_unit(self):
+        _super = super(RMALineMixin, self)
+        self.price_unit = 0.0
+        if self.lot_id and self.uom_quantity and self.uom_quantity != 0.0:
+            self.price_unit = 7.0
+            if self.lot_id.quant_ids:
+
+                quant = self.lot_id.quant_ids[-1]
+                self.price_unit = quant.value / self.uom_quantity
+        else:
+            _super.onchange_price_unit()
+
     def _get_rma_move_qty(self, states, direction):
         result = 0.0
         rma_location = self.order_id.route_template_id.location_id
@@ -222,6 +268,8 @@ def _get_receipt_procurement_data(self):
             "product_uom": self.uom_id.id,
             "location_id": location,
             "route_ids": route,
+            "price_unit": self.price_unit,
+            "forced_lot_id": self.lot_id and self.lot_id.id,
         }
         if self._name == "rma_customer_line":
             result.update(
@@ -242,6 +290,10 @@ def _get_delivery_procurement_data(self):
         origin = self.order_id.name
         warehouse = self.order_id.route_template_id.outbound_warehouse_id
         route = self.order_id.route_template_id.outbound_route_id
+        location = (
+            self.order_id.route_template_id.partner_location_id
+            or self.order_id.partner_id.property_stock_customer,
+        )  # TODO
         result = {
             "name": self.order_id.name,
             "group_id": group,
@@ -252,8 +304,9 @@ def _get_delivery_procurement_data(self):
             "product_qty": self.qty_to_deliver,
             "partner_id": self.order_id.partner_id.id,
             "product_uom": self.uom_id.id,
-            "location_id": self.order_id.partner_id.property_stock_customer,
+            "location_id": location,
             "route_ids": route,
+            "forced_lot_id": self.lot_id and self.lot_id.id,
         }
         if self._name == "rma_customer_line":
             result.update(
diff --git a/ssi_rma/models/rma_mixin.py b/ssi_rma/models/rma_mixin.py
index ec5bf2a..986838e 100644
--- a/ssi_rma/models/rma_mixin.py
+++ b/ssi_rma/models/rma_mixin.py
@@ -100,6 +100,7 @@ class RMAMixin(models.AbstractModel):
         comodel_name="procurement.group",
         string="Procurement Group",
         ondelete="restrict",
+        readonly=True,
     )
     line_ids = fields.One2many(
         comodel_name="rma_line_mixin",
diff --git a/ssi_rma/models/rma_route_template.py b/ssi_rma/models/rma_route_template.py
index e6a8707..591b481 100644
--- a/ssi_rma/models/rma_route_template.py
+++ b/ssi_rma/models/rma_route_template.py
@@ -40,6 +40,12 @@ class RMARouteTemplate(models.Model):
         required=True,
         ondelete="restrict",
     )
+    partner_location_id = fields.Many2one(
+        comodel_name="stock.location",
+        string="Partner Location",
+        required=False,
+        ondelete="restrict",
+    )
     customer_to_supplier = fields.Boolean(
         string="The customer will send to the supplier", default=False
     )
diff --git a/ssi_rma/models/stock_rule.py b/ssi_rma/models/stock_rule.py
index cc62878..ea929ea 100644
--- a/ssi_rma/models/stock_rule.py
+++ b/ssi_rma/models/stock_rule.py
@@ -12,7 +12,9 @@ class StockRule(models.Model):
     def _get_custom_move_fields(self):
         _super = super(StockRule, self)
         result = _super._get_custom_move_fields()
-        result.append(
+        result += [
             "customer_rma_line_ids",
-        )
+            "price_unit",
+            "forced_lot_id",
+        ]
         return result
diff --git a/ssi_rma/views/rma_mixin_view.xml b/ssi_rma/views/rma_mixin_view.xml
index 04c364c..ddc3a4f 100644
--- a/ssi_rma/views/rma_mixin_view.xml
+++ b/ssi_rma/views/rma_mixin_view.xml
@@ -92,7 +92,15 @@
                                     <field name="sequence" />
                                     <field name="product_id" />
                                     <field name="name" />
-                                    <field name="lot_id" />
+                                    <field
+                                        name="allowed_lot_ids"
+                                        widget="many2many_tags"
+                                        invisible="1"
+                                    />
+                                    <field
+                                        name="lot_id"
+                                        domain="[('id','in',allowed_lot_ids)]"
+                                    />
                                 </group>
                                 <group name="form_detail2" colspan="1" col="2">
                                     <field
@@ -153,6 +161,7 @@
                         class="oe_highlight"
                         icon="fa-sign-in"
                         attrs="{'invisible':[('receipt_ok','=',False)]}"
+                        style="margin-right:4px"
                     />
                     <button
                         name="action_create_delivery"
diff --git a/ssi_rma/views/rma_route_template_view.xml b/ssi_rma/views/rma_route_template_view.xml
index 0de81a5..f7d952d 100644
--- a/ssi_rma/views/rma_route_template_view.xml
+++ b/ssi_rma/views/rma_route_template_view.xml
@@ -41,6 +41,7 @@
             <data>
                 <xpath expr="//field[@name='active']" position="after">
                     <field name="location_id" />
+                    <field name="partner_location_id" />
                     <field name="customer_to_supplier" />
                     <field name="supplier_to_customer" />
                 </xpath>