old playlist imports

playlist/views.py

@@ -11,7 +11,7 @@ from datetime import UTC, datetime
 from collections import OrderedDict
 from dotenv import load_dotenv
 from .models import Track, Artist, Album, Vote, Background, Profile, Playlist
-from .utils import from_json, get_unvoted
+from .utils import track_from_json, get_unvoted
 from .spotify import spt
 from .tasks import get_banter, get_and_dither_image
 
@@ -28,6 +28,8 @@ class VoteView(LoginRequiredMixin, View):
     def get(self, request, undo=None):
         (track, skipped) = get_unvoted(request.user)
         if track and not track.album.image:
+            if not track.album.image_url:
+                get_and_dither_image("https://picsum.photos/256", track.album, "image")
             get_and_dither_image(track.album.image_url, track.album, "image")
         return TemplateResponse(request, "Vote.html", {"track": track, "undo": undo, "skipped": skipped})
     def post(self, request):
@@ -86,7 +88,7 @@ class NominateView(LoginRequiredMixin, View):
                 get_banter.delay(track.id)
                 return TemplateResponse(request, "Nominate.html", {
                     "message_type": "warning",
-                    "message": "Hey een klassieker.", 
+                    "message": f"Hey een klassieker, die stond al in {track.from_playlist}. Geen probleem, we proberen hem nog een keer dit jaar!", 
                     "message_title": str(track)
                 })
             return TemplateResponse(request, "Nominate.html", {
@@ -101,31 +103,8 @@ class NominateView(LoginRequiredMixin, View):
                 print(e)
                 return TemplateResponse(request, "Nominate.html", {"message_type": "error", "message": "Fout bij het ophalen van het liedje."})
 
-            # Album artists
-            album_artists = []
-            for artist in json["album"]["artists"]:
-                album_artists.append(from_json(Artist, artist))
-            del json["album"]["artists"]
-
-            # Album
-            album = from_json(Album, json["album"])
-            album.image_url = json["album"]["images"][0]["url"]
-            album.save()
-            del json["album"]
-            for artist in album_artists:
-                album.artists.add(artist)
-
-            # Track artists
-            track_artists = []
-            for artist in json["artists"]:
-                track_artists.append(from_json(Artist, artist))
-            del json["artists"]
-
-            # Track
-            track = from_json(Track, json)
+            track = track_from_json(json)
             track.nominated_by = request.user
-            for artist in track_artists:
-                track.artists.add(artist)
             track.save()
             dup = track.is_duplicate_of()
             if dup:
@@ -135,8 +114,6 @@ class NominateView(LoginRequiredMixin, View):
                     "message": "Die hadden we al knul.",
                     "message_title": str(dup)
                 })
-            track.album = album
-            track.save()
             get_banter.delay(track.id)
             profile.quota -= 1
             profile.save()
@@ -246,7 +223,6 @@ class PlaylistListView(LoginRequiredMixin, View):
             return TemplateResponse(request, "PlaylistList.html")
             return self.get(request, count=count)
 
-
 class PlaylistDetailView(LoginRequiredMixin, View):
     def get(self, request, count=None):
         context = {}
@@ -281,3 +257,34 @@ class SpotifyCallbackView(LoginRequiredMixin, View):
         else:
             url = spt.get_oauth_redirect()
             return HttpResponseRedirect(url)
+
+class PlaylistImportView(LoginRequiredMixin, View):
+    def get(self, request):
+        return HttpResponseRedirect('/settings/')
+    def post(self, request):
+        print("...")
+        playlist_link = self.request.POST['playlist_link']
+        m = re.match(r"^https:\/\/open.spotify.com\/playlist\/([a-zA-Z0-9]+)\??", playlist_link)
+        playlist_id = m.group(1)
+        print(playlist_id)
+        json = spt.get_playlist_info(playlist_id)
+        tracks = json["tracks"]["items"]
+        playlist_name = json["name"]
+
+        for track in tracks:
+            track_id = track["track"]["id"]
+            try:
+                json = spt.get_song_info(track_id)
+                track = track_from_json(json)
+                dup = track.is_duplicate_of()
+                if dup:
+                    print(f"{track} hadden we al")
+                    track.delete()
+                else:
+                    track.old = True
+                    track.from_playlist = playlist_name
+                    track.save()
+            except Exception as e:
+                print(e)
+
+        return self.get(request)