From 22faec68b77a48681df12036713f44f9d8df7bfd Mon Sep 17 00:00:00 2001 From: Alban Gruin Date: Sun, 22 Jan 2017 20:00:52 +0100 Subject: Regroupement des cours effectué à l'aide d'une fonction pour éviter la duplication de code --- utils.py | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'utils.py') diff --git a/utils.py b/utils.py index 5cc6648..3479b47 100644 --- a/utils.py +++ b/utils.py @@ -8,3 +8,13 @@ def get_week(year, week): end = start + datetime.timedelta(weeks=1) return start, end + +def group_courses(courses): + grouped_courses = [] + for i, course in enumerate(courses): + if i == 0 or courses[i - 1].begin.day != course.begin.day: + grouped_courses.append([course]) + else: + grouped_courses[-1].append(course) + + return grouped_courses -- cgit v1.2.1