Initial commit: Odoo 18.0-20251222 extra-addons

sale_partner_primeship/models/sale_order.py

@@ -0,0 +1,111 @@
+# Copyright 2021 Akretion France (http://www.akretion.com/)
+# Copyright 2024 Akretion France (http://www.akretion.com/)
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
+
+from dateutil.relativedelta import relativedelta
+
+from odoo import api, fields, models
+
+
+class SaleOrder(models.Model):
+    _inherit = "sale.order"
+
+    def _action_confirm(self):
+        super()._action_confirm()
+        sale_primeship = self.env["sale.primeship"]
+        for record in self:
+            partner = record.partner_id.commercial_partner_id
+
+            for line in record.order_line:
+                product_template = line.product_template_id
+
+                if product_template.primeship_activation:
+                    # We do not take in account product_qty for now:
+                    duration = product_template.primeship_duration
+                    start = fields.Date.context_today(record)
+                    end = start + relativedelta(months=duration)
+
+                    # If we have already some primeships, we need to check for overlaps
+                    if partner.primeship_ids:
+                        # We assume no overlaps between partner primeships
+                        for primeship in partner.primeship_ids.sorted("start_date"):
+                            if primeship.overlaps(start, end):
+                                start = primeship.end_date
+                                end = start + relativedelta(months=duration)
+
+                    vals = {
+                        "start_date": start,
+                        "end_date": end,
+                        "partner_id": partner.id,
+                        "order_line_id": line.id,
+                        # this is to reactivate a maybe deactivated existing primeship
+                        "active": True,
+                    }
+                    if line.primeship_id:
+                        # Hm... something seems to have gone wrong here,
+                        # but we handle it nonetheless.
+                        line.primeship_id.write(vals)
+                    else:
+                        # We may have a deactivated primeship because of an order
+                        # cancellation.
+                        primeship = sale_primeship.with_context(
+                            active_test=False,
+                        ).search([("order_line_id", "=", line.id)])
+                        if primeship:
+                            primeship.write(vals)
+                        else:
+                            sale_primeship.create(vals)
+
+        return True
+
+    def _action_cancel(self):
+        rv = super()._action_cancel()
+        for record in self:
+            record.order_line.mapped("primeship_id").active = False
+        return rv
+
+
+class SaleOrderLine(models.Model):
+    _inherit = "sale.order.line"
+
+    primeship_id = fields.Many2one(
+        string="Primeships",
+        comodel_name="sale.primeship",
+        compute="_compute_primeship_id",
+        inverse="_inverse_primeship_id",
+    )
+
+    # One2one impl
+    primeship_ids = fields.One2many(
+        comodel_name="sale.primeship", inverse_name="order_line_id"
+    )
+
+    @api.depends("primeship_ids")
+    def _compute_primeship_id(self):
+        for record in self:
+            record.primeship_id = record.primeship_ids[:1]
+
+    def _inverse_primeship_id(self):
+        for record in self:
+            if record.primeship_ids:
+                primeship = record.env["sale.primeship"].browse(
+                    record.primeship_ids[0].id
+                )
+                primeship.order_line_id = record
+
+            record.primeship_id.order_line_id = record
+
+    def _prepare_invoice_line(self, **optional_values):
+        """Update invoice start/end dates.
+        Set invoice start/end dates to primeship start/end dates
+        In case of multi quantity, this assumes continuous date ranges."""
+        self.ensure_one()
+        res = super()._prepare_invoice_line(**optional_values)
+        if self.primeship_id:
+            res.update(
+                {
+                    "start_date": self.primeship_id.start_date,
+                    "end_date": self.primeship_id.end_date,
+                }
+            )
+        return res