You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
65 lines
1.9 KiB
65 lines
1.9 KiB
from urllib.parse import urlparse, parse_qs
|
|
|
|
import regex
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class URLType:
|
|
Song = "song"
|
|
Album = "album"
|
|
Playlist = "playlist"
|
|
Artist = "artist"
|
|
|
|
|
|
class AppleMusicURL(BaseModel):
|
|
url: str
|
|
storefront: str
|
|
type: str
|
|
id: str
|
|
|
|
@classmethod
|
|
def parse_url(cls, url: str):
|
|
if not regex.match(r"https://music.apple.com/(.{2})/(song|album|playlist|artist).*/(pl.*|\d*)", url):
|
|
return None
|
|
parsed_url = urlparse(url)
|
|
paths = parsed_url.path.split("/")
|
|
storefront = paths[1]
|
|
url_type = paths[2]
|
|
match url_type:
|
|
case URLType.Song:
|
|
url_id = paths[-1]
|
|
return Song(url=url, storefront=storefront, id=url_id, type=URLType.Song)
|
|
case URLType.Album:
|
|
if not parsed_url.query:
|
|
url_id = paths[-1]
|
|
return Album(url=url, storefront=storefront, id=url_id, type=URLType.Album)
|
|
else:
|
|
url_query = parse_qs(parsed_url.query)
|
|
if url_query.get("i"):
|
|
url_id = url_query.get("i")[0]
|
|
return Song(url=url, storefront=storefront, id=url_id, type=URLType.Song)
|
|
else:
|
|
url_id = paths[-1]
|
|
return Album(url=url, storefront=storefront, id=url_id, type=URLType.Album)
|
|
case URLType.Artist:
|
|
url_id = paths[-1]
|
|
return Artist(url=url, storefront=storefront, id=url_id, type=URLType.Artist)
|
|
case URLType.Playlist:
|
|
url_id = paths[-1]
|
|
return Playlist(url=url, storefront=storefront, id=url_id, type=URLType.Playlist)
|
|
|
|
|
|
class Song(AppleMusicURL):
|
|
...
|
|
|
|
|
|
class Album(AppleMusicURL):
|
|
...
|
|
|
|
|
|
class Playlist(AppleMusicURL):
|
|
...
|
|
|
|
|
|
class Artist(AppleMusicURL):
|
|
...
|
|
|