|
1 | 1 | # Copyright (C) 2019 Brian McMaster
|
2 | 2 | # Copyright (C) 2019 Open Source Integrators
|
3 | 3 | # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
|
4 |
| -from odoo import api, fields, models |
| 4 | +from odoo import Command, api, fields, models |
5 | 5 |
|
6 | 6 |
|
7 | 7 | class SaleOrderLine(models.Model):
|
@@ -53,8 +53,31 @@ def create(self, vals_list):
|
53 | 53 | line.order_id._field_service_generation()
|
54 | 54 | return lines
|
55 | 55 |
|
| 56 | + def _get_invoiceable_fsm_order_domain(self): |
| 57 | + """ |
| 58 | + Override this method to define more search criteria for invoiceable |
| 59 | + fsm order |
| 60 | + :return: |
| 61 | + """ |
| 62 | + invoiceable_stage_ids = self.env["fsm.stage"]._get_invoiceable_stage() |
| 63 | + dom = [ |
| 64 | + "|", |
| 65 | + ("sale_line_id", "=", self.id), |
| 66 | + ("sale_id", "=", self.order_id.id), |
| 67 | + ("invoice_lines", "=", False), |
| 68 | + ] |
| 69 | + if invoiceable_stage_ids: |
| 70 | + dom.append(("stage_id", "in", invoiceable_stage_ids.ids)) |
| 71 | + return dom |
| 72 | + |
| 73 | + def _get_invoiceable_fsm_order(self): |
| 74 | + dom = self._get_invoiceable_fsm_order_domain() |
| 75 | + return self.env["fsm.order"].search(dom) |
| 76 | + |
56 | 77 | def _prepare_invoice_line(self, **optional_values):
|
57 | 78 | res = super()._prepare_invoice_line(**optional_values)
|
58 | 79 | if self.fsm_order_id:
|
59 |
| - res.update({"fsm_order_ids": [(4, self.fsm_order_id.id)]}) |
| 80 | + fsm_orders = self._get_invoiceable_fsm_order() |
| 81 | + if fsm_orders: |
| 82 | + res.update({"fsm_order_ids": [Command.set(fsm_orders.ids)]}) |
60 | 83 | return res
|
0 commit comments