Spotify user api integration

En een begin aan een playlist maker. Doet op het moment niet veel.

playlist/models/__init__.py

@@ -3,3 +3,4 @@ from .artist import Artist
 from .album import Album
 from .vote import Vote
 from .profile import Profile, Background
+from .playlist import Playlist

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)

playlist/models/profile.py

@@ -5,6 +5,7 @@ from django.db.models.signals import post_save
 from django.dispatch import receiver
 from django.utils import timezone
 from datetime import datetime, timedelta
+from ..spotify import spt
 
 NOMINATIONS_PER_DAY = 10
 MAX_QUOTA = 22
@@ -25,6 +26,10 @@ class Profile(models.Model):
     quota = models.FloatField(default=10.0)
     quota_last_updated = models.DateTimeField(default=timezone.now, null=False)
 
+    spt_access_token = models.CharField(max_length=128, default="", null=True)
+    spt_refresh_token = models.CharField(max_length=128, default="", null=True)
+    spt_expires = models.DateTimeField(default=None, null=True)
+
     @property
     def quota_display(self):
        return int(self.quota)
@@ -38,6 +43,53 @@ class Profile(models.Model):
             print(f"Er zijn {elapsed.seconds}s voorbij, je nieuwe quota is {self.quota}")
             self.quota_last_updated = timezone.now()
 
+    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)

playlist/models/track.py

@@ -2,6 +2,7 @@ from django.db import models
 from random import choice
 from .album import Album
 from .artist import Artist
+from .vote import Vote
 import re
 
 class TrackManager(models.Manager):
@@ -30,6 +31,10 @@ class Track(models.Model):
     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()]
@@ -37,6 +42,7 @@ class Track(models.Model):
             tracks = tracks.filter(artists__name__exact=artist)
         return tracks.first()
 
+
     id = models.CharField(primary_key=True, max_length=128)
     name = models.CharField(max_length=255)
     artists = models.ManyToManyField(Artist)

playlist/models/vote.py

@@ -1,10 +1,9 @@
 from django.db import models
 from django.contrib.auth.models import User
-from . import Track
 
 class Vote(models.Model):
     class Meta:
         unique_together = ["user", "track"]
-    track = models.ForeignKey(Track, on_delete=models.CASCADE)
+    track = models.ForeignKey('Track', on_delete=models.CASCADE)
     user = models.ForeignKey(User, on_delete=models.CASCADE)
     points = models.IntegerField()