matthew/enhanced-discord.py
1
0
Fork 0
Code Issues 4 Pull Requests 21 Projects Releases Wiki Activity

Handle voice websocket closure if it's a successful close.

Browse Source
This commit is contained in:
Rapptz 2016-06-10 21:45:42 -04:00
parent bde266bbb9
commit fe5c369fe9
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
discord/voice_client.py
View File

@ -57,7 +57,7 @@ log = logging.getLogger(__name__)
from . import utils, opus from . import utils, opus
from .gateway import * from .gateway import *
from .errors import ClientException, InvalidArgument from .errors import ClientException, InvalidArgument, ConnectionClosed
class StreamPlayer(threading.Thread): class StreamPlayer(threading.Thread):
def __init__(self, stream, encoder, connected, player, after, **kwargs): def __init__(self, stream, encoder, connected, player, after, **kwargs):
@ -233,7 +233,13 @@ class VoiceClient:
Reads from the voice websocket while connected. Reads from the voice websocket while connected.
""" """
while self._connected.is_set(): while self._connected.is_set():
yield from self.ws.poll_event() try:
yield from self.ws.poll_event()
except ConnectionClosed as e:
if e.code == 1000:
break
else:
raise
@asyncio.coroutine @asyncio.coroutine
def disconnect(self): def disconnect(self):