class StudentSignUp(CreateView):
model = UserRole
form_class = StudentSignupForm
template_name = 'register/signup.html'
def get_context_data(self, **kwargs):
kwargs['user_type'] = 'student'
return super().get_context_data(**kwargs)
def student_valid_form(self, student_form):
user = student_form.save()
login(self.request, user)
return redirect('students:student_index')
class StaffSignup(CreateView):
model = UserRole
form_class = StaffSignupForm
template_name = 'register/signup.html'
def get_context_data(self, **kwargs):
kwargs['user_type'] = 'staff'
return super().get_context_data(**kwargs)
def staff_valid_form(self, staff_form):
user = staff_form.save()
login(self.request, user)
return redirect('staff:staff_index')