aboutsummaryrefslogtreecommitdiff
path: root/models.py
blob: d01dbf08d223b3b41a0fef7b7788fbb36e799b46 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
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 LastUpdate(models.Model):
    timetable = models.ForeignKey(Timetable, on_delete=models.CASCADE, verbose_name="emploi du temps")
    week = models.IntegerField(verbose_name="semaine")
    year = models.IntegerField(verbose_name="année")
    date = models.DateTimeField(verbose_name="date de mise à jour")


    class Meta:
        unique_together = ("timetable", "week", "year",)
        verbose_name = "dernière mise à jour"
        verbose_name_plural = "dernières mises à jour"


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"