Fix justice.gov.uk drift

ministryofjustice · Aug 16, 2023 · 395e58c · 395e58c
1 parent d21825b
commit 395e58c
Show file tree

Hide file tree

Showing 3 changed files with 1,260 additions and 598 deletions.
diff --git a/scripts/create-zone.py b/scripts/create-zone.py
@@ -8,7 +8,7 @@
 # USER ACTION
 # ----------------------------------------------------
 # List of zone names to process
-ZONE_NAMES = ["zone1", "zone2"]
+ZONE_NAMES = ["justice.gov.uk"]
 
 # Toggle auto-import behavior
 AUTO_IMPORT = False

diff --git a/scripts/lib/HostedZone.py b/scripts/lib/HostedZone.py
@@ -48,13 +48,16 @@ def get_records(self):
         """Returns Records in Hosted Zone owned by this object
 
         Returns:
-            dict: Dict holding all records in Hosted Zone owned by this object
+            list: List holding all records in Hosted Zone owned by this object
         """
 
-        return self.client.list_resource_record_sets(
-            HostedZoneId=self.id,
-            MaxItems='250'
-        )['ResourceRecordSets']
+        records = []
+        paginator = self.client.get_paginator('list_resource_record_sets')
+
+        for page in paginator.paginate(HostedZoneId=self.id):
+            records.extend(page['ResourceRecordSets'])
+
+        return records
 
     def get_tags(self):
         """Returns tags for Hosted Zone owned by this object