from django.db import models


class Timetable(models.Model):
    name = models.CharField(max_length=64, unique=True, verbose_name="nom")
    url = models.URLField(max_length=255, unique=True, verbose_name="URL")
    slug = models.SlugField(max_length=64, unique=True, default="")

    def __str__(self):
        return self.name


    class Meta:
        verbose_name = "emploi du temps"
        verbose_name_plural = "emplois du temps"


class Group(models.Model):
    name = models.CharField(max_length=255, verbose_name="nom")
    timetable = models.ForeignKey(Timetable, on_delete=models.CASCADE, verbose_name="emploi du temps")

    mention = models.CharField(max_length=32, db_index=True)
    subgroup = models.CharField(max_length=1, verbose_name="sous-groupe", db_index=True)
    td = models.IntegerField(verbose_name="groupe de TD", null=True, db_index=True)
    tp = models.IntegerField(verbose_name="groupe de TP", null=True, db_index=True)

    def __str__(self):
        return self.name

    def save(self):
        parts = self.name.split("-")[-2:]
        group = parts[1].strip()[2:]

        self.mention = parts[0].strip()
        self.subgroup = group[0]

        if len(group) > 1:
            self.td = int(group[1])
        if len(group) > 2:
            self.tp = int(group[2])

        super(Group, self).save()


    class Meta:
        verbose_name = "groupe"
        verbose_name_plural = "groupes"
        unique_together = ("name", "timetable",)


class Room(models.Model):
    name = models.CharField(max_length=255, unique=True, verbose_name="nom")

    def __str__(self):
        return self.name


    class Meta:
        verbose_name = "salle"
        verbose_name_plural = "salles"


class Course(models.Model):
    name = models.CharField(max_length=255, verbose_name="nom")
    type_ = models.CharField(name="type", max_length=255, verbose_name="type de cours")
    timetable = models.ForeignKey(Timetable, on_delete=models.CASCADE, verbose_name="emploi du temps")

    groups = models.ManyToManyField(Group, verbose_name="groupes")
    rooms = models.ManyToManyField(Room, verbose_name="salles")

    begin = models.DateTimeField(verbose_name="début du cours", db_index=True)
    end = models.DateTimeField(verbose_name="fin du cours")

    def __str__(self):
        return self.name


    class Meta:
        verbose_name = "cours"
        verbose_name_plural = "cours"