67 lines
2.3 KiB
Python
67 lines
2.3 KiB
Python
|
import discord
|
||
|
|
from discord import app_commands
|
||
|
|
import spotipy
|
||
|
|
from spotipy.oauth2 import SpotifyOAuth
|
||
|
|
import config
|
||
|
|
|
||
|
|
|
||
|
|
class SpotifyCog:
|
||
|
|
def __init__(self, bot):
|
||
|
|
self.bot = bot
|
||
|
|
self.sp = spotipy.Spotify(
|
||
|
|
auth_manager=SpotifyOAuth(
|
||
|
|
client_id=config.SPOTIPY_CLIENT_ID,
|
||
|
|
client_secret=config.SPOTIPY_CLIENT_SECRET,
|
||
|
|
redirect_uri=config.SPOTIPY_REDIRECT_URI,
|
||
|
|
scope=(
|
||
|
|
"user-library-read user-read-playback-state "
|
||
|
|
"user-modify-playback-state user-read-currently-playing"
|
||
|
|
)
|
||
|
|
)
|
||
|
|
)
|
||
|
|
self.add_commands()
|
||
|
|
|
||
|
|
def add_commands(self):
|
||
|
|
@app_commands.command(
|
||
|
|
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']})"
|
||
|
|
)
|
||
|
|
|
||
|
|
@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
|
||
|
|
|
||
|
|
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']})"
|
||
|
|
)
|
||
|
|
|
||
|
|
self.bot.tree.add_command(current_track)
|
||
|
|
self.bot.tree.add_command(play_track)
|
||
|
|
|
||
|
|
|
||
|
|
async def setup(bot):
|
||
|
|
SpotifyCog(bot)
|