aboutsummaryrefslogtreecommitdiff
path: root/management/commands/_private.py
diff options
context:
space:
mode:
Diffstat (limited to 'management/commands/_private.py')
-rw-r--r--management/commands/_private.py7
1 files changed, 3 insertions, 4 deletions
diff --git a/management/commands/_private.py b/management/commands/_private.py
index 65772a4..f5a3ebc 100644
--- a/management/commands/_private.py
+++ b/management/commands/_private.py
@@ -1,6 +1,8 @@
from bs4 import BeautifulSoup
from django.utils import timezone
+
from edt.models import Group, Room, Course
+from edt.utils import get_week
import datetime
import requests
@@ -23,10 +25,7 @@ def add_time(date, time):
return date + delta
def delete_courses_in_week(timetable, year, week):
- start = timezone.make_aware(datetime.datetime.strptime(
- "{0}-W{1:02d}-1".format(year, week), "%Y-W%W-%w"))
- end = start + datetime.timedelta(weeks=1)
-
+ start, end = get_week(year, week)
Course.objects.filter(begin__gte=start, begin__lt=end,
timetable=timetable).delete()