From 8b7f041966462925ede9dcf1687a5efd255bd85a Mon Sep 17 00:00:00 2001 From: Benjamin Dauvergne Date: Fri, 26 Jul 2019 16:07:58 +0200 Subject: [PATCH] add kwargs mellon_template_prefix to LoginView (#35083) --- mellon/views.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/mellon/views.py b/mellon/views.py index 6228a93..abbecf6 100644 --- a/mellon/views.py +++ b/mellon/views.py @@ -124,6 +124,10 @@ class ProfileMixin(object): class LoginView(ProfileMixin, LogMixin, View): + def dispatch(self, request, *args, **kwargs): + self.mellon_template_prefix = kwargs.pop('mellon_template_prefix', 'mellon') + return super(LoginView, self).dispatch(request, *args, **kwargs) + def get_idp(self, request): entity_id = request.POST.get('entityID') or request.GET.get('entityID') if not entity_id: @@ -180,7 +184,7 @@ class LoginView(ProfileMixin, LogMixin, View): if error_url: error_url = resolve_url(error_url) next_url = error_url or self.get_next_url(default=resolve_url(settings.LOGIN_REDIRECT_URL)) - return render(request, 'mellon/authentication_failed.html', + return render(request, '%s/authentication_failed.html' % self.mellon_template_prefix, { 'debug': settings.DEBUG, 'reason': reason, @@ -248,12 +252,12 @@ class LoginView(ProfileMixin, LogMixin, View): else: self.log.warning('user %s (NameID is %r) is inactive, login refused', user, attributes['name_id_content']) - return render(request, 'mellon/inactive_user.html', { + return render(request, '%s/inactive_user.html' % self.mellon_template_prefix, { 'user': user, 'saml_attributes': attributes}) else: self.log.warning('no user found for NameID %r', attributes['name_id_content']) - return render(request, 'mellon/user_not_found.html', + return render(request, '%s/user_not_found.html' % self.mellon_template_prefix, {'saml_attributes': attributes}) request.session['lasso_session_dump'] = login.session.dump() -- 2.22.0