Skip to content

Commit

Permalink
style: move payload data logs to debug log type
Browse files Browse the repository at this point in the history
  • Loading branch information
aaronhnsy committed Jun 15, 2022
1 parent b1a6476 commit c8677b9
Show file tree
Hide file tree
Showing 2 changed files with 20 additions and 14 deletions.
30 changes: 18 additions & 12 deletions slate/node.py
Original file line number Diff line number Diff line change
Expand Up @@ -257,11 +257,11 @@ async def _listen(self) -> None:
break

else:
payload = message.json(loads=self._json_loads)
payload = self._json_loads(message.data)
asyncio.create_task(
self._receive_payload(
payload["op"],
data=payload["d"] if "d" in payload else payload
payload.pop("op", "missing-op"),
data=payload.get("d", payload),
)
)

Expand All @@ -275,28 +275,34 @@ async def _receive_payload(
LOGGER.debug(f"Node '{self.identifier}' received a payload with op '{op}'.\nData: {data}")

if op in (1, "stats"):
self._stats = None # Stats(data)
self._stats = None # TODO: Implement
return

player = self._players.get(int(data["guild_id"] if self.provider is Provider.OBSIDIAN else data["guildId"]))
guild_id = int(data.get("guild_id", data.get("guildId")))
player = self._players.get(guild_id)

if op in (4, "event"):

if not player:
LOGGER.warning(f"Node '{self.identifier}' received a player event for a guild without a player.\nData: {data}")
LOGGER.warning(
f"Node '{self.identifier}' received an event for guild '{guild_id}' which does not "
f"have a player."
)
else:
player._dispatch_event(data)
return

if op in (5, "playerUpdate"):
elif op in (5, "playerUpdate"):

if not player:
LOGGER.warning(f"Node '{self.identifier}' received a player update for a guild without a player.\nData: {data}")
LOGGER.warning(
f"Node '{self.identifier}' received an update for guild '{guild_id}' which does not "
f"have a player."
)
else:
player._update_state(data)
return

LOGGER.warning(f"Node '{self.identifier}' received a payload with an unknown op '{op}'.\nData: {data}")
else:
LOGGER.warning(f"Node '{self.identifier}' received a payload with an unhandled op code: '{op}'.")

async def _send_payload(
self,
Expand Down Expand Up @@ -334,7 +340,7 @@ async def _send_payload(
_json = _json.decode("utf-8")

await self._websocket.send_str(_json)
LOGGER.debug(f"Node '{self.identifier}' sent a payload with op '{op}'.\nData: {data}")
# LOGGER.debug(f"Node '{self.identifier}' sent a payload with op '{op}'.\nData: {data}")

# REST

Expand Down
4 changes: 2 additions & 2 deletions slate/player.py
Original file line number Diff line number Diff line change
Expand Up @@ -147,13 +147,13 @@ def _dispatch_event(
_type = data["type"]

if not (event := OBSIDIAN_EVENT_MAPPING.get(_type) if self._node.provider is Provider.OBSIDIAN else LAVALINK_EVENT_MAPPING.get(_type)):
LOGGER.error(f"Player '{self.channel.guild.id}' received an event with an unknown type '{_type}'.\nData: {data}")
LOGGER.error(f"Player '{self.channel.guild.id}' received an event with an unknown type: '{_type}'.")
return

event = event(data)

LOGGER.info(f"Player '{self.channel.guild.id}' dispatched an event with type '{_type}'.\nData: {data}")
self.bot.dispatch(f"slate_{event.type.lower()}", self, event)
LOGGER.info(f"Player '{self.channel.guild.id}' dispatched '{_type}' event.")

def _update_state(
self,
Expand Down

0 comments on commit c8677b9

Please sign in to comment.