from django.db import models

from users.models import CustomUser


class Event(models.Model):
    MORNING, AFTERNOON, EVENING = range(3)
    TIME_SLOTS = {
        MORNING:   "Voormiddag",
        AFTERNOON: "Namiddag",
        EVENING:   "Avond",
    }
    date = models.DateField()
    time = models.IntegerField(choices=TIME_SLOTS.items(), default=MORNING)
    capacity = models.IntegerField(default=6)


    def __str__(self):
        return f"{self.date} {self.TIME_SLOTS[self.time]}"


    @property
    def time_str(self):
        return Event.TIME_SLOTS[self.time]


class EventRegistration(models.Model):
    INTERESTED = "I"
    ADMITTED   = "A"
    DENIED     = "D"
    REGISTRATION_STATE = {
        INTERESTED: "Interested",
        ADMITTED:   "Admitted",
        DENIED:     "Denied",
    }
    event = models.ForeignKey(Event, on_delete=models.CASCADE)
    user = models.ForeignKey(CustomUser, on_delete=models.CASCADE)
    state = models.CharField(max_length=1, choices=REGISTRATION_STATE.items())

    def __str__(self):
        return f"Reservation[{self.user.username}:{self.event.date}:{self.state}]"


    @property
    def state_str(self):
        return EventRegistration.REGISTRATION_STATE[self.state]