Add retry_delay in invoke method

Author: KurimuzonAkuma

pyrogram/methods/advanced/invoke.py

@@ -34,6 +34,7 @@ async def invoke(
         retries: int = Session.MAX_RETRIES,
         timeout: float = Session.WAIT_TIMEOUT,
         sleep_threshold: float = None,
+        retry_delay: float = Session.RETRY_DELAY,
         business_connection_id: str = None
     ):
         """Invoke raw Telegram functions.
@@ -54,15 +55,18 @@ async def invoke(
             query (``RawFunction``):
                 The API Schema function filled with proper arguments.
 
-            retries (``int``):
+            retries (``int``, *optional*):
                 Number of retries.
 
-            timeout (``float``):
+            timeout (``float`, *optional*`):
                 Timeout in seconds.
 
-            sleep_threshold (``float``):
+            sleep_threshold (``float``, *optional*):
                 Sleep threshold in seconds.
 
+            retry_delay (``float``, *optional*):
+                Retry delay in seconds on errors.
+
             business_connection_id (``str``, *optional*):
                 Unique identifier of the business connection.
 
@@ -92,10 +96,11 @@ async def invoke(
             query = raw.functions.InvokeWithTakeout(takeout_id=self.takeout_id, query=query)
 
         r = await session.invoke(
-            query, retries, timeout,
-            (sleep_threshold
-             if sleep_threshold is not None
-             else self.sleep_threshold)
+            query=query, retries=retries, timeout=timeout,
+            sleep_threshold=(
+                sleep_threshold if sleep_threshold is not None else self.sleep_threshold
+            ),
+            retry_delay=retry_delay
         )
 
         await self.fetch_peers(getattr(r, "users", []))

pyrogram/session/session.py

@@ -516,7 +516,8 @@ async def invoke(
         query: TLObject,
         retries: int = MAX_RETRIES,
         timeout: float = WAIT_TIMEOUT,
-        sleep_threshold: float = SLEEP_THRESHOLD
+        sleep_threshold: float = SLEEP_THRESHOLD,
+        retry_delay: float = RETRY_DELAY
     ):
         try:
             await asyncio.wait_for(self.is_started.wait(), self.WAIT_TIMEOUT)
@@ -554,7 +555,7 @@ async def invoke(
                     '[%s] Retrying "%s" due to: %s', attempt, query_name, str(e) or repr(e)
                 )
 
-                await asyncio.sleep(self.RETRY_DELAY)
+                await asyncio.sleep(retry_delay)
 
         raise TimeoutError(f'Failed to invoke "{query_name}" after {retries} retries')