code cleanup

This commit is contained in:
Crussader 2021-10-28 18:41:28 +04:00 committed by GitHub
parent bf134a5105
commit 3b207bb528
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 7 deletions

View File

@ -1,6 +1,7 @@
from .pool import NodePool
from .utils import ClientType
class PomiceEvent:
"""The base class for all events dispatched by a node.
Every event must be formatted within your bot's code as a listener.
@ -24,7 +25,7 @@ class TrackStartEvent(PomiceEvent):
name = "track_start"
def __init__(self, data: dict):
self.player = NodePool.get_node().get_player(int(data["guildId"]))
self.player = NodePool.get_player(int(data["guildId"]))
self.track = self.player._current
# on_pomice_track_start(player, track)
@ -41,7 +42,7 @@ class TrackEndEvent(PomiceEvent):
name = "track_end"
def __init__(self, data: dict):
self.player = NodePool.get_node().get_player(int(data["guildId"]))
self.player = NodePool.get_player(int(data["guildId"]))
self.track = self.player._ending_track
self.reason: str = data["reason"]
@ -49,17 +50,21 @@ class TrackEndEvent(PomiceEvent):
self.handler_args = self.player, self.track, self.reason
def __repr__(self) -> str:
return f"<Pomice.TrackEndEvent player={self.player} track_id={self.track.track_id} reason={self.reason}>"
return (
f"<Pomice.TrackEndEvent player={self.player} track_id={self.track.track_id} "
f"reason={self.reason}>"
)
class TrackStuckEvent(PomiceEvent):
"""Fired when a track is stuck and cannot be played. Returns the player
associated with the event along with the pomice.Track object to be further parsed by the end user.
associated with the event along with the pomice.Track object
to be further parsed by the end user.
"""
name = "track_stuck"
def __init__(self, data: dict):
self.player = NodePool.get_node().get_player(int(data["guildId"]))
self.player = NodePool.get_player(int(data["guildId"]))
self.track = self.player._ending_track
self.threshold: float = data["thresholdMs"]
@ -78,7 +83,7 @@ class TrackExceptionEvent(PomiceEvent):
name = "track_exception"
def __init__(self, data: dict):
self.player = NodePool.get_node().get_player(int(data["guildId"]))
self.player = NodePool.get_player(int(data["guildId"]))
self.track = self.player._ending_track
self.error: str = data["error"]
@ -91,7 +96,7 @@ class TrackExceptionEvent(PomiceEvent):
class WebSocketClosedPayload:
def __init__(self, data: dict):
self.guild = NodePool.get_node().get_player(int(data["guildId"]))._guild
self.guild = NodePool.get_player(int(data["guildId"]))._guild
self.code: int = data["code"]
self.reason: str = data["code"]
self.by_remote: bool = data["byRemote"]
@ -100,6 +105,7 @@ class WebSocketClosedPayload:
return f"<Pomice.WebSocketClosedPayload guild={self.guild!r} code={self.code!r} " \
f"reason={self.reason!r} by_remote={self.by_remote!r}>"
class WebSocketClosedEvent(PomiceEvent):
"""Fired when a websocket connection to a node has been closed.
Returns the reason and the error code.