advtech/selena
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

FEAT: Improved Spotify Controls adding Pause, Resume, Next, and Previous. Also Added a check for no devices

Browse Source
This commit is contained in:
Dan 2024-06-20 22:31:27 -04:00
parent 726ec91a79
commit ed66d6438f
1 changed files with 86 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

113
modules/spotify_module.py
View File

@ -5,7 +5,7 @@ from spotipy.oauth2 import SpotifyOAuth
import config
class SpotifyCog:
class SpotifyModule:
def __init__(self, bot):
self.bot = bot
self.sp = spotipy.Spotify(
@ -26,41 +26,100 @@ class SpotifyCog:
name="current_track", description="Get the currently playing song"
)
async def current_track(interaction: discord.Interaction):
current = self.sp.currently_playing()
if current is None or current["item"] is None:
await interaction.response.send_message(
"No song is currently playing"
)
else:
track = current["item"]
artist = ", ".join([artist["name"] for artist in track[
"artists"]])
await interaction.response.send_message(
f"Currently playing: {track['name']} by {artist} "
f"({track['album']['name']})"
)
await interaction.response.defer()
try:
current = self.sp.currently_playing()
if current is None or current["item"] is None:
await interaction.followup.send("No song is currently playing")
else:
track = current["item"]
artist = ", ".join([artist["name"] for artist in track["artists"]])
await interaction.followup.send(
f"Currently playing: {track['name']} by {artist} "
f"({track['album']['name']})"
)
except Exception as e:
await interaction.followup.send(f"An error occurred: {e}")
@app_commands.command(
name="play_track", description="Play a track by searching for it"
)
async def play_track(interaction: discord.Interaction, query: str):
results = self.sp.search(q=query, limit=1, type="track")
if not results["tracks"]["items"]:
await interaction.response.send_message("No results found")
return
await interaction.response.defer()
try:
results = self.sp.search(q=query, limit=1, type="track")
if not results["tracks"]["items"]:
await interaction.followup.send("No results found")
return
track = results["tracks"]["items"][0]
uri = track["uri"]
self.sp.start_playback(uris=[uri])
await interaction.response.send_message(
f"Now playing: {track['name']} by "
f"{', '.join([artist['name'] for artist in track['artists']])}"
f"({track['album']['name']})"
)
track = results["tracks"]["items"][0]
uri = track["uri"]
devices = self.sp.devices()
if not devices["devices"]:
await interaction.followup.send("No active devices found. Please open Spotify on a device.")
return
self.sp.start_playback(uris=[uri])
await interaction.followup.send(
f"Now playing: {track['name']} by "
f"{', '.join([artist['name'] for artist in track['artists']])} "
f"({track['album']['name']})"
)
except Exception as e:
await interaction.followup.send(f"An error occurred: {e}")
@app_commands.command(
name="pause", description="Pause the currently playing track"
)
async def pause(interaction: discord.Interaction):
await interaction.response.defer()
try:
self.sp.pause_playback()
await interaction.followup.send("Playback paused.")
except Exception as e:
await interaction.followup.send(f"An error occurred: {e}")
@app_commands.command(
name="resume", description="Resume the currently playing track"
)
async def resume(interaction: discord.Interaction):
await interaction.response.defer()
try:
self.sp.start_playback()
await interaction.followup.send("Playback resumed.")
except Exception as e:
await interaction.followup.send(f"An error occurred: {e}")
@app_commands.command(
name="next", description="Skip to the next track"
)
async def next_track(interaction: discord.Interaction):
await interaction.response.defer()
try:
self.sp.next_track()
await interaction.followup.send("Skipped to the next track.")
except Exception as e:
await interaction.followup.send(f"An error occurred: {e}")
@app_commands.command(
name="previous", description="Go back to the previous track"
)
async def previous_track(interaction: discord.Interaction):
await interaction.response.defer()
try:
self.sp.previous_track()
await interaction.followup.send("Returned to the previous track.")
except Exception as e:
await interaction.followup.send(f"An error occurred: {e}")
self.bot.tree.add_command(current_track)
self.bot.tree.add_command(play_track)
self.bot.tree.add_command(pause)
self.bot.tree.add_command(resume)
self.bot.tree.add_command(next_track)
self.bot.tree.add_command(previous_track)
async def setup(bot):
SpotifyCog(bot)
SpotifyModule(bot)