Gmail Search all mailboxes: retry mechanism for first time

Packs/Gmail/Integrations/Gmail/Gmail.py

@@ -825,7 +825,6 @@ def get_millis_from_date(date, arg_name):
 
 
 def cutting_for_batches(list_accounts: list) -> List[list]:
-
     accounts: list = []
     rest_accounts: list = []
 
@@ -845,13 +844,11 @@ def cutting_for_batches(list_accounts: list) -> List[list]:
 
 
 def scheduled_commands_for_more_users(accounts: list, next_page_token: str) -> List[CommandResults]:
-
     accounts_batches = cutting_for_batches(accounts)
 
     command_results: list[CommandResults] = []
     args = copy.deepcopy(demisto.args())
     for batch in accounts_batches:
-
         args.update({'list_accounts': batch})
         command_results.append(
             CommandResults(
@@ -907,7 +904,6 @@ def get_mailboxes(max_results: int, users_next_page_token: str = None):
 
 
 def information_search_process(length_accounts: int, search_from: int | None, search_to: int | None) -> CommandResults:
-
     if search_from is None or search_to is None:
         readable_output = f'Searching the first {length_accounts} accounts'
         search_from = 0
@@ -1350,7 +1346,6 @@ def search_in_mailboxes(accounts: list[str], only_return_account_names: bool) ->
 
 
 def search_all_mailboxes():
-
     args = demisto.args()
     only_return_account_names = argToBoolean(args.get('show-only-mailboxes', 'true'))
     list_accounts = argToList(args.get('list_accounts', ''))
@@ -1392,7 +1387,8 @@ def search_all_mailboxes():
             search_all_mailboxes()
 
 
-def search_command(mailbox: str = None, only_return_account_names: bool = False) -> dict[str, Any] | None:
+def search_command(mailbox: str = None, only_return_account_names: bool = False, first_time: bool = True) -> dict[
+        str, Any] | None:
     """
     Searches for Gmail records of a specified Google user.
     """
@@ -1428,8 +1424,19 @@ def search_command(mailbox: str = None, only_return_account_names: bool = False)
                           include_spam_trash, has_attachments, only_return_account_names,
                           )
     except HttpError as err:
-        if only_return_account_names and err.status_code == 429:
+        if err.status_code == 500 and first_time:
+            # retry mechanism - try just one time more
+            demisto.debug(f'Gmail Integration: Got an error {err.status_code} for {user_id=}, trying again to search fot it')
+            search_command(mailbox, only_return_account_names, False)
+        elif (
+            err.status_code == 500
+            or (err.status_code == 429
+                and only_return_account_names)
+        ):
+            demisto.debug(f'Gmail Integration: Got another time the {err.status_code} error for {user_id=}, '
+                          f'continuing to the next user')
             return {'Mailbox': mailbox, 'Error': {'message': str(err.error_details), 'status_code': err.status_code}}
+
         raise
 
     # In case the user wants only account list without content.

Packs/Gmail/ReleaseNotes/1_3_25.md

@@ -0,0 +1,6 @@
+
+#### Integrations
+
+##### Gmail
+
+- Added support for retry mechanism in ***gmail-search*** command when getting an error due to limited resources on Google's side.

Packs/Gmail/pack_metadata.json

@@ -2,7 +2,7 @@
     "name": "Gmail",
     "description": "Gmail API and user management (This integration replaces the Gmail functionality in the GoogleApps API and G Suite integration).",
     "support": "xsoar",
-    "currentVersion": "1.3.24",
+    "currentVersion": "1.3.25",
     "author": "Cortex XSOAR",
     "url": "https://www.paloaltonetworks.com/cortex",
     "email": "",