nambrogi
/
pomice
mirror of https://github.com/cloudwithax/pomice.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

code cleanup

This commit is contained in:
Crussader 2021-10-28 18:53:02 +04:00 committed by GitHub
parent 46f1aaa698
commit c8cd3fe599
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
pomice/player.py
View File

@ -15,7 +15,7 @@ from discord.ext import commands, tasks
from . import events from . import events
from .enums import SearchType from .enums import SearchType
from .events import PomiceEvent, TrackStartEvent from .events import PomiceEvent, TrackEndEvent, TrackStartEvent
from .exceptions import TrackInvalidPosition from .exceptions import TrackInvalidPosition
from .filters import Filter from .filters import Filter
from .objects import Track from .objects import Track
@ -188,6 +188,10 @@ class Player(VoiceProtocol):
async def _dispatch_event(self, data: dict): async def _dispatch_event(self, data: dict):
event_type = data.get("type") event_type = data.get("type")
event: PomiceEvent = getattr(events, event_type)(data) event: PomiceEvent = getattr(events, event_type)(data)
if isinstance(event, TrackEndEvent):
self._current = None
event.dispatch(self._bot) event.dispatch(self._bot)
if isinstance(event, TrackStartEvent): if isinstance(event, TrackStartEvent):
@ -231,7 +235,15 @@ class Player(VoiceProtocol):
async def destroy(self): async def destroy(self):
"""Disconnects a player and destroys the player instance.""" """Disconnects a player and destroys the player instance."""
try:
await self.disconnect() await self.disconnect()
except AttributeError:
# 'NoneType' has no attribute '_get_voice_client_key' raised by self.cleanup() ->
# assume we're already disconnected and cleaned up
assert self.channel is None \
and not self.is_connected \
and self.guild.id not in self._node._players
await self._node.send(op="destroy", guildId=str(self.guild.id)) await self._node.send(op="destroy", guildId=str(self.guild.id))
async def play(self, track: Track, *, start_position: int = 0) -> Track: async def play(self, track: Track, *, start_position: int = 0) -> Track: