aboutsummaryrefslogtreecommitdiff
path: root/management/commands/timetables.py
diff options
context:
space:
mode:
Diffstat (limited to 'management/commands/timetables.py')
-rw-r--r--management/commands/timetables.py10
1 files changed, 9 insertions, 1 deletions
diff --git a/management/commands/timetables.py b/management/commands/timetables.py
index 6412b7e..21148c8 100644
--- a/management/commands/timetables.py
+++ b/management/commands/timetables.py
@@ -1,7 +1,7 @@
from django.core.management.base import BaseCommand, CommandError
from django.db import transaction
from django.utils import timezone
-from edt.models import Timetable, Group, Room, Course
+from edt.models import Timetable, LastUpdate, Group, Room, Course
from bs4 import BeautifulSoup
@@ -25,6 +25,14 @@ def process_timetable_week(timetable, year, week, soup, weeks_in_soup):
course.save()
+ date = timezone.make_aware(datetime.datetime.now())
+ try:
+ lastUpdate = LastUpdate.objects.get(timetable=timetable, year=year, week=week)
+ lastUpdate.date = date
+ except:
+ lastUpdate = LastUpdate(timetable=timetable, year=year, week=week, date=date)
+ lastUpdate.save()
+
def process_timetable(timetable, year, weeks):
soup = get_xml(timetable.url)
weeks_in_soup = get_weeks(soup)