velvox
/
litecord
mirror of https://gitlab.com/litecord/litecord.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

gateway.websocket: decrease amount of public methods

This commit is contained in:
Luna 2018-12-04 16:56:53 -03:00
parent 6b35c73ad6
commit d87ff940f6
1 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
litecord/gateway/websocket.py
View File

@ -237,7 +237,7 @@ class GatewayWebsocket:
for guild_id in guild_ids for guild_id in guild_ids
] ]
async def guild_dispatch(self, unavailable_guilds: List[Dict[str, Any]]): async def _guild_dispatch(self, unavailable_guilds: List[Dict[str, Any]]):
"""Dispatch GUILD_CREATE information.""" """Dispatch GUILD_CREATE information."""
# Users don't get asynchronous guild dispatching. # Users don't get asynchronous guild dispatching.
@ -255,7 +255,7 @@ class GatewayWebsocket:
await self.dispatch('GUILD_CREATE', guild) await self.dispatch('GUILD_CREATE', guild)
async def user_ready(self): async def _user_ready(self) -> dict:
"""Fetch information about users in the READY packet. """Fetch information about users in the READY packet.
This part of the API is completly undocumented. This part of the API is completly undocumented.
@ -300,7 +300,7 @@ class GatewayWebsocket:
uready = {} uready = {}
if not self.state.bot: if not self.state.bot:
# user, fetch info # user, fetch info
uready = await self.user_ready() uready = await self._user_ready()
await self.dispatch('READY', {**{ await self.dispatch('READY', {**{
'v': 6, 'v': 6,
@ -314,7 +314,7 @@ class GatewayWebsocket:
}, **uready}) }, **uready})
# async dispatch of guilds # async dispatch of guilds
self.ext.loop.create_task(self.guild_dispatch(guilds)) self.ext.loop.create_task(self._guild_dispatch(guilds))
async def _check_shards(self, shard, user_id): async def _check_shards(self, shard, user_id):
current_shard, shard_count = shard current_shard, shard_count = shard
@ -338,7 +338,11 @@ class GatewayWebsocket:
if current_shard > shard_count: if current_shard > shard_count:
raise InvalidShard('Shard count > Total shards') raise InvalidShard('Shard count > Total shards')
async def _guild_ids(self): async def _guild_ids(self) -> list:
"""Get a list of Guild IDs that are tied to this connection.
The implementation is shard-aware.
"""
guild_ids = await self.user_storage.get_user_guilds( guild_ids = await self.user_storage.get_user_guilds(
self.state.user_id self.state.user_id
) )
@ -359,8 +363,8 @@ class GatewayWebsocket:
async def subscribe_all(self): async def subscribe_all(self):
"""Subscribe to all guilds, DM channels, and friends. """Subscribe to all guilds, DM channels, and friends.
Subscribing to channels is already handled Note: subscribing to channels is already handled
by GuildDispatcher.sub by GuildDispatcher.sub
""" """
user_id = self.state.user_id user_id = self.state.user_id