deadlines

playlist/models/profile.py

@@ -4,8 +4,12 @@ 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
+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
@@ -42,6 +46,39 @@ class Profile(models.Model):
             self.quota = min(self.quota, MAX_QUOTA)
             print(f"Er zijn {elapsed.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,)
+        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,)
+
+    @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"""