34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
from django.db.models import ForeignKey, Model, IntegerField, BooleanField, CASCADE, PROTECT
|
|
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):
|
|
return f"{encode_id(self.pk)}"
|
|
|
|
@classmethod
|
|
def decode_id(cls, hashed_id:str) -> id:
|
|
return decode_id(hashed_id)
|
|
|
|
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() |