Add deezer module
This commit is contained in:
92
bot/modules/deezer/song.py
Normal file
92
bot/modules/deezer/song.py
Normal file
@@ -0,0 +1,92 @@
|
||||
from attrs import define
|
||||
|
||||
from .driver import DeezerDriver
|
||||
|
||||
|
||||
@define
|
||||
class SongItem:
|
||||
name: str
|
||||
id: int
|
||||
artist: str
|
||||
preview_url: str | None
|
||||
thumbnail: str
|
||||
|
||||
@classmethod
|
||||
def from_deezer(cls, song_item: dict):
|
||||
return cls(
|
||||
name=song_item['title'],
|
||||
id=song_item['id'],
|
||||
artist=song_item['artist']['name'],
|
||||
preview_url=song_item.get('preview'),
|
||||
thumbnail=song_item['album']['cover_medium']
|
||||
)
|
||||
|
||||
@property
|
||||
def full_name(self):
|
||||
return f"{self.artist} - {self.name}"
|
||||
|
||||
def __str__(self):
|
||||
return self.full_name
|
||||
|
||||
|
||||
@define
|
||||
class FullSongItem:
|
||||
name: str
|
||||
id: int
|
||||
artists: list[str]
|
||||
preview_url: str | None
|
||||
duration: int
|
||||
thumbnail: str
|
||||
|
||||
@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'])
|
||||
)
|
||||
|
||||
@property
|
||||
def all_artists(self):
|
||||
return ', '.join(self.artists)
|
||||
|
||||
@property
|
||||
def full_name(self):
|
||||
return f"{self.all_artists} - {self.name}"
|
||||
|
||||
def __str__(self):
|
||||
return self.full_name
|
||||
|
||||
|
||||
@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: int) -> FullSongItem | None:
|
||||
r = await self.driver.reverse_get_track(song_id)
|
||||
|
||||
if r is None:
|
||||
return None
|
||||
|
||||
return await FullSongItem.from_deezer(r)
|
||||
Reference in New Issue
Block a user