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()