majjo/Vocard
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Handle Playlist type in track search results

Browse Source 
Added a check to handle cases where the returned tracks are a Playlist object, ensuring the correct list of tracks is used for app command choices.
This commit is contained in:
Choco
2025-08-11 17:44:46 +08:00
parent f6c25118e1
commit e7aa431ae2
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
cogs/basic.py
View File

@@ -93,8 +93,15 @@ class Basic(commands.Cog):
node = voicelink.NodePool.get_node()
if not node:
return []
tracks: list[voicelink.Track] = await node.get_tracks(current, requester=interaction.user)
return [app_commands.Choice(name=truncate_string(f"🎵 {track.author} - {track.title}", 100), value=truncate_string(f"{track.author} - {track.title}", 100)) for track in tracks] if tracks else []
if not tracks:
return []
if isinstance(tracks, voicelink.Playlist):
tracks = tracks.tracks
return [app_commands.Choice(name=truncate_string(f"🎵 {track.author} - {track.title}", 100), value=truncate_string(f"{track.author} - {track.title}", 100)) for track in tracks]
history = {track["identifier"]: track for track_id in reversed(await get_user(interaction.user.id, "history")) if (track := voicelink.decode(track_id))["uri"]}
return [app_commands.Choice(name=truncate_string(f"🕒 {track['author']} - {track['title']}", 100), value=track['uri']) for track in history.values() if len(track['uri']) <= 100][:25]