Commit b83e5628 by Kálmán Viktor

dashboard: handle not existing profiles in 2fa

profile is created after 1st login
parent 509c7ce7
...@@ -111,7 +111,7 @@ class CircleLoginView(LoginView): ...@@ -111,7 +111,7 @@ class CircleLoginView(LoginView):
def form_valid(self, form): def form_valid(self, form):
user = form.get_user() user = form.get_user()
if user.profile.two_factor_secret: if hasattr(user, "profile") and user.profile.two_factor_secret:
self.request.session['two-fa-user'] = user.pk self.request.session['two-fa-user'] = user.pk
self.request.session['two-fa-redirect'] = self.get_success_url() self.request.session['two-fa-redirect'] = self.get_success_url()
self.request.session['login-type'] = "password" self.request.session['login-type'] = "password"
...@@ -719,7 +719,7 @@ if hasattr(settings, 'SAML_ORG_ID_ATTRIBUTE'): ...@@ -719,7 +719,7 @@ if hasattr(settings, 'SAML_ORG_ID_ATTRIBUTE'):
relay_state = settings.LOGIN_REDIRECT_URL relay_state = settings.LOGIN_REDIRECT_URL
logger.debug('Redirecting to the RelayState: ' + relay_state) logger.debug('Redirecting to the RelayState: ' + relay_state)
if user.profile.two_factor_secret: if hasattr(user, "profile") and user.profile.two_factor_secret:
request.session['two-fa-user'] = user.pk request.session['two-fa-user'] = user.pk
request.session['two-fa-redirect'] = relay_state request.session['two-fa-redirect'] = relay_state
request.session['login-type'] = "saml2" request.session['login-type'] = "saml2"
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment