Update voice client main ws references when reconnecting.

This commit is contained in:
Rapptz 2016-06-01 05:20:22 -04:00
parent e0a91df32b
commit 8638ff4414
2 changed files with 6 additions and 0 deletions

View File

@ -199,6 +199,8 @@ class DiscordWebSocket(websockets.client.WebSocketClientProtocol):
ws._dispatch = client.dispatch
ws.gateway = gateway
client.connection._update_references(ws)
log.info('Created websocket connected to {}'.format(gateway))
if not resume:
yield from ws.identify()

View File

@ -107,6 +107,10 @@ class ConnectionState:
def _remove_voice_client(self, guild_id):
self._voice_clients.pop(guild_id, None)
def _update_references(self, ws):
for vc in self.voice_clients:
vc.main_ws = ws
@property
def servers(self):
return self._servers.values()