fixed music processing

This commit is contained in:
Alexander Karpov 2023-12-18 03:50:51 +03:00
parent 51f3a6bb57
commit 75d787f460

View File

@ -62,8 +62,6 @@ def load_file_meta(track: int, user_id: int) -> str:
que.delete() que.delete()
return "" return ""
print("Start downloading")
filename = f"_{str(randint(10000, 9999999))}" filename = f"_{str(randint(10000, 9999999))}"
orig_path = f"{settings.MEDIA_ROOT}/{filename}.mp3" orig_path = f"{settings.MEDIA_ROOT}/{filename}.mp3"
album = track.albums[0] album = track.albums[0]
@ -76,13 +74,10 @@ def load_file_meta(track: int, user_id: int) -> str:
except NotFoundError: except NotFoundError:
img_pth = None img_pth = None
print("Downloaded file")
try: try:
lyrics = track.get_lyrics("LRC").fetch_lyrics() lyrics = track.get_lyrics("LRC").fetch_lyrics()
except NotFoundError: except NotFoundError:
lyrics = "" lyrics = ""
print("Start loading")
song = load_track( song = load_track(
orig_path, orig_path,
img_pth, img_pth,
@ -96,7 +91,6 @@ def load_file_meta(track: int, user_id: int) -> str:
explicit=track.explicit, explicit=track.explicit,
track_source=track.track_source, track_source=track.track_source,
) )
print("Loaded")
if os.path.exists(orig_path): if os.path.exists(orig_path):
os.remove(orig_path) os.remove(orig_path)
if os.path.exists(img_pth): if os.path.exists(img_pth):
@ -134,10 +128,11 @@ def update_album_info(album: AlbumModel) -> None:
} }
authors = [] authors = []
if search_album.artists: if search_album.artists:
authors = [ for x in search_album.artists:
Author.objects.get_or_create(name=x.name)[0] try:
for x in search_album.artists authors.append(Author.objects.get_or_create(name=x.name)[0])
] except Author.MultipleObjectsReturned:
authors.append(Author.objects.filter(name=x.name).first())
album.authors.set(authors) album.authors.set(authors)
album.meta = data album.meta = data
image_path = str(settings.MEDIA_ROOT + f"/_{str(randint(10000, 99999))}.png") image_path = str(settings.MEDIA_ROOT + f"/_{str(randint(10000, 99999))}.png")
@ -155,8 +150,6 @@ def update_author_info(author: Author) -> None:
client = login() client = login()
search = client.search(author.name, type_="artist") # type: Search search = client.search(author.name, type_="artist") # type: Search
print("Loading author info " + author.name)
if search.artists: if search.artists:
search_artist = search.artists.results[0] search_artist = search.artists.results[0]
data = { data = {
@ -171,7 +164,6 @@ def update_author_info(author: Author) -> None:
if not search_artist.cover: if not search_artist.cover:
author.save() author.save()
return return
print("Downloading image")
search_artist.cover.download(filename=image_path) search_artist.cover.download(filename=image_path)
with open(image_path, "rb") as f: with open(image_path, "rb") as f:
author.image = File(f, name=image_path.split("/")[-1]) author.image = File(f, name=image_path.split("/")[-1])