Added track number

- Refactoring code
- Remove Poetry
- Added track number

service/provider.py

@@ -0,0 +1,57 @@
+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"]
+        )
+    )