8000 Fix missing imports and general code smells. · devoxin/Lavalink.py@64a538e · GitHub
[go: up one dir, main page]

Skip to content

Commit 64a538e

Browse files
committed
Fix missing imports and general code smells.
1 parent f37dadd commit 64a538e

File tree

4 files changed

+11
-9
lines changed

4 files changed

+11
-9
lines changed

lavalink/__init__.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
__author__ = 'Devoxin'
55
__license__ = 'MIT'
66
__copyright__ = 'Copyright 2017-present Devoxin'
7-
__version__ = '5.8.0'
7+
__version__ = '5.8.1'
88

99

1010
from typing import Type

lavalink/abc.py

+7-6
Original file line numberDiff line numberDiff line change
@@ -24,12 +24,12 @@
2424
import logging
2525
from abc import ABC, abstractmethod
2626
from typing import (TYPE_CHECKING, Any, Dict, Generic, List, Optional, TypeVar,
27-
Union)
27+
Union, cast)
2828

2929
from .common import MISSING
3030
from .errors import InvalidTrack, LoadError
3131
from .events import Event, TrackLoadFailedEvent
32-
from .server import AudioTrack
32+
from .server import AudioTrack, RawPlayer
3333

3434
if TYPE_CHECKING:
3535
from .client import Client
@@ -106,7 +106,7 @@ async def play_track(self,
106106
no_replace: bool = MISSING,
107107
volume: int = MISSING,
108108
pause: bool = MISSING,
109-
**kwargs) -> Optional[Dict[str, Any]]:
109+
**kwargs) -> Optional[RawPlayer]:
110110
"""|coro|
111111
112112
.. _player object: https://lavalink.dev/api/rest.html#Player
@@ -147,7 +147,7 @@ async def play_track(self,
147147
148148
Returns
149149
-------
150-
Optional[Dict[:class:`str`, Any]]
150+
Optional[:class:`RawPlayer`]
151151
The updated `player object`_, or ``None`` if a request wasn't made due to an empty payload.
152152
"""
153153
if track is MISSING or not isinstance(track, AudioTrack):
@@ -203,7 +203,8 @@ async def play_track(self,
203203
return
204204

205205
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)
207208

208209
def cleanup(self):
209210
pass
@@ -240,7 +241,7 @@ async def _voice_state_update(self, data):
240241

241242
async def _dispatch_voice_update(self):
242243
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
244245

245246
@abstractmethod
246247
async def node_unavailable(self):

lavalink/playermanager.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -250,7 +250,8 @@ def create(self,
250250
raise ClientError('No available nodes!')
251251

252252
id_int = int(guild_id)
253-
self.players[id_int] = player = cls(id_int, best_node)
253+
player = cls(id_int, best_node)
254+
self.players[id_int] = player # type: ignore
254255
_log.debug('Created player with GuildId %d on node \'%s\'', id_int, best_node.name)
255256
return player
256257

lavalink/server.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626
"""
2727
from enum import Enum as _Enum
2828
from typing import (TYPE_CHECKING, Any, Dict, List, Literal, Optional, Type,
29-
TypedDict, TypeVar, Union)
29+
TypedDict, TypeVar, Union, cast)
3030

3131
from .errors import InvalidTrack
3232

0 commit comments

Comments
 (0)
0