Try to handle google upload error 429 tpp many requests (EOF...)

with google grive api lib Signed-off-by: anasty17 <[email protected]>
anasty17 · Dec 28, 2023 · 86234e1 · 86234e1
1 parent 02154c8
commit 86234e1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/bot/helper/mirror_utils/gdrive_utils/download.py b/bot/helper/mirror_utils/gdrive_utils/download.py
@@ -115,7 +115,7 @@ def _download_file(self, file_id, path, filename, mime_type):
             try:
                 self.status, done = downloader.next_chunk()
             except HttpError as err:
-                if err.resp.status in [500, 502, 503, 504] and retries < 10:
+                if err.resp.status in [500, 502, 503, 504, 429] and retries < 10:
                     retries += 1
                     continue
                 if err.resp.get("content-type", "").startswith("application/json"):

diff --git a/bot/helper/mirror_utils/gdrive_utils/upload.py b/bot/helper/mirror_utils/gdrive_utils/upload.py
@@ -176,7 +176,7 @@ def _upload_file(self, file_path, file_name, mime_type, dest_id, is_dir=True):
             try:
                 self.status, response = drive_file.next_chunk()
             except HttpError as err:
-                if err.resp.status in [500, 502, 503, 504] and retries < 10:
+                if err.resp.status in [500, 502, 503, 504, 429] and retries < 10:
                     retries += 1
                     continue
                 if err.resp.get("content-type", "").startswith("application/json"):