56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
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
|
|
|
|
# Create your models here.
|
|
class DraftSession(Model):
|
|
season = ForeignKey(Season, on_delete=CASCADE)
|
|
is_active = BooleanField()
|
|
current_nomination_index = IntegerField()
|
|
|
|
@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)
|
|
user = ForeignKey(User, on_delete=CASCADE)
|
|
budget = IntegerField()
|
|
|
|
class DraftMoviePool(Model):
|
|
draft = ForeignKey(DraftSession, on_delete=CASCADE)
|
|
movie = ForeignKey(Movie, on_delete=CASCADE)
|
|
nominated = BooleanField()
|
|
|
|
class DraftPick(Model):
|
|
draft = ForeignKey(DraftSession, on_delete=CASCADE)
|
|
movie = ForeignKey(Movie, on_delete=CASCADE)
|
|
winner = ForeignKey(User, on_delete=CASCADE)
|
|
bid_amount = 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" |