-
Notifications
You must be signed in to change notification settings - Fork 1
/
fixRoomsToPrefs.py
46 lines (31 loc) · 1.3 KB
/
fixRoomsToPrefs.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
from app.models import Course, RoomPreferences
courses = Course.select().where(Course.rid != None)
for course in courses:
pref, created = RoomPreferences.get_or_create(course = course.cId)
if pref.pref_1 == None:
print("pref_1 is none")
if pref.any_Choice == None:
print("any_choice is none")
if pref.none_Choice == None:
print("none_choice is none")
pref.any_Choice = 1
else:
print("Already had a pref")
# if pref.pref_1 != None and pref.none_Choice == None:
# pref.any_Choice = 2
if pref.pref_1 == None:
pref.pref_1 = course.rid
pref.any_Choice = 2
print("Updated Preference for course {1} to {0}".format(pref.pref_1.rID, course.cId))
else:
print("Course {0} has pref_1 of {1}".format(course.cId, pref.pref_1.rID))
pref.save()
print("Saved all things for {0}: any_choice: {1}".format(pref.course.cId, pref.any_Choice))
course.rid = None
course.save()
courses = Course.select().where(Course.rid == None)
prefs = RoomPreferences.select()
for pref in prefs:
if pref.pref_1 == None and pref.any_Choice == None and pref.none_Choice == None:
pref.any_Choice = 1
pref.save()