This commit is contained in:
codey 2024-08-11 17:40:10 -04:00
parent 7baa1c5d25
commit 72abef1092

View File

@ -11,6 +11,15 @@ class ValidArtistSearchRequest(BaseModel):
a: str a: str
class ValidAlbumDetailRequest(BaseModel):
"""
- **a**: artist name
- **a2**: album/release name (as sourced from here/LastFM)
"""
a: str
a2: str
class LastFM(FastAPI): class LastFM(FastAPI):
"""Last.FM Endpoints""" """Last.FM Endpoints"""
def __init__(self, app: FastAPI, util, constants): # pylint: disable=super-init-not-called def __init__(self, app: FastAPI, util, constants): # pylint: disable=super-init-not-called
@ -22,6 +31,7 @@ class LastFM(FastAPI):
self.endpoints = { self.endpoints = {
"get_artist_by_name": self.artist_by_name_handler, "get_artist_by_name": self.artist_by_name_handler,
"get_artist_albums": self.artist_album_handler, "get_artist_albums": self.artist_album_handler,
"get_release": self.release_detail_handler,
#tbd #tbd
} }
@ -30,6 +40,7 @@ class LastFM(FastAPI):
async def artist_by_name_handler(self, data: ValidArtistSearchRequest): async def artist_by_name_handler(self, data: ValidArtistSearchRequest):
""" """
/get_artist_by_name/
Get artist info Get artist info
""" """
artist = data.a.strip() artist = data.a.strip()
@ -53,6 +64,7 @@ class LastFM(FastAPI):
async def artist_album_handler(self, data: ValidArtistSearchRequest): async def artist_album_handler(self, data: ValidArtistSearchRequest):
""" """
/get_artist_albums/
Get artist's albums/releases Get artist's albums/releases
""" """
artist = data.a.strip() artist = data.a.strip()
@ -72,8 +84,36 @@ class LastFM(FastAPI):
continue continue
seen_release_titles.append(release_title.lower()) seen_release_titles.append(release_title.lower())
album_result_out.append(release) album_result_out.append(release)
return { return {
'success': True, 'success': True,
'result': album_result_out 'result': album_result_out
} }
async def release_detail_handler(self, data: ValidAlbumDetailRequest):
"""
/get_release/
Get details of a particular release by an artist
"""
artist = data.a.strip()
release = data.a2.strip()
if not artist or not release:
return {
'err': True,
'errorText': 'Invalid request'
}
release_result = await self.lastfm.get_release(artist=artist, album=release)
ret_obj = {
'id': release_result.get('id'),
'artists': release_result.get('artists'),
'title': release_result.get('title'),
'summary': release_result.get('summary'),
'tracks': release_result.get('tracks')
}
return {
'success': True,
'result': ret_obj
}