From 5b062cf72bcb1dde2d29f053f618612dabcd1089 Mon Sep 17 00:00:00 2001
From: Roberto Fichera <robyf@tekno-soft.it>
Date: Tue, 23 Jan 2024 16:15:06 +0100
Subject: [PATCH] [14.0][l10n_it_intrastat_statement] Code optimization to
 remove unnecessary triple for loop when generating the statement

---
 .../models/intrastat_statement.py             | 55 +++++++++++--------
 1 file changed, 31 insertions(+), 24 deletions(-)

diff --git a/l10n_it_intrastat_statement/models/intrastat_statement.py b/l10n_it_intrastat_statement/models/intrastat_statement.py
index 4e6cb7e39a9d..4c736fdf0d4f 100644
--- a/l10n_it_intrastat_statement/models/intrastat_statement.py
+++ b/l10n_it_intrastat_statement/models/intrastat_statement.py
@@ -779,33 +779,40 @@ def compute_statement(self):
             inv_type += ["in_invoice", "in_refund"]
         domain.append(("move_type", "in", inv_type))
 
-        statement_data = dict()
+        statement_data = {}
+        section_field_reverse = {}
+        for section_type in ["purchase", "sale"]:
+            for section_number in range(1, 5):
+                statement_section_field = self.get_section_field_name(
+                    section_type, section_number
+                )
+                statement_data[statement_section_field] = list()
+                section_field_reverse['{}_s{}'.format(section_type,
+                    section_number)] = (section_type, section_number)
+
         invoices = self.env["account.move"].search(domain)
 
         for inv_intra_line in invoices.mapped("intrastat_line_ids"):
-            for section_type in ["purchase", "sale"]:
-                for section_number in range(1, 5):
-                    section_details = (section_type, section_number)
-                    statement_section = "%s_s%s" % section_details
-                    if inv_intra_line.statement_section != statement_section:
-                        continue
-                    statement_section_model_name = self.get_section_model(
-                        *section_details
-                    )
-                    st_line = self.env[
-                        statement_section_model_name
-                    ]._prepare_statement_line(inv_intra_line, self)
-                    if not st_line:
-                        continue
-                    statement_section_field = self.get_section_field_name(
-                        *section_details
-                    )
-                    if statement_section_field not in statement_data:
-                        statement_data[statement_section_field] = list()
-                    st_line["sequence"] = (
-                        len(statement_data[statement_section_field]) + 1
-                    )
-                    statement_data[statement_section_field].append((0, 0, st_line))
+            section_details = section_field_reverse[
+                inv_intra_line.statement_section
+            ]
+            statement_section_model_name = self.get_section_model(
+                *section_details
+            )
+            st_line = self.env[
+                statement_section_model_name
+            ]._prepare_statement_line(inv_intra_line, self)
+            if not st_line:
+                continue
+            statement_section_field = self.get_section_field_name(
+                *section_details
+            )
+            if statement_section_field not in statement_data:
+                statement_data[statement_section_field] = list()
+            st_line["sequence"] = (
+                len(statement_data[statement_section_field]) + 1
+            )
+            statement_data[statement_section_field].append((0, 0, st_line))
 
         self.write(statement_data)
         # Group refund to sale lines if they have the same period of ref