70 lines
1.9 KiB
Python
70 lines
1.9 KiB
Python
from attrs import define
|
|
|
|
from .driver import DeezerDriver
|
|
|
|
from ..common.song import BaseSongItem
|
|
|
|
|
|
@define
|
|
class SongItem(BaseSongItem):
|
|
@classmethod
|
|
def from_deezer(cls, song_item: dict):
|
|
return cls(
|
|
name=song_item["title"],
|
|
id=str(song_item["id"]),
|
|
artists=[song_item["artist"]["name"]],
|
|
preview_url=song_item.get("preview"),
|
|
thumbnail=song_item["album"]["cover_medium"],
|
|
)
|
|
|
|
|
|
@define
|
|
class FullSongItem(BaseSongItem):
|
|
duration: int
|
|
track_dict: dict
|
|
|
|
@classmethod
|
|
async def from_deezer(cls, song_item: dict):
|
|
if song_item.get("results"):
|
|
song_item = song_item["results"]
|
|
|
|
return cls(
|
|
name=song_item["SNG_TITLE"],
|
|
id=song_item["SNG_ID"],
|
|
artists=[artist["ART_NAME"] for artist in song_item["ARTISTS"]],
|
|
preview_url=(
|
|
song_item.get("MEDIA").get("HREF")
|
|
if type(song_item.get("MEDIA")) is dict
|
|
and song_item.get("MEDIA").get("TYPE") == "preview"
|
|
else None
|
|
),
|
|
thumbnail=f"https://e-cdns-images.dzcdn.net/images/cover/"
|
|
f'{song_item["ALB_PICTURE"]}/320x320.jpg',
|
|
duration=int(song_item["DURATION"]),
|
|
track_dict=song_item,
|
|
)
|
|
|
|
|
|
@define
|
|
class Songs(object):
|
|
driver: DeezerDriver
|
|
|
|
async def search(self, query: str, limit: int = 30) -> list[SongItem] | None:
|
|
r = await self.driver.search(query, limit=limit)
|
|
|
|
if r is None:
|
|
return None
|
|
|
|
return [SongItem.from_deezer(item) for item in r]
|
|
|
|
async def search_one(self, query: str) -> SongItem | None:
|
|
return (await self.search(query, limit=1) or [None])[0]
|
|
|
|
async def from_id(self, song_id: str) -> FullSongItem | None:
|
|
r = await self.driver.reverse_get_track(song_id)
|
|
|
|
if r is None:
|
|
return None
|
|
|
|
return await FullSongItem.from_deezer(r)
|