From a59c252ef1af0a81b5220b418dcc52e3baadf229 Mon Sep 17 00:00:00 2001
From: Cristiano Sticca
 <148890858+cristianosticca-pagopa@users.noreply.github.com>
Date: Thu, 16 Jan 2025 09:03:46 +0100
Subject: [PATCH] Update station-maintenance-aggregated.py

---
 .../scripts/station-maintenance-aggregated.py     | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/src/main/scripts/station-maintenance-aggregated.py b/src/main/scripts/station-maintenance-aggregated.py
index de0f630..d066fc4 100644
--- a/src/main/scripts/station-maintenance-aggregated.py
+++ b/src/main/scripts/station-maintenance-aggregated.py
@@ -56,22 +56,23 @@ def get_pa_for_broker(broker_code):
     }
     
     all_creditor_institutions = []
-
     while True:
         params = {'page': page, 'limit': limit}
         response = requests.get(base_url, headers=headers, params=params)
         response.raise_for_status()
-
         data = response.json()
-        creditor_institutions = data.get('creditorInstitutions', [])
-        all_creditor_institutions.extend(creditor_institutions)
+        
+        all_creditor_institutions.extend(data['creditorInstitutions'])
 
-        total_pages = data.get('pageInfo', {}).get('totalPages', 1)
-        if page >= total_pages - 1:
+        # Verifica se siamo arrivati all'ultima pagina
+        if len(data['creditorInstitutions']) < limit:
             break
         page += 1
 
-    return all_creditor_institutions
+    # Restituire la struttura originale
+    return {
+        'creditorInstitutions': all_creditor_institutions
+    }
 
 
 def create_betterstack_maintenance(data):