Parker
382c422629
- Split custom sources into helper functions - Add proper logging and handling - Fix LoadError embed messsage
91 lines
2.7 KiB
Python
91 lines
2.7 KiB
Python
import datetime
|
|
import discord
|
|
import requests
|
|
from typing import Tuple, Optional
|
|
from requests.exceptions import JSONDecodeError
|
|
|
|
from utils.config import BOT_COLOR, LOG
|
|
|
|
|
|
async def load(
|
|
headers: dict,
|
|
query: str,
|
|
user: discord.User,
|
|
) -> Tuple[Optional[dict], Optional[discord.Embed]]:
|
|
"""
|
|
Get the playlist info from the Apple Music API
|
|
"""
|
|
playlist_id = query.split("/playlist/")[1].split("/")[1]
|
|
try:
|
|
# Get all of the tracks in the playlist (limit at 100)
|
|
response = requests.get(
|
|
f"https://api.music.apple.com/v1/catalog/us/playlists/{playlist_id}/tracks?limit=100",
|
|
headers=headers,
|
|
)
|
|
|
|
if response.status_code == 404:
|
|
embed = discord.Embed(
|
|
title="Playlist Not Found",
|
|
description=(
|
|
"The playlist could not be found as the provided link is"
|
|
" invalid. Please try again."
|
|
),
|
|
color=BOT_COLOR,
|
|
)
|
|
return None, embed
|
|
|
|
if response.status_code == 401:
|
|
LOG.error(
|
|
"Could not authorize with Apple Music API. Likely need to"
|
|
" restart the bot."
|
|
)
|
|
return None, None
|
|
|
|
response.raise_for_status()
|
|
playlist = response.json()
|
|
|
|
# Get the general playlist info (name, artwork)
|
|
response = requests.get(
|
|
f"https://api.music.apple.com/v1/catalog/us/playlists/{playlist_id}",
|
|
headers=headers,
|
|
)
|
|
|
|
response.raise_for_status()
|
|
# Unpack the playlist info
|
|
playlist_info = response.json()
|
|
name = playlist_info["data"][0]["attributes"]["name"]
|
|
num_tracks = len(playlist["data"])
|
|
except IndexError:
|
|
LOG.error("Failed unpacking Apple Music playlist info")
|
|
return None, None
|
|
except (JSONDecodeError, requests.HTTPError):
|
|
LOG.error("Failed making request to Apple Music API")
|
|
return None, None
|
|
|
|
# Extract artwork URL, if available
|
|
artwork_url = (
|
|
playlist_info["data"][0]["attributes"]
|
|
.get("artwork", {})
|
|
.get("url", None)
|
|
)
|
|
if artwork_url:
|
|
artwork_url = artwork_url.replace("{w}x{h}", "300x300")
|
|
|
|
embed = discord.Embed(
|
|
title="Playlist Queued",
|
|
description=(
|
|
f"**{name}**\n` {num_tracks} ` tracks\n\nQueued by: {user.mention}"
|
|
),
|
|
color=BOT_COLOR,
|
|
)
|
|
|
|
# Add small alert if the playlist is the max size
|
|
if len(playlist["data"]) == 100:
|
|
embed.description += (
|
|
"\n\n*This playlist is longer than the 100 song"
|
|
" maximum. Only the first 100 songs will be"
|
|
" queued.*"
|
|
)
|
|
|
|
return playlist, embed
|