increase attempt

nath1996rahul · Dec 9, 2022 · ea66b0f · ea66b0f
1 parent 65a03f2
commit ea66b0f
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/bot/helper/mirror_utils/upload_utils/gdriveTools.py b/bot/helper/mirror_utils/upload_utils/gdriveTools.py
@@ -138,7 +138,7 @@ def __getIdFromUrl(link: str):
         parsed = urlparse(link)
         return parse_qs(parsed.query)['id'][0]
 
-    @retry(wait=wait_exponential(multiplier=2, min=3, max=6), stop=stop_after_attempt(3),
+    @retry(wait=wait_exponential(multiplier=2, min=3, max=6), stop=stop_after_attempt(10),
            retry=retry_if_exception_type(Exception))
     def __set_permission(self, file_id):
         permissions = {
@@ -149,13 +149,13 @@ def __set_permission(self, file_id):
         }
         return self.__service.permissions().create(fileId=file_id, body=permissions, supportsAllDrives=True).execute()
 
-    @retry(wait=wait_exponential(multiplier=2, min=3, max=6), stop=stop_after_attempt(3),
+    @retry(wait=wait_exponential(multiplier=2, min=3, max=6), stop=stop_after_attempt(10),
            retry=retry_if_exception_type(Exception))
     def __getFileMetadata(self, file_id):
         return self.__service.files().get(fileId=file_id, supportsAllDrives=True,
                                           fields='name, id, mimeType, size').execute()
 
-    @retry(wait=wait_exponential(multiplier=2, min=3, max=6), stop=stop_after_attempt(3),
+    @retry(wait=wait_exponential(multiplier=2, min=3, max=6), stop=stop_after_attempt(10),
            retry=retry_if_exception_type(Exception))
     def __getFilesByFolderId(self, folder_id):
         page_token = None
@@ -272,7 +272,7 @@ def __upload_dir(self, input_directory, dest_id, user_id):
                 break
         return new_id
 
-    @retry(wait=wait_exponential(multiplier=2, min=3, max=6), stop=stop_after_attempt(3),
+    @retry(wait=wait_exponential(multiplier=2, min=3, max=6), stop=stop_after_attempt(10),
            retry=retry_if_exception_type(Exception))
     def __create_directory(self, directory_name, dest_id, user_id):
         # Change file name
@@ -292,7 +292,7 @@ def __create_directory(self, directory_name, dest_id, user_id):
         LOGGER.info("Created G-Drive Folder:\nName: {}\nID: {} ".format(file.get("name"), file_id))
         return file_id
 
-    @retry(wait=wait_exponential(multiplier=2, min=3, max=6), stop=stop_after_attempt(3),
+    @retry(wait=wait_exponential(multiplier=2, min=3, max=6), stop=stop_after_attempt(10),
            retry=(retry_if_exception_type(Exception)))
     def __upload_file(self, file_path, file_name, mime_type, dest_id, user_id):
         # Change file name
@@ -475,7 +475,7 @@ def __cloneFolder(self, name, local_path, folder_id, dest_id, user_id):
             if self.__is_cancelled:
                 break
 
-    @retry(wait=wait_exponential(multiplier=2, min=3, max=6), stop=stop_after_attempt(3),
+    @retry(wait=wait_exponential(multiplier=2, min=3, max=6), stop=stop_after_attempt(10),
            retry=retry_if_exception_type(Exception))
     def __copyFile(self, file_id, dest_id, file_name, user_id):
         # Change file name
@@ -913,7 +913,7 @@ def __download_folder(self, folder_id, path, folder_name):
             if self.__is_cancelled:
                 break
 
-    @retry(wait=wait_exponential(multiplier=2, min=3, max=6), stop=stop_after_attempt(3),
+    @retry(wait=wait_exponential(multiplier=2, min=3, max=6), stop=stop_after_attempt(10),
            retry=(retry_if_exception_type(Exception)))
     def __download_file(self, file_id, path, filename, mime_type):
         request = self.__service.files().get_media(fileId=file_id, supportsAllDrives=True)