From 7234eda85f352b50114b3c21073bca2b8ff2af03 Mon Sep 17 00:00:00 2001 From: msramalho <19508417+msramalho@users.noreply.github.com> Date: Sat, 5 Jul 2025 18:29:33 +0100 Subject: [PATCH] expands Sheets API retries for really large spreadsheets --- .../modules/gsheet_feeder_db/gsheet_feeder_db.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/auto_archiver/modules/gsheet_feeder_db/gsheet_feeder_db.py b/src/auto_archiver/modules/gsheet_feeder_db/gsheet_feeder_db.py index cb2051c..83738ac 100644 --- a/src/auto_archiver/modules/gsheet_feeder_db/gsheet_feeder_db.py +++ b/src/auto_archiver/modules/gsheet_feeder_db/gsheet_feeder_db.py @@ -32,6 +32,10 @@ class GsheetsFeederDB(Feeder, Database): if not self.sheet and not self.sheet_id: raise ValueError("You need to define either a 'sheet' name or a 'sheet_id' in your manifest.") + @retry( + wait_exponential_multiplier=1, + stop_max_attempt_number=6, + ) def open_sheet(self) -> gspread.Spreadsheet: if self.sheet: return self.gsheets_client.open(self.sheet) @@ -40,7 +44,7 @@ class GsheetsFeederDB(Feeder, Database): @retry( wait_exponential_multiplier=1, - stop_max_attempt_number=5, + stop_max_attempt_number=6, ) def enumerate_sheets(self, sheet) -> Iterator[gspread.Worksheet]: for worksheet in sheet.worksheets():