fix: Force new user workflow through right forms

Resolves #112
Resolves #68
This commit is contained in:
James Graham
2021-04-25 17:13:59 +01:00
parent 7681e78a50
commit adc9021002
2 changed files with 42 additions and 27 deletions

View File

@@ -6,7 +6,7 @@ These views don't represent any of the models in the apps.
from django.conf import settings
from django.contrib.auth import get_user_model
from django.contrib.auth.mixins import LoginRequiredMixin
from django.urls import reverse_lazy
from django.urls import reverse
from django.views.generic import TemplateView
from django.views.generic.edit import UpdateView
@@ -25,7 +25,13 @@ class ConsentTextView(LoginRequiredMixin, UpdateView):
model = User
form_class = forms.ConsentForm
template_name = 'consent.html'
success_url = reverse_lazy('index')
def get_success_url(self) -> str:
try:
return reverse('people:person.detail', kwargs={'pk': self.request.user.person.pk})
except AttributeError:
return reverse('index')
def get_object(self, *args, **kwargs) -> User:
return self.request.user