58 lines
1.8 KiB
Python
58 lines
1.8 KiB
Python
import itertools
|
|
from typing import Any
|
|
|
|
from api import shazam
|
|
from models.track import TrackInfo, Album
|
|
|
|
|
|
async def __extract_data(data: dict[str, Any]) -> dict[str, Any]:
|
|
track_data = data["track"]
|
|
sections_data = track_data["sections"]
|
|
items = list(itertools.chain(*[x["metadata"] for x in sections_data if "metadata" in x]))
|
|
|
|
def album_data(t: str) -> str | int:
|
|
for item in items:
|
|
if item["title"] == t:
|
|
return item["text"]
|
|
|
|
def seek_track(which: str, ad: dict[str, Any]) -> dict[str, Any]:
|
|
album_tracks = ad["data"][0]["relationships"]["tracks"]["data"]
|
|
for track in album_tracks:
|
|
if which == track["attributes"]["name"]:
|
|
return track["attributes"]
|
|
return {}
|
|
|
|
title = track_data["title"]
|
|
artist = track_data["subtitle"]
|
|
album_id = int(track_data['albumadamid'])
|
|
album_name = album_data("Album")
|
|
album_release_year = album_data("Released")
|
|
album_data = seek_track(title, await shazam.album(album_id=album_id))
|
|
|
|
return {
|
|
"title": title,
|
|
"artist": artist,
|
|
"track_number": album_data["trackNumber"],
|
|
"album": {
|
|
"id": album_id,
|
|
"name": album_name,
|
|
"released": album_release_year
|
|
}
|
|
}
|
|
|
|
|
|
async def search_song(song_file: str) -> TrackInfo:
|
|
raw_data: dict[str, Any] = await shazam.recognize(song_file)
|
|
track_data = await __extract_data(data=raw_data)
|
|
album_data = track_data["album"]
|
|
return TrackInfo(
|
|
title=track_data["title"],
|
|
artist=track_data["artist"],
|
|
track_number=track_data["track_number"],
|
|
album=Album(
|
|
album_id=album_data["id"],
|
|
name=album_data["name"],
|
|
released=album_data["released"]
|
|
)
|
|
)
|