Spotify user api integration

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

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)