def setUp(self):
self.test_exam1 = make_test_exam(10000)
self.test_exam2 = make_test_exam(20000)
self.test_exam3 = make_test_exam(30000)
self.permission1 = InstructorPermission.objects.get(pk=100000)
self.permission2 = InstructorPermission.objects.get(pk=200000)
self.permission3 = InstructorPermission.objects.get(pk=300000)
self.dept_cs = Department(long_name='Computer Science',
short_name='CS',
abbreviation='COMPSCI')
self.dept_cs.save()
self.course_cs_1 = Course(department=self.dept_cs, number='1')
self.course_cs_1.save()
self.instructor_cs = Instructor(first_name='Tau',
last_name='Bate',
department=self.dept_cs)
self.instructor_cs.save()
self.term = Term(term='sp', year=2016, current=True)
self.term.save()
self.course_instance_cs_1 = CourseInstance(term=self.term,
course=self.course_cs_1)
self.course_instance_cs_1.save()
self.course_instance_cs_1.instructors.add(self.instructor_cs)
self.course_instance_cs_1.save()
self.permission4 = InstructorPermission(instructor=self.instructor_cs,
permission_allowed=True)
self.permission4.save()
def save(self, *args, **kwargs):
permission_allowed = self.cleaned_data.get('permission_allowed')
if permission_allowed is not None:
if not self.permission:
self.permission = InstructorPermission(
instructor=self.instance)
self.permission.permission_allowed = permission_allowed
self.permission.correspondence = self.cleaned_data.get(
'correspondence')
self.permission.save()
return super(InstructorEditForm, self).save(*args, **kwargs)
