Merge d4aaa05eb3 into 43a6e8d779

akarpov/music/apps.py

@@ -7,8 +7,10 @@ class MusicConfig(AppConfig):
     name = "akarpov.music"
 
     def ready(self):
-        import akarpov.music.signals  # noqa F401
+        try:
-
+            import akarpov.music.signals  # noqa F401
+        except ImportError:
+            pass
         try:
             from akarpov.music.tasks import start_next_song
 

akarpov/music/migrations/0012_album_meta_author_albums_author_meta.py

@@ -1,27 +0,0 @@
-# Generated by Django 4.2.8 on 2023-12-17 22:22
-
-from django.db import migrations, models
-
-
-class Migration(migrations.Migration):
-    dependencies = [
-        ("music", "0011_alter_playlist_private_userlistenhistory"),
-    ]
-
-    operations = [
-        migrations.AddField(
-            model_name="album",
-            name="meta",
-            field=models.JSONField(blank=True, null=True),
-        ),
-        migrations.AddField(
-            model_name="author",
-            name="albums",
-            field=models.ManyToManyField(related_name="authors", to="music.album"),
-        ),
-        migrations.AddField(
-            model_name="author",
-            name="meta",
-            field=models.JSONField(blank=True, null=True),
-        ),
-    ]

akarpov/music/models.py

@@ -10,8 +10,6 @@
 class Author(BaseImageModel, ShortLinkModel):
     name = models.CharField(max_length=200)
     link = models.URLField(blank=True)
-    meta = models.JSONField(blank=True, null=True)
-    albums = models.ManyToManyField("Album", related_name="authors")
 
     def get_absolute_url(self):
         return reverse("music:author", kwargs={"slug": self.slug})
@@ -23,7 +21,6 @@ def __str__(self):
 class Album(BaseImageModel, ShortLinkModel):
     name = models.CharField(max_length=200)
     link = models.URLField(blank=True)
-    meta = models.JSONField(blank=True, null=True)
 
     def get_absolute_url(self):
         return reverse("music:album", kwargs={"slug": self.slug})

akarpov/music/services/db.py

@@ -78,7 +78,7 @@ def load_track(
         str(
             slugify(
                 GoogleTranslator(source="auto", target="en").translate(
-                    f"{song.name} {' '.join([x.name for x in authors])}",
+                    f"{song.name} {' '.join([x.name for x in song.authors])}",
                     target_language="en",
                 )
             )
@@ -124,9 +124,9 @@ def load_track(
                     data=f.read(),
                 )
             )
-    if "release" in kwargs and kwargs["release"]:
+    if "release" in kwargs:
         tag.tags.add(TORY(text=kwargs["release"]))
-    if "genre" in kwargs and kwargs["genre"]:
+    if "genre" in kwargs:
         tag.tags.add(TCON(text=kwargs["genre"]))
     tag.save()
 

akarpov/music/services/spotify.py

@@ -2,6 +2,8 @@
 from django.conf import settings
 from spotipy.oauth2 import SpotifyClientCredentials
 
+from akarpov.music.services.yandex import search_ym
+
 
 def login() -> spotipy.Spotify:
     if not settings.MUSIC_SPOTIFY_ID or not settings.MUSIC_SPOTIFY_SECRET:
@@ -45,7 +47,15 @@ def get_track_info(name: str) -> dict:
     # try to get genre
     sp = login()
     genres = sp.album(res["album"]["external_urls"]["spotify"])["genres"]
-    if genres:
+    if not genres:
+        ym_info = search_ym(info["artist"] + " " + info["title"])
+        if ym_info and "genre" in ym_info:
+            info["genre"] = ym_info["genre"]
+        else:
+            genres = sp.artist(res["artists"][0]["external_urls"]["spotify"])["genres"]
+            if genres:
+                info["genre"] = genres[0]
+    else:
         info["genre"] = genres[0]
 
     return info

akarpov/music/services/yandex.py

@@ -2,13 +2,11 @@
 from random import randint
 
 from django.conf import settings
-from django.core.files import File
+from django.utils.text import slugify
 from yandex_music import Client, Playlist, Search, Track
-from yandex_music.exceptions import NotFoundError
 
 from akarpov.music import tasks
-from akarpov.music.models import Album as AlbumModel
+from akarpov.music.models import Song, SongInQue
-from akarpov.music.models import Author, Song, SongInQue
 from akarpov.music.services.db import load_track
 
 
@@ -45,7 +43,7 @@ def search_ym(name: str):
     return info
 
 
-def load_file_meta(track: int, user_id: int) -> str:
+def load_file_meta(track: int, user_id: int):
     que = SongInQue.objects.create()
     client = login()
     track = client.tracks(track)[0]  # type: Track
@@ -57,32 +55,19 @@ def load_file_meta(track: int, user_id: int) -> str:
             name=track.title, album__name=track.albums[0].title
         ):
             que.delete()
-            return str(sng.first())
+            return sng.first()
     except IndexError:
         que.delete()
-        return ""
+        return
 
-    print("Start downloading")
+    filename = slugify(f"{track.artists[0].name} - {track.title}")
-
-    filename = f"_{str(randint(10000, 9999999))}"
     orig_path = f"{settings.MEDIA_ROOT}/{filename}.mp3"
     album = track.albums[0]
 
     track.download(filename=orig_path, codec="mp3")
     img_pth = str(settings.MEDIA_ROOT + f"/_{str(randint(10000, 99999))}.png")
 
-    try:
+    track.download_cover(filename=img_pth)
-        track.download_cover(filename=img_pth)
-    except NotFoundError:
-        img_pth = None
-
-    print("Downloaded file")
-
-    try:
-        lyrics = track.get_lyrics("LRC").fetch_lyrics()
-    except NotFoundError:
-        lyrics = ""
-    print("Start loading")
     song = load_track(
         orig_path,
         img_pth,
@@ -92,11 +77,7 @@ def load_file_meta(track: int, user_id: int) -> str:
         track.title,
         release=album.release_date,
         genre=album.genre,
-        lyrics=lyrics,
-        explicit=track.explicit,
-        track_source=track.track_source,
     )
-    print("Loaded")
     if os.path.exists(orig_path):
         os.remove(orig_path)
     if os.path.exists(img_pth):
@@ -115,65 +96,3 @@ def load_playlist(link: str, user_id: int):
         tasks.load_ym_file_meta.apply_async(
             kwargs={"track": track.track.id, "user_id": user_id}
         )
-
-
-def update_album_info(album: AlbumModel) -> None:
-    client = login()
-    search = client.search(album.name, type_="album")  # type: Search
-
-    if search.albums:
-        search_album = search.albums.results[0]
-        data = {
-            "name": search_album.title,
-            "tracks": search_album.track_count,
-            "explicit": search_album.explicit,
-            "year": search_album.year,
-            "genre": search_album.genre,
-            "description": search_album.description,
-            "type": search_album.type,
-        }
-        authors = []
-        if search_album.artists:
-            authors = [
-                Author.objects.get_or_create(name=x.name)[0]
-                for x in search_album.artists
-            ]
-        album.authors.set(authors)
-        album.meta = data
-        image_path = str(settings.MEDIA_ROOT + f"/_{str(randint(10000, 99999))}.png")
-        if not search_album.cover_uri:
-            album.save()
-            return
-        search_album.download_cover(filename=image_path)
-        with open(image_path, "rb") as f:
-            album.image = File(f, name=image_path.split("/")[-1])
-            album.save()
-        os.remove(image_path)
-
-
-def update_author_info(author: Author) -> None:
-    client = login()
-    search = client.search(author.name, type_="artist")  # type: Search
-
-    print("Loading author info " + author.name)
-
-    if search.artists:
-        search_artist = search.artists.results[0]
-        data = {
-            "name": search_artist.name,
-            "description": search_artist.description,
-            "genres": search_artist.genres,
-        }
-
-        author.meta = data
-
-        image_path = str(settings.MEDIA_ROOT + f"/_{str(randint(10000, 99999))}.png")
-        if not search_artist.cover:
-            author.save()
-            return
-        print("Downloading image")
-        search_artist.cover.download(filename=image_path)
-        with open(image_path, "rb") as f:
-            author.image = File(f, name=image_path.split("/")[-1])
-            author.save()
-        os.remove(image_path)

akarpov/music/signals.py

@@ -4,7 +4,6 @@
 from django.dispatch import receiver
 
 from akarpov.music.models import Album, Author, PlaylistSong, Song, SongUserRating
-from akarpov.music.services.yandex import update_album_info, update_author_info
 
 
 @receiver(post_delete, sender=Song)
@@ -17,13 +16,15 @@ def auto_delete_file_on_delete(sender, instance, **kwargs):
 @receiver(post_save, sender=Author)
 def author_create(sender, instance, created, **kwargs):
     if created:
-        update_author_info(instance)
+        # TODO: add logic to retrieve author info here
+        return
 
 
 @receiver(post_save, sender=Album)
 def album_create(sender, instance, created, **kwargs):
     if created:
-        update_album_info(instance)
+        # TODO: add logic to retrieve author info here
+        return
 
 
 @receiver(post_save)