Show error message for echo_attributes when not logged in via SAML

This fixes issue #72
IdentityPython · Aug 16, 2017 · d987dc0 · d987dc0
1 parent 5b5577c
commit d987dc0
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/djangosaml2/views.py b/djangosaml2/views.py
@@ -323,8 +323,12 @@ def echo_attributes(request,
     client = Saml2Client(conf, state_cache=state,
                          identity_cache=IdentityCache(request.session))
     subject_id = _get_subject_id(request.session)
-    identity = client.users.get_identity(subject_id,
-                                         check_not_on_or_after=False)
+    try:
+        identity = client.users.get_identity(subject_id,
+                                             check_not_on_or_after=False)
+    except AttributeError:
+        return HttpResponse("No active SAML identity found. Are you sure you have logged in via SAML?")
+
     return render(request, template, {'attributes': identity[0]})