X-Git-Url: https://git.auf.org/?p=auf_django_saml.git;a=blobdiff_plain;f=auf%2Fdjango%2Fsaml%2Fviews.py;h=f2af6f91d736368bd43c19e0a90e7b87e88a62b3;hp=d8990228132322fbc4f2b7411916074fcfd4fadc;hb=e1e432321118aa864931ccfed479be343291c8d4;hpb=aa2ccaae52702b7c13ef794e9c958e06d526d8d1

diff --git a/auf/django/saml/views.py b/auf/django/saml/views.py
index d899022..f2af6f9 100644
--- a/auf/django/saml/views.py
+++ b/auf/django/saml/views.py
@@ -11,13 +11,18 @@ from settings import SAML_REDIRECT_FIELD_NAME,\
         SAML_MELLON_LOGIN_URL,\
         SAML_MELLON_LOGOUT_URL,\
         SAML_CHANGE_PASSWORD_URL,\
-        SAML_LOGOUT_REDIRECT_URL
+        SAML_LOGOUT_REDIRECT_URL, \
+        SAML_AUTH
 
 
 def redirect_to_login(request, redirect_to=None, do_redirect=True):
     if redirect_to is None:
         redirect_to = request.get_full_path()
-    url = "%s?%s=%s" % (SAML_MELLON_LOGIN_URL,
+    if SAML_AUTH:
+        base_url = SAML_MELLON_LOGIN_URL
+    else:
+        base_url = reverse('sandbox_login')
+    url = "%s?%s=%s" % (base_url,
             SAML_REDIRECT_FIELD_NAME,
             redirect_to,
             )
@@ -66,7 +71,11 @@ def local_logout(request, ):
     """
     query_string = request.META['QUERY_STRING']
     auth_logout(request)
-    logout_url = "%s?%s" % (SAML_MELLON_LOGOUT_URL, query_string)
+    if SAML_AUTH:
+        base_url = SAML_MELLON_LOGOUT_URL
+    else:
+        base_url = reverse('sandbox_logout')
+    logout_url = "%s?%s" % (base_url, query_string)
     response = HttpResponse(content="", status=303)
     response["Location"] = logout_url
     return response