first backend frontend split

with docker compose file
nextjs something something
also includes migrations

backend/playlist/models/__init__.py

@@ -0,0 +1,6 @@
+from .track import Track
+from .artist import Artist
+from .album import Album
+from .vote import Vote
+from .profile import Profile, Background
+from .playlist import Playlist

backend/playlist/models/album.py

@@ -0,0 +1,13 @@
+from django.db import models
+from .artist import Artist
+
+class Album(models.Model):
+    def __str__(self):
+        return self.name
+    id = models.CharField(primary_key=True, max_length=128)
+    name = models.CharField(max_length=255)
+    artists = models.ManyToManyField(Artist)
+    album_type = models.CharField(max_length=255)
+    total_tracks = models.PositiveSmallIntegerField(null=True)
+    image = models.ImageField(upload_to="albums/")
+    image_url = models.URLField(null=True)

backend/playlist/models/artist.py

@@ -0,0 +1,9 @@
+from django.db import models
+
+class Artist(models.Model):
+    def __str__(self):
+        return self.name
+    id = models.CharField(primary_key=True, max_length=128)
+    name = models.CharField(max_length=255)
+    popularity = models.SmallIntegerField(null=True, blank=True)
+    genres = models.JSONField(null=True)

backend/playlist/models/playlist.py

@@ -0,0 +1,8 @@
+from django.db import models
+from .track import Track
+
+class Playlist(models.Model):
+	def __str__(self):
+		return self.name
+	name = models.CharField(max_length=255)
+	tracks = models.ManyToManyField(Track)

backend/playlist/models/profile.py

@@ -0,0 +1,145 @@
+from django.db import models
+from django.contrib.auth.models import User
+from .track import Track
+from django.db.models.signals import post_save
+from django.dispatch import receiver
+from django.utils import timezone
+from datetime import datetime, timedelta, UTC
+from ..spotify import spt
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+NOMINATIONS_PER_DAY = 10
+MAX_QUOTA = 22
+
+class Background(models.Model):
+    def __str__(self):
+        return self.name
+    name = models.CharField(max_length=128)
+    image = models.ImageField(upload_to="wallpapers/", null=True, blank=True)
+    repeat = models.BooleanField(default=False)
+    cover = models.BooleanField(default=False)
+    color = models.CharField(max_length=32, blank=True)
+
+class Profile(models.Model):
+    def __str__(self):
+        return str(self.user)
+    user = models.OneToOneField(User, on_delete=models.CASCADE)
+    background = models.ForeignKey(Background, null=True, on_delete=models.SET_NULL)
+    last_voted = models.ForeignKey(Track, null=True, on_delete=models.SET_NULL)
+
+    quota = models.FloatField(default=10.0)
+    quota_last_updated = models.DateTimeField(default=timezone.now, null=False)
+
+    spt_access_token = models.CharField(max_length=256, default="", blank=True, null=True)
+    spt_refresh_token = models.CharField(max_length=256, default="", blank=True, null=True)
+    spt_expires = models.DateTimeField(default=None, null=True)
+
+    @property
+    def quota_display(self):
+       return int(self.quota)
+
+    def update_quota(self):
+        elapsed = timezone.now() - self.quota_last_updated
+        if elapsed.total_seconds() > 300:
+            noms_per_second = NOMINATIONS_PER_DAY/(60*60*24)
+            self.quota += noms_per_second * elapsed.total_seconds()
+            self.quota = min(self.quota, MAX_QUOTA)
+            print(f"{self.user.username}: Er zijn {elapsed.total_seconds()}s voorbij, je nieuwe quota is {self.quota}")
+            self.quota_last_updated = timezone.now()
+
+    @property
+    def can_nominate(self):
+        """Is this user allowed to nominate at this point in time?"""
+        end_date = os.getenv('DATE_NOM_END')
+        if self.user.is_superuser:
+            return (True, None)
+        if end_date:
+            end_date = datetime.fromisoformat(end_date)
+            if end_date <= datetime.now(tz=UTC):
+                return (False, "over")
+        if self.quota < 1:
+            return (False, "quota")
+        return (True, None)
+
+    @property
+    def can_vote(self):
+        """Is this user allowed to vote at this point in time?"""
+        end_date = os.getenv('DATE_VOTE_END')
+        if self.user.is_superuser or not end_date:
+            return True
+        end_date = datetime.fromisoformat(end_date)
+        if end_date <= datetime.now(tz=UTC):
+            return False
+        return True
+
+    @property
+    def vote_end(self):
+        return os.getenv('DATE_VOTE_END')
+
+    @property
+    def nominate_end(self):
+        return os.getenv('DATE_NOM_END')
+
+    def init_spt(self, code):
+        """Initial setting of spotify tokens from user accept code"""
+        data = spt.get_token_json(
+            grant_type='authorization_code',
+            code=code
+        )
+        self.save_spt(data)
+
+    def refresh_spt(self):
+        """Refreshes expired access token using stored refresh token"""
+        if self.spt_refresh_token:
+            data = spt.get_token_json(
+                grant_type='refresh_token',
+                refresh_token=self.spt_refresh_token
+            )
+            self.save_spt(data)
+
+    def save_spt(self, data):
+        """Saves tokens from json response"""
+        access = data.get('access_token', None)
+        refresh = data.get('refresh_token', self.spt_refresh_token)
+        expires = data.get('expires_in', 3600)
+        self.spt_access_token = access
+        self.spt_refresh_token = refresh
+        self.spt_expires = timezone.now() + timedelta(seconds=expires - 300)
+        self.save()
+
+    def unlink_spt(self):
+        self.spt_access_token = None
+        self.spt_refresh_token = None
+        self.spt_expires = None
+
+    @property
+    def can_spt(self):
+        return (
+            self.spt_access_token and
+            self.spt_expires and
+            (timezone.now() < self.spt_expires)
+        )
+
+    def get_or_update_spt(self):
+        if self.spt_refresh_token:
+            if not self.can_spt:
+                self.refresh_spt()
+            return self.spt_access_token
+        return None
+
+    def save(self, *args, **kwargs):
+        self.update_quota()
+        super().save(*args, **kwargs)
+
+@receiver(post_save, sender=User)
+def create_user_profile(sender, instance, created, **kwargs):
+    if created:
+        Profile.objects.create(user=instance)
+
+@receiver(post_save, sender=User)
+def save_user_profile(sender, instance, **kwargs):
+    Profile.objects.get_or_create(user=instance)
+    instance.profile.save()

backend/playlist/models/track.py

@@ -0,0 +1,144 @@
+from django.db import models
+from django.contrib.auth.models import User
+from random import choice
+from .album import Album
+from .artist import Artist
+from .vote import Vote
+from playlist.spotify import spt
+#from playlist.utils import track_from_json
+import re
+
+class TrackManager(models.Manager):
+    def get_queryset(self):
+        return super().get_queryset().filter(old=False)
+
+    def from_json(self, cls, json):
+        try:
+            return cls.objects.get(pk=json["id"])
+        except cls.DoesNotExist:
+            keys = [f.name for f in cls._meta.get_fields()]
+            subset = {key:json[key] for key in set(keys) & set(json.keys())}
+            return cls.objects.create(**subset)
+
+    def track_from_json(self, json):
+        """Creates the track, album and artist from spotify response."""
+        # Album artists
+        album_artists = []
+        for artist in json["album"]["artists"]:
+            album_artists.append(self.from_json(Artist, artist))
+        del json["album"]["artists"]
+
+        # Album
+        album = self.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(self.from_json(Artist, artist))
+        del json["artists"]
+
+        # Track
+        track = self.from_json(Track, json)
+        for artist in track_artists:
+            track.artists.add(artist)
+        track.album = album
+        track.save()
+        return track
+
+    def create_from_spotify(self, spotify_link, profile):
+        if re.match(r"^https:\/\/spotify.link.*", spotify_link):
+            spotify_link = spt.unshort(spotify_link)
+
+        m = re.match(r"^https:\/\/open.spotify.com\/track\/([a-zA-Z0-9]+)\??", spotify_link)
+        if m is None:
+            return None, Exception("Invalid Spotify track URL")
+
+        spotify_id = m.group(1)
+
+        try:
+            existing_track = self.model.all_tracks.get(pk=spotify_id)
+            if existing_track.old:
+                existing_track.old = False
+                existing_track.user = profile.user
+                existing_track.save()
+                return (existing_track, "Updated old track")
+            else:
+                return (None, "Track already exists")
+        except:
+           pass
+        try:
+            json = spt.get_song_info(spotify_id)
+        except Exception as e:
+            print(e)
+            return (None, "Error fetching track")
+            #return TemplateResponse(request, "Nominate.html", {"message_type": "error", "message": "Fout bij het ophalen van het liedje."})
+
+        track = self.track_from_json(json)
+        track.nominated_by = profile.user
+        track.save()
+        dup = track.is_duplicate_of()
+        if dup:
+            track.delete()
+            return (None, "Track already exists")
+        #get_banter.delay(track.id)
+        profile.quota -= 1
+        profile.save()
+        return (track, None)
+
+class Track(models.Model):
+    def __str__(self):
+        s = f"'{self.name}'"
+        if self.artists.count() > 0:
+            artists = list(map(lambda a:a.name, self.artists.all()))
+            s += " door "
+            s += " & ".join(artists)
+        return s
+    @property
+    def artist(self):
+        if self.artists.count() > 0:
+            artists = list(map(lambda a:a.name, self.artists.all()))
+            return " & ".join(artists)
+
+    @property
+    def name_sanitized(self):
+        return re.match("[^\-\(]*\w", self.name)[0]
+
+    @property
+    def link(self):
+        return f'https://open.spotify.com/track/{self.id}'
+
+    @property
+    def score(self):
+        return Vote.objects.filter(track=self.id).aggregate(rating=models.Avg("points"))["rating"]
+
+    def is_duplicate_of(self):
+        tracks = Track.objects.exclude(id=self.id).filter(name__startswith=self.name_sanitized)
+        artists = [artist.name for artist in self.artists.all()]
+        for artist in artists:
+            tracks = tracks.filter(artists__name__exact=artist)
+        return tracks.first()
+
+
+    id = models.CharField(primary_key=True, max_length=128)
+    nominated_by = models.ForeignKey(User, blank=True, null=True, default=None, on_delete=models.SET_NULL)
+    name = models.CharField(max_length=255)
+    artists = models.ManyToManyField(Artist)
+    album = models.ForeignKey(Album, blank=True, null=True, on_delete=models.SET_NULL)
+
+    explicit = models.BooleanField(default=False)
+    duration_ms = models.PositiveIntegerField(blank=True, null=True)
+    popularity = models.PositiveIntegerField(blank=True, null=True)
+
+    banter = models.TextField(null=True, blank=True)
+    banter_done = models.BooleanField(default=False)
+
+    old = models.BooleanField(default=False)
+    from_playlist = models.CharField(max_length=255, null=True, blank=True, default=None)
+
+    objects = TrackManager()
+    all_tracks = models.Manager()

backend/playlist/models/vote.py

@@ -0,0 +1,11 @@
+from django.db import models
+from django.contrib.auth.models import User
+from django.core.validators import MinValueValidator, MaxValueValidator
+
+class Vote(models.Model):
+    class Meta:
+        unique_together = ["user", "track"]
+    track = models.ForeignKey('Track', on_delete=models.CASCADE)
+    user = models.ForeignKey(User, on_delete=models.CASCADE)
+    points = models.IntegerField(validators=[MinValueValidator(1), MaxValueValidator(5)])
+    skipped = models.BooleanField(default=False)