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)
    updatedDate = models.DateTimeField(auto_now=True)
    colorCode = models.CharField(max_length=7, default='#FFFFFF')

    def __str__(self):
        return self.name

class Teacher(models.Model):
    lastName = models.CharField(max_length=100)
    firstName = models.CharField(max_length=100)
    email = models.EmailField(unique=True)
    specialities = models.ManyToManyField(Speciality, related_name='teachers')
    associatedProfile = models.ForeignKey(Profile, on_delete=models.CASCADE, null=True, blank=True)
    updatedDate = models.DateTimeField(auto_now=True)

    def __str__(self):
        return f"{self.lastName} {self.firstName}"

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

    atmosphereName = models.CharField(max_length=255, null=True, blank=True)
    ageGroup = models.JSONField()
    numberOfStudents = models.PositiveIntegerField()
    teachingLanguage = models.CharField(max_length=255)
    schoolYear = models.CharField(max_length=9)
    updatedDate = models.DateTimeField(auto_now_add=True)
    teachers = models.ManyToManyField(Teacher, related_name='schoolClasses')
    levels = ArrayField(models.IntegerField(choices=LEVEL_CHOICES), default=list)
    type = models.IntegerField(choices=PLANNING_TYPE_CHOICES, default=1)
    schedule = models.JSONField(default=list)
    openingDays = ArrayField(models.IntegerField(), default=list)

    def __str__(self):
        return self.atmosphereName

class Planning(models.Model):
    level = models.IntegerField(choices=LEVEL_CHOICES, null=True, blank=True)
    classModel = 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.classModel.atmosphereName}'