from urllib import urlencode
import urllib2
import urlparse
+from functools import wraps
+
+from django.utils.decorators import available_attrs
+from django.views.decorators.debug import sensitive_post_parameters
+from django.views.decorators.cache import cache_control
+from django.utils.cache import patch_cache_control
+from django.views.decorators.csrf import csrf_protect
from django.http import HttpResponse, HttpResponseRedirect
from django.conf import settings
logger = logging.getLogger(__name__)
+_never_cache = cache_control(no_cache=True, must_revalidate=True)
+
+
+def never_cache(view_func):
+ """
+ Decorator that adds headers to a response so that it will
+ never be cached.
+ """
+ @wraps(view_func, assigned=available_attrs(view_func))
+ def _wrapped_view_func(request, *args, **kwargs):
+ response = view_func(request, *args, **kwargs)
+ patch_cache_control(response, no_cache=True,
+ must_revalidate=True, proxy_revalidate=True)
+ response['Pragma'] = 'no-cache'
+ return response
+ return _wrapped_view_func
+
+
+@sensitive_post_parameters()
+@csrf_protect
+@never_cache
def login(request, template_name='cas/login.html',
success_redirect=settings.LOGIN_REDIRECT_URL,
warn_template_name='cas/warn.html', **kwargs):
url = '%s?%s' % (base_url, args)
logging.debug('Redirecting to %s', url)
return HttpResponseRedirect(url)
-
+
if user is None:
errors.append('Incorrect username and/or password.')
else:
for receiver, response in signals.on_cas_login.send(sender=login, request=request, **kwargs):
if isinstance(response, HttpResponse):
return response
-
+
if service is None:
# Try and pull the service off the session
service = request.session.pop('service', service)
-
+
+ signals.on_cas_login_success.send(sender=login, request=request,
+ service=service, **kwargs)
+
if service is None:
# Normal internal success redirection.
logging.debug('Redirecting to %s', success_redirect)
'service': service,
'warn': False
}, context_instance=RequestContext(request))
-
+
# Create a service ticket and redirect to the service.
ticket = ServiceTicket.objects.create(service=service, user=user)
if 'service' in request.session:
return render_to_response(template_name, {'form': form, 'errors': errors}, context_instance=RequestContext(request))
+@never_cache
def validate(request):
"""Validate ticket via CAS v.1 protocol
"""
username = ticket.user.username
ticket.delete()
- results = signals.on_cas_collect_histories.send(sender=validate, for_email=ticket.user.email)
+ results = signals.on_cas_collect_histories.send(sender=validate, for_user=ticket.user)
histories = '\n'.join('\n'.join(rs) for rc, rs in results)
logger.info('Validated %s %s', username, "(also %s)" % histories if histories else '')
return HttpResponse("yes\n%s\n%s" % (username, histories))
logger.info('Validation failed.')
return HttpResponse("no\n\n")
-
+
+@never_cache
def logout(request, template_name='cas/logout.html',
auto_redirect=settings.CAS_AUTO_REDIRECT_AFTER_LOGOUT):
url = request.GET.get('url', None)
context_instance=RequestContext(request))
+@never_cache
def proxy(request):
targetService = request.GET['targetService']
pgt_id = request.GET['pgt']
pgt = None
user = ticket.user
+ ticket.delete()
return _cas2_sucess_response(user, pgtIouId, proxies)
+@never_cache
def service_validate(request):
"""Validate ticket via CAS v.2 protocol"""
service = request.GET.get('service', None)
return ticket_validate(service, ticket_string, pgtUrl)
+@never_cache
def proxy_validate(request):
"""Validate ticket via CAS v.2 protocol"""
service = request.GET.get('service', None)
username = etree.SubElement(auth_success, CAS + 'user')
username.text = user.username
- if settings.CAS_CUSTOM_ATTRIBUTES_CALLBACK:
- callback = get_callable(settings.CAS_CUSTOM_ATTRIBUTES_CALLBACK)
- attrs = callback(user)
- if len(attrs) > 0:
- formater = get_callable(settings.CAS_CUSTOM_ATTRIBUTES_FORMATER)
- formater(auth_success, attrs)
+ attrs = {}
+ for receiver, custom in signals.cas_collect_custom_attributes.send(sender=auth_success_response, user=user):
+ if custom:
+ attrs.update(custom)
+
+ identifiers = [i for sr, rr in signals.on_cas_collect_histories.send(sender=validate, for_user=user)
+ for i in rr]
+
+ if identifiers:
+ # Singular `identifier`, as that is the name of the element tag(s).
+ attrs['identifier'] = identifiers
+
+ if attrs:
+ formatter = get_callable(settings.CAS_CUSTOM_ATTRIBUTES_FORMATER)
+ formatter(auth_success, attrs)
if pgt:
pgtElement = etree.SubElement(auth_success, CAS + 'proxyGrantingTicket')
fnp / django-cas-provider.git / blobdiff