sanspie/akarpov
1
1
Fork 0
mirror of https://github.com/Alexander-D-Karpov/akarpov synced 2024-11-24 08:53:44 +03:00
Code Issues Packages Projects Releases Wiki Activity

updated music fetching

Browse Source
This commit is contained in:
Alexander Karpov 2024-04-06 23:48:03 +03:00
parent 00668b6f18
commit b148a3d591
1 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
akarpov/music/services/yandex.py
View File

@ -77,18 +77,24 @@ def load_file_meta(track: int, user_id: int) -> str:
def load_url(link: str, user_id: int): def load_url(link: str, user_id: int):
client = login() client = login()
obj_id = link.split("/")[-1]
obj_id = obj_id.split("?")[0]
try:
obj_id = int(obj_id)
except ValueError:
print("Invalid link")
return None
if "/playlists/" in link: if "/playlists/" in link:
author = link.split("/")[4] author = link.split("/")[4]
playlist_id = link.split("/")[-1]
playlist = client.users_playlists(int(playlist_id), author) # type: Playlist playlist = client.users_playlists(obj_id, author) # type: Playlist
for track in playlist.fetch_tracks(): for track in playlist.fetch_tracks():
tasks.load_ym_file_meta.apply_async( tasks.load_ym_file_meta.apply_async(
kwargs={"track": track.track.id, "user_id": user_id} kwargs={"track": track.track.id, "user_id": user_id}
) )
elif "/album/" in link: elif "/album/" in link:
album_id = link.split("/")[-1] album = client.albums_with_tracks(obj_id)
album = client.albums_with_tracks(int(album_id))
tracks = [] tracks = []
for volume in album.volumes: for volume in album.volumes:
for track in volume: for track in volume:
@ -99,14 +105,12 @@ def load_url(link: str, user_id: int):
kwargs={"track": track.track.id, "user_id": user_id} kwargs={"track": track.track.id, "user_id": user_id}
) )
elif "/artist/" in link: elif "/artist/" in link:
author = link.split("/")[-1] artist = client.artists(obj_id)[0]
artist = client.artists([int(author)])[0]
for track in artist.popular_tracks: for track in artist.popular_tracks:
tasks.load_ym_file_meta.apply_async( tasks.load_ym_file_meta.apply_async(
kwargs={"track": track.id, "user_id": user_id} kwargs={"track": track.id, "user_id": user_id}
) )
else: else:
track_id = link.split("/")[-1]
tasks.load_ym_file_meta.apply_async( tasks.load_ym_file_meta.apply_async(
kwargs={"track": track_id, "user_id": user_id} kwargs={"track": obj_id, "user_id": user_id}
) )