From b0212d5fda001fd1bec6ac70b33058947022685a Mon Sep 17 00:00:00 2001
From: Daniil Digtyar Vasilieva <daniildigtyar@gmail.com>
Date: Mon, 30 Oct 2023 12:02:55 +0100
Subject: [PATCH] [REF] energy_selfconsumption: use method to get contract line
 values

---
 .../wizards/define_invoicing_mode_wizard.py   | 21 ++++++++++++-------
 1 file changed, 13 insertions(+), 8 deletions(-)

diff --git a/energy_selfconsumption/wizards/define_invoicing_mode_wizard.py b/energy_selfconsumption/wizards/define_invoicing_mode_wizard.py
index 063ea6939..5ffa60d27 100644
--- a/energy_selfconsumption/wizards/define_invoicing_mode_wizard.py
+++ b/energy_selfconsumption/wizards/define_invoicing_mode_wizard.py
@@ -82,6 +82,16 @@ class ContractGenerationWizard(models.TransientModel):
             "recurring_invoicing_type": "post-paid",
         }
 
+    def _prepare_contract_line_template_values(self, product_id, formula_contract_id):
+        return {
+            "product_id": product_id.id,
+            "automatic_price": True,
+            "company_id": self.env.company.id,
+            "qty_type": "variable",
+            "qty_formula_id": formula_contract_id.id,
+            "name": "",
+        }
+
     def save_data_to_selfconsumption(self):
         if self.invoicing_mode == "energy_delivered_variable":
             raise UserError(_("This invoicing mode is not yet implemented"))
@@ -146,14 +156,9 @@ result = line.supply_point_assignation_id.distribution_table_id.selfconsumption_
             (
                 0,
                 0,
-                {
-                    "product_id": product_id.id,
-                    "automatic_price": True,
-                    "company_id": self.env.company.id,
-                    "qty_type": "variable",
-                    "qty_formula_id": formula_contract_id.id,
-                    "name": "",
-                },
+                self._prepare_contract_line_template_values(
+                    product_id, formula_contract_id
+                ),
             )
         ]
 
-- 
GitLab