from django.db import models
from Auth.models import Profile
from django.db.models import JSONField
from django.dispatch import receiver
from django.contrib.postgres.fields import ArrayField

LEVEL_CHOICES = [
    (1, 'Très Petite Section (TPS)'),
    (2, 'Petite Section (PS)'),
    (3, 'Moyenne Section (MS)'),
    (4, 'Grande Section (GS)'),
    (5, 'Cours Préparatoire (CP)'),
    (6, 'Cours Élémentaire 1 (CE1)'),
    (7, 'Cours Élémentaire 2 (CE2)'),
    (8, 'Cours Moyen 1 (CM1)'),
    (9, 'Cours Moyen 2 (CM2)')
]

class Speciality(models.Model):
    name = models.CharField(max_length=100)
    updated_date = models.DateTimeField(auto_now=True)
    color_code = models.CharField(max_length=7, default='#FFFFFF')

    def __str__(self):
        return self.name

class Teacher(models.Model):
    last_name = models.CharField(max_length=100)
    first_name = models.CharField(max_length=100)
    email = models.EmailField(unique=True)
    specialities = models.ManyToManyField(Speciality, blank=True)
    associated_profile = models.ForeignKey(Profile, on_delete=models.CASCADE, null=True, blank=True)
    updated_date = models.DateTimeField(auto_now=True)

    def __str__(self):
        return f"{self.last_name} {self.first_name}"

class SchoolClass(models.Model):
    PLANNING_TYPE_CHOICES = [
        (1, 'Annuel'),
        (2, 'Semestriel'),
        (3, 'Trimestriel')
    ]

    atmosphere_name = models.CharField(max_length=255, null=True, blank=True)
    age_range = models.JSONField(blank=True)
    number_of_students = models.PositiveIntegerField(blank=True)
    teaching_language = models.CharField(max_length=255, blank=True)
    school_year = models.CharField(max_length=9, blank=True)
    updated_date = models.DateTimeField(auto_now_add=True)
    teachers = models.ManyToManyField(Teacher, blank=True)
    levels = ArrayField(models.IntegerField(choices=LEVEL_CHOICES), default=list)
    type = models.IntegerField(choices=PLANNING_TYPE_CHOICES, default=1)
    time_range = models.JSONField(default=list)
    opening_days = ArrayField(models.IntegerField(), default=list)

    def __str__(self):
        return self.atmosphere_name

class Planning(models.Model):
    level = models.IntegerField(choices=LEVEL_CHOICES, null=True, blank=True)
    school_class = models.ForeignKey(SchoolClass, null=True, blank=True, related_name='plannings', on_delete=models.CASCADE)
    schedule = JSONField(default=dict)

    def __str__(self):
        return f'Planning for {self.level} of {self.school_class.atmosphere_name}'