mirror
/
auto-archiver
kopia lustrzana https://github.com/bellingcat/auto-archiver
1
0
Forkuj 0
Kod Zgłoszenia Packages Projekty Wydania Wiki Aktywność

expands Sheets API retries for really large spreadsheets

pull/341/head
msramalho 2025-07-05 18:29:33 +01:00
rodzic a8c1ef3912
commit 7234eda85f
Nie znaleziono w bazie danych klucza dla tego podpisu
1 zmienionych plików z 5 dodań i 1 usunięć
Pokaż statystyki Ściągnij plik aktualizacji Ściągnij plik porównania Expand all files Collapse all files

6
src/auto_archiver/modules/gsheet_feeder_db/gsheet_feeder_db.py
Wyświetl plik

@ -32,6 +32,10 @@ class GsheetsFeederDB(Feeder, Database):
if not self.sheet and not self.sheet_id: 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.") 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: def open_sheet(self) -> gspread.Spreadsheet:
if self.sheet: if self.sheet:
return self.gsheets_client.open(self.sheet) return self.gsheets_client.open(self.sheet)
@ -40,7 +44,7 @@ class GsheetsFeederDB(Feeder, Database):
@retry( @retry(
wait_exponential_multiplier=1, wait_exponential_multiplier=1,
stop_max_attempt_number=5, stop_max_attempt_number=6,
) )
def enumerate_sheets(self, sheet) -> Iterator[gspread.Worksheet]: def enumerate_sheets(self, sheet) -> Iterator[gspread.Worksheet]:
for worksheet in sheet.worksheets(): for worksheet in sheet.worksheets():