from django.db import models
from django.db.models import Q
from django.utils.text import slugify


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)

    slug = models.SlugField(max_length=64, unique=True, default="")

    def __str__(self):
        return self.name

    def save(self):
        try:
            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])
        except:
            print("Malformed name: {0}. Ignoring".format(self.name))
        finally:
            self.slug = slugify(self.name)
            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 CourseManager(models.Manager):
    def get_courses_for_group(self, group, **filters):
        return self.get_queryset().filter(Q(groups__td__isnull=True) | Q(groups__td=group.td), Q(groups__tp__isnull=True) | Q(groups__tp=group.tp), groups__mention=group.mention, groups__subgroup=group.subgroup, **filters).order_by("begin")


class Course(models.Model):
    objects = CourseManager()

    name = models.CharField(max_length=255, verbose_name="nom", null=True)
    type_ = models.CharField(name="type", max_length=255, verbose_name="type de cours", null=True)
    timetable = models.ForeignKey(Timetable, on_delete=models.CASCADE, verbose_name="emploi du temps")
    notes = models.TextField(verbose_name="remarques", blank=True, null=True)

    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"