68 lines
1.9 KiB
Python
68 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)
|