def create_class(request):
if not is_teacher(request.user):
return redirect('index')
class_form = ClassForm(request.POST or None)
if class_form.is_valid():
classe = class_form.save()
classe.teachers.add(Teacher.objects.get(user=request.user))
classe.save()
return redirect('Classes:index')
return render(request, 'classes/form.html', {
'form': class_form,
'edit': False
})
def edit_class(request, id):
if not is_teacher(request.user):
return redirect('index')
if not Class.objects.filter(
pk=id, teachers__in=[Teacher.objects.get(user=request.user)]):
return redirect('Classes:index')
my_class = Class.objects.get(pk=id)
class_form = ClassForm(request.POST or None, instance=my_class)
if class_form.is_valid():
my_class.save()
return redirect('Classes:index')
return render(request, 'classes/form.html', {
'form': class_form,
'edit': True
})
