2025-07-26

draft/models.py

@@ -1,4 +1,4 @@
-from django.db.models import ForeignKey, Model, IntegerField, BooleanField, CASCADE, PROTECT
+from django.db.models import ForeignKey, Model, IntegerField, BooleanField, CASCADE, PROTECT, OneToOneField
 from boxofficefantasy.models import Season, User, Movie
 from boxofficefantasy_project.utils import encode_id, decode_id
 
@@ -10,11 +10,19 @@ class DraftSession(Model):
 
     @property
     def hashed_id(self):
+        if not self.pk: return ""
         return f"{encode_id(self.pk)}"
     
     @classmethod
     def decode_id(cls, hashed_id:str) -> id:
         return decode_id(hashed_id)
+    
+    def save(self, *args, **kwargs):
+        is_new = self.pk is None
+        super().save(*args, **kwargs)
+        if is_new and not hasattr(self, 'settings'):
+            DraftSessionSettings.objects.create(draft_session=self)
+
 
 class DraftParticipant(Model):
     draft = ForeignKey(DraftSession, on_delete=CASCADE)
@@ -31,4 +39,18 @@ class DraftPick(Model):
     movie = ForeignKey(Movie, on_delete=CASCADE)
     winner = ForeignKey(User, on_delete=CASCADE)
     bid_amount = IntegerField()
-    nomination_order = IntegerField()
+    nomination_order = IntegerField()
+
+class DraftSessionSettings(Model):
+    starting_budget = IntegerField(default=100)
+    draft_session = OneToOneField(
+        DraftSession,
+        on_delete=CASCADE,
+        related_name="settings"
+    )
+
+    def __str__(self):
+        return f"Settings for {self.draft_session}"
+    
+    class Meta:
+        verbose_name_plural = "Draft session settings"