|
24 | 24 | import logging
|
25 | 25 | from abc import ABC, abstractmethod
|
26 | 26 | from typing import (TYPE_CHECKING, Any, Dict, Generic, List, Optional, TypeVar,
|
27 |
| - Union) |
| 27 | + Union, cast) |
28 | 28 |
|
29 | 29 | from .common import MISSING
|
30 | 30 | from .errors import InvalidTrack, LoadError
|
31 | 31 | from .events import Event, TrackLoadFailedEvent
|
32 |
| -from .server import AudioTrack |
| 32 | +from .server import AudioTrack, RawPlayer |
33 | 33 |
|
34 | 34 | if TYPE_CHECKING:
|
35 | 35 | from .client import Client
|
@@ -106,7 +106,7 @@ async def play_track(self,
|
106 | 106 | no_replace: bool = MISSING,
|
107 | 107 | volume: int = MISSING,
|
108 | 108 | pause: bool = MISSING,
|
109 |
| - **kwargs) -> Optional[Dict[str, Any]]: |
| 109 | + **kwargs) -> Optional[RawPlayer]: |
110 | 110 | """|coro|
|
111 | 111 |
|
112 | 112 | .. _player object: https://lavalink.dev/api/rest.html#Player
|
@@ -147,7 +147,7 @@ async def play_track(self,
|
147 | 147 |
|
148 | 148 | Returns
|
149 | 149 | -------
|
150 |
| - Optional[Dict[:class:`str`, Any]] |
| 150 | + Optional[:class:`RawPlayer`] |
151 | 151 | The updated `player object`_, or ``None`` if a request wasn't made due to an empty payload.
|
152 | 152 | """
|
153 | 153 | if track is MISSING or not isinstance(track, AudioTrack):
|
@@ -203,7 +203,8 @@ async def play_track(self,
|
203 | 203 | return
|
204 | 204 |
|
205 | 205 | self._next = track
|
206 |
| - return await self.node.update_player(guild_id=self._internal_id, encoded_track=playable_track, **options) |
| 206 | + response = await self.node.update_player(guild_id=self._internal_id, encoded_track=playable_track, **options) |
| 207 | + return cast(RawPlayer, response) |
207 | 208 |
|
208 | 209 | def cleanup(self):
|
209 | 210 | pass
|
@@ -240,7 +241,7 @@ async def _voice_state_update(self, data):
|
240 | 241 |
|
241 | 242 | async def _dispatch_voice_update(self):
|
242 | 243 | if {'sessionId', 'endpoint', 'token'} == self._voice_state.keys():
|
243 |
| - await self.node.update_player(guild_id=self._internal_id, voice_state=self._voice_state) |
| 244 | + await self.node.update_player(guild_id=self._internal_id, voice_state=self._voice_state) # type: ignore |
244 | 245 |
|
245 | 246 | @abstractmethod
|
246 | 247 | async def node_unavailable(self):
|
|
0 commit comments