aboutsummaryrefslogtreecommitdiff
path: root/models.py
diff options
context:
space:
mode:
Diffstat (limited to 'models.py')
-rw-r--r--models.py19
1 files changed, 19 insertions, 0 deletions
diff --git a/models.py b/models.py
index 09bf876..93d5e27 100644
--- a/models.py
+++ b/models.py
@@ -19,9 +19,28 @@ 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)
+ subgroup = models.CharField(max_length=1, verbose_name="sous-groupe")
+ td = models.IntegerField(verbose_name="groupe de TD", null=True)
+ tp = models.IntegerField(verbose_name="groupe de TP", null=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"