From ba7d2e5c5a78170fa178cd6f9a07a35ec385b63a Mon Sep 17 00:00:00 2001
From: Alban Gruin
Date: Mon, 27 Feb 2017 10:56:54 +0100
Subject: Léger reformatage

---
 management/commands/_private.py | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

(limited to 'management/commands/_private.py')

diff --git a/management/commands/_private.py b/management/commands/_private.py
index f14dae6..5b4a7ff 100644
--- a/management/commands/_private.py
+++ b/management/commands/_private.py
@@ -27,7 +27,8 @@ import requests
 class Week:
     def __init__(self, number, start):
         self.number = number
-        self.start = timezone.make_aware(datetime.datetime.strptime(start, "%d/%m/%Y"))
+        self.start = timezone.make_aware(
+            datetime.datetime.strptime(start, "%d/%m/%Y"))
 
     def get_day(self, id):
         return self.start + datetime.timedelta(id)
@@ -58,7 +59,7 @@ def get_from_db_or_create(cls, timetable=None, **kwargs):
 
     return obj
 
-def get_events(soup, weeks, year, week, timetable):
+def get_events(timetable, year, week, soup, weeks_in_soup):
     for event in soup.find_all("event"):
         title = None
         type_ = None
@@ -66,11 +67,12 @@ def get_events(soup, weeks, year, week, timetable):
         rooms = None
         notes = None
 
-        if weeks[event.rawweeks.text].number == week and \
-           weeks[event.rawweeks.text].year == year and \
+        if weeks_in_soup[event.rawweeks.text].number == week and \
+           weeks_in_soup[event.rawweeks.text].year == year and \
            event.resources.group is not None and \
            event.starttime is not None and event.endtime is not None:
-            date = weeks[event.rawweeks.text].get_day(int(event.day.text))
+            date = weeks_in_soup[event.rawweeks.text].get_day(int(
+                event.day.text))
 
             begin = add_time(date, datetime.datetime.strptime(
                 event.starttime.text, "%H:%M"))
@@ -104,7 +106,8 @@ def get_events(soup, weeks, year, week, timetable):
 def get_weeks(soup):
     weeks = {}
     for span in soup.find_all("span"):
-        weeks[span.alleventweeks.text] = Week(int(span.title.text), span["date"])
+        weeks[span.alleventweeks.text] = Week(int(span.title.text),
+                                              span["date"])
 
     return weeks
 
-- 
cgit v1.2.1


From 246d7a2e3a22548ce355a551fb19936dc420f695 Mon Sep 17 00:00:00 2001
From: Alban Gruin
Date: Tue, 28 Feb 2017 09:32:01 +0100
Subject: Léger nettoyage

---
 management/commands/_private.py | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

(limited to 'management/commands/_private.py')

diff --git a/management/commands/_private.py b/management/commands/_private.py
index 5b4a7ff..3cd23ca 100644
--- a/management/commands/_private.py
+++ b/management/commands/_private.py
@@ -46,15 +46,12 @@ def delete_courses_in_week(timetable, year, week):
     Course.objects.filter(begin__gte=start, begin__lt=end,
                           timetable=timetable).delete()
 
-def get_from_db_or_create(cls, timetable=None, **kwargs):
+def get_from_db_or_create(cls, **kwargs):
     obj = cls.objects.all().filter(**kwargs)
-    if timetable is not None:
-        obj = obj.filter(timetable=timetable)
 
     obj = obj.first()
     if obj is None:
         obj = cls(**kwargs)
-        obj.timetable = timetable
         obj.save()
 
     return obj
@@ -79,7 +76,7 @@ def get_events(timetable, year, week, soup, weeks_in_soup):
             end = add_time(date, datetime.datetime.strptime(
                 event.endtime.text, "%H:%M"))
 
-            groups = [get_from_db_or_create(Group, timetable,
+            groups = [get_from_db_or_create(Group, timetable=timetable,
                                             celcat_name=item.text)
                       for item in event.resources.group.find_all("item")]
 
-- 
cgit v1.2.1