From 60092220ca3c47a20f50b677a2bb2173ae3f89a3 Mon Sep 17 00:00:00 2001 From: Alban Gruin Date: Mon, 27 Feb 2017 09:52:08 +0100 Subject: Parsage du nom des groupes basé sur une regex pour éviter les problèmes liés à l'ancienne méthode --- utils.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) (limited to 'utils.py') diff --git a/utils.py b/utils.py index 4ce3a21..090142b 100644 --- a/utils.py +++ b/utils.py @@ -17,6 +17,7 @@ from django.utils import timezone import datetime +import re def get_current_week(): return timezone.now().isocalendar()[:2] @@ -44,3 +45,17 @@ def group_courses(courses): grouped_courses[-1].append(course) return grouped_courses + +def parse_group(name): + group_regex = re.compile("^(.+?)\s*\-\s*(((CM)(\w))|((TD)(\w)(\d))|((TP)(\w)(\d)(\d)))$") + search = group_regex.search(name) + if search is None: + return None, None, None, None + + parts = search.groups(0) + if parts[3] == "CM": + return parts[0], parts[4], None, None + elif parts[6] == "TD": + return parts[0], parts[7], parts[8], None + elif parts[10] == "TP": + return parts[0], parts[11], parts[12], parts[13] -- cgit v1.2.1