From 017efcb1cc2f3e2a74b73e1bd1b15b683c1fb152 Mon Sep 17 00:00:00 2001
From: Alexander-D-Karpov <alexandr.d.karpov@gmail.com>
Date: Thu, 11 Jan 2024 23:23:29 +0300
Subject: [PATCH] fixed music handling

---
 akarpov/music/services/youtube.py | 83 +++++++++++++++++--------------
 1 file changed, 45 insertions(+), 38 deletions(-)

diff --git a/akarpov/music/services/youtube.py b/akarpov/music/services/youtube.py
index 229ec01..49bc709 100644
--- a/akarpov/music/services/youtube.py
+++ b/akarpov/music/services/youtube.py
@@ -101,6 +101,46 @@ def download_from_youtube_link(link: str, user_id: int) -> Song:
             chapter_path = path.split(".")[0] + chapters[i][2] + ".mp3"
             info = search_all_platforms(chapters[i][2])
             audio.export(chapter_path, format="mp3")
+            if not info["album_image"].startswith("/"):
+                r = requests.get(info["album_image"])
+                img_pth = str(
+                    settings.MEDIA_ROOT
+                    + f"/{info['album_image'].split('/')[-1]}_{str(randint(100, 999))}"
+                )
+                with open(img_pth, "wb") as f:
+                    f.write(r.content)
+
+                im = Image.open(img_pth)
+                im.save(str(f"{img_pth}.png"))
+                img_pth = f"{img_pth}.png"
+            else:
+                img_pth = info["album_image"]
+
+            if "genre" in info:
+                song = load_track(
+                    chapter_path,
+                    img_pth,
+                    user_id,
+                    info["artists"],
+                    info["album_name"],
+                    chapters[i][2],
+                    genre=info["genre"],
+                )
+            else:
+                song = load_track(
+                    chapter_path,
+                    img_pth,
+                    user_id,
+                    info["artists"],
+                    info["album_name"],
+                    chapters[i][2],
+                )
+            os.remove(chapter_path)
+    else:
+        print(f"[processing] loading {title}")
+
+        info = search_all_platforms(title)
+        if not info["album_image"].startswith("/"):
             r = requests.get(info["album_image"])
             img_pth = str(
                 settings.MEDIA_ROOT
@@ -113,46 +153,13 @@ def download_from_youtube_link(link: str, user_id: int) -> Song:
             im.save(str(f"{img_pth}.png"))
 
             os.remove(img_pth)
-            if "genre" in info:
-                song = load_track(
-                    chapter_path,
-                    f"{img_pth}.png",
-                    user_id,
-                    info["artists"],
-                    info["album_name"],
-                    chapters[i][2],
-                    genre=info["genre"],
-                )
-            else:
-                song = load_track(
-                    chapter_path,
-                    f"{img_pth}.png",
-                    user_id,
-                    info["artists"],
-                    info["album_name"],
-                    chapters[i][2],
-                )
-            os.remove(chapter_path)
-    else:
-        print(f"[processing] loading {title}")
-
-        info = search_all_platforms(title)
-        r = requests.get(info["album_image"])
-        img_pth = str(
-            settings.MEDIA_ROOT
-            + f"/{info['album_image'].split('/')[-1]}_{str(randint(100, 999))}"
-        )
-        with open(img_pth, "wb") as f:
-            f.write(r.content)
-
-        im = Image.open(img_pth)
-        im.save(str(f"{img_pth}.png"))
-
-        os.remove(img_pth)
+            img_pth = f"{img_pth}.png"
+        else:
+            img_pth = info["album_image"]
         if "genre" in info:
             song = load_track(
                 path,
-                f"{img_pth}.png",
+                img_pth,
                 user_id,
                 info["artists"],
                 info["album_name"],
@@ -162,7 +169,7 @@ def download_from_youtube_link(link: str, user_id: int) -> Song:
         else:
             song = load_track(
                 path,
-                f"{img_pth}.png",
+                img_pth,
                 user_id,
                 info["artists"],
                 info["album_name"],