mirrors/discord.py
1
0
Fork 0
mirror of https://github.com/Rapptz/discord.py.git synced 2025-04-19 15:36:02 +00:00
Code Issues Projects Releases Wiki Activity

Add shard_connect_timeout parameter for AutoShardedClient

Browse Source
This commit is contained in:
Rapptz 2024-01-07 19:45:29 -05:00
parent 9a1b91dc9f
commit cb3ea9b889
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
discord/shard.py
View File

@ -326,6 +326,11 @@ class AutoShardedClient(Client):
------------
shard_ids: Optional[List[:class:`int`]]
An optional list of shard_ids to launch the shards with.
shard_connect_timeout: Optional[:class:`float`]
The maximum number of seconds to wait before timing out when launching a shard.
Defaults to 180 seconds.
.. versionadded:: 2.4
"""
if TYPE_CHECKING:
@ -334,6 +339,8 @@ class AutoShardedClient(Client):
def __init__(self, *args: Any, intents: Intents, **kwargs: Any) -> None:
kwargs.pop('shard_id', None)
self.shard_ids: Optional[List[int]] = kwargs.pop('shard_ids', None)
self.shard_connect_timeout: Optional[float] = kwargs.pop('shard_connect_timeout', 180.0)
super().__init__(*args, intents=intents, **kwargs)
if self.shard_ids is not None:
@ -411,7 +418,7 @@ class AutoShardedClient(Client):
async def launch_shard(self, gateway: yarl.URL, shard_id: int, *, initial: bool = False) -> None:
try:
coro = DiscordWebSocket.from_client(self, initial=initial, gateway=gateway, shard_id=shard_id)
ws = await asyncio.wait_for(coro, timeout=180.0)
ws = await asyncio.wait_for(coro, timeout=self.shard_connect_timeout)
except Exception:
_log.exception('Failed to connect for shard_id: %s. Retrying...', shard_id)
await asyncio.sleep(5.0)