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