iatoolkit 2.60.1__tar.gz → 2.60.2__tar.gz

{iatoolkit-2.60.1/src/iatoolkit.egg-info → iatoolkit-2.60.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 2.60.1
+Version: 2.60.2
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT

{iatoolkit-2.60.1 → iatoolkit-2.60.2}/src/iatoolkit/__init__.py

@@ -3,7 +3,7 @@
 #
 # IAToolkit is open source software.
 
-__version__ = "2.60.1"
+__version__ = "2.60.2"
 
 # Expose main classes and functions at the top level of the package
 

{iatoolkit-2.60.1 → iatoolkit-2.60.2}/src/iatoolkit/infra/connectors/google_drive_connector.py

@@ -104,6 +104,9 @@ class DriveChangesPage:
 
 class GoogleDriveConnector(FileConnector):
     PAGE_SIZE = 1000
+    # Reads retry Google's transient errors (5xx, 429) with exponential backoff,
+    # as googleapiclient does with num_retries; writes are not retried.
+    NUM_RETRIES = 3
 
     def __init__(
         self,
@@ -193,7 +196,7 @@ class GoogleDriveConnector(FileConnector):
                 fileId=file_id,
                 fields=FILE_FIELDS,
                 supportsAllDrives=True,
-            ).execute()
+            ).execute(num_retries=self.NUM_RETRIES)
         except HttpError as exc:
             if getattr(exc, "status_code", None) in (403, 404) or getattr(getattr(exc, "resp", None), "status", None) in (403, 404):
                 return None
@@ -211,7 +214,7 @@ class GoogleDriveConnector(FileConnector):
                 pageToken=page_token,
                 supportsAllDrives=True,
                 includeItemsFromAllDrives=True,
-            ).execute()
+            ).execute(num_retries=self.NUM_RETRIES)
             for payload in response.get("files", []):
                 yield DriveFile.from_api(payload)
             page_token = response.get("nextPageToken")
@@ -269,14 +272,14 @@ class GoogleDriveConnector(FileConnector):
         downloader = MediaIoBaseDownload(buffer, request)
         done = False
         while not done:
-            _, done = downloader.next_chunk()
+            _, done = downloader.next_chunk(num_retries=GoogleDriveConnector.NUM_RETRIES)
         return buffer.getvalue()
 
     # --- Changes ------------------------------------------------------------
 
     def get_start_page_token(self) -> str:
         """Where to start reading changes from: everything after now."""
-        response = self.drive_service.changes().getStartPageToken(supportsAllDrives=True).execute()
+        response = self.drive_service.changes().getStartPageToken(supportsAllDrives=True).execute(num_retries=self.NUM_RETRIES)
         return str(response.get("startPageToken") or "")
 
     def list_changes(self, page_token: str) -> DriveChangesPage:
@@ -292,7 +295,7 @@ class GoogleDriveConnector(FileConnector):
             supportsAllDrives=True,
             includeItemsFromAllDrives=True,
             spaces="drive",
-        ).execute()
+        ).execute(num_retries=self.NUM_RETRIES)
         changes = [
             DriveChange(
                 file_id=str(change.get("fileId") or ""),

{iatoolkit-2.60.1 → iatoolkit-2.60.2/src/iatoolkit.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 2.60.1
+Version: 2.60.2
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT