feat: Add find autocomplete.

This commit is contained in:
Lemon4ksan
2025-01-25 19:57:55 +03:00
parent 6a20ab11d1
commit 85f7ee6c6c
5 changed files with 157 additions and 64 deletions

View File

@@ -121,7 +121,7 @@ class PlayButton(Button, VoiceExtension):
current_player = None
if guild['current_player']:
current_player = await self.get_player_message(interaction, guild['current_player'])
current_player = await self.get_menu_message(interaction, guild['current_player'])
if current_player and interaction.message:
logging.debug(f"Deleting interaction message {interaction.message.id}: current player {current_player.id} found")
@@ -130,7 +130,7 @@ class PlayButton(Button, VoiceExtension):
await interaction.respond(response_message, delete_after=15)
class ListenView(View):
def __init__(self, item: Track | Album | Artist | Playlist | list[Track], *items: Item, timeout: float | None = None, disable_on_timeout: bool = False):
def __init__(self, item: Track | Album | Artist | Playlist | list[Track], *items: Item, timeout: float | None = 3600, disable_on_timeout: bool = False):
super().__init__(*items, timeout=timeout, disable_on_timeout=disable_on_timeout)
logging.debug(f"Creating view for type: '{type(item).__name__}'")

View File

@@ -140,7 +140,7 @@ class LyricsButton(Button, VoiceExtension):
class MenuView(View, VoiceExtension):
def __init__(self, ctx: ApplicationContext | Interaction, *items: Item, timeout: float | None = 3600, disable_on_timeout: bool = True):
def __init__(self, ctx: ApplicationContext | Interaction | RawReactionActionEvent, *items: Item, timeout: float | None = 3600, disable_on_timeout: bool = False):
View.__init__(self, *items, timeout=timeout, disable_on_timeout=disable_on_timeout)
VoiceExtension.__init__(self, None)
if not ctx.guild_id:
@@ -167,7 +167,7 @@ class MenuView(View, VoiceExtension):
self.add_item(self.next_button)
self.add_item(self.shuffle_button)
if len(cast(VoiceChannel, self.ctx.channel).members) > 2:
if isinstance(self.ctx, RawReactionActionEvent) or len(cast(VoiceChannel, self.ctx.channel).members) > 2:
self.like_button.disabled = True
elif likes and current_track and str(current_track['id']) in [str(like.id) for like in likes]:
self.like_button.style = ButtonStyle.success
@@ -178,4 +178,18 @@ class MenuView(View, VoiceExtension):
self.add_item(self.like_button)
self.add_item(self.lyrics_button)
return self
return self
async def on_timeout(self) -> None:
logging.debug('Menu timed out...')
if not self.ctx.guild_id:
return
if self.guild['current_player']:
self.db.update(self.ctx.guild_id, {'current_player': None, 'previous_tracks': []})
message = await self.get_menu_message(self.ctx, self.guild['current_player'])
if message:
await message.delete()
logging.debug('Successfully deleted menu message')
else:
logging.debug('No menu message found')