From 31769b1f70ce77b815a64dd2a0dba525d48ace16 Mon Sep 17 00:00:00 2001 From: Alex Dehnert Date: Sat, 29 May 2010 21:10:59 -0400 Subject: [PATCH 1/1] Add scripts_login, to do cert login if possible --- django/mit/__init__.py | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) diff --git a/django/mit/__init__.py b/django/mit/__init__.py index ff296a8..8858248 100644 --- a/django/mit/__init__.py +++ b/django/mit/__init__.py @@ -1,7 +1,11 @@ from django.contrib.auth.middleware import RemoteUserMiddleware from django.contrib.auth.backends import RemoteUserBackend +from django.contrib.auth.views import login +from django.contrib.auth import REDIRECT_FIELD_NAME +from django.http import HttpResponseRedirect from django.contrib import auth from django.core.exceptions import ObjectDoesNotExist +import settings def zephyr(msg, clas='remit', instance='log', rcpt='adehnert',): import os @@ -40,3 +44,26 @@ class ScriptsRemoteUserBackend(RemoteUserBackend): except ObjectDoesNotExist: print "Failed to retrieve autocreated group" return user + +def scripts_login(request, **kwargs): + if request.META['HTTP_HOST'] == 'localhost': + return login(request, **kwargs) + elif request.META['SERVER_PORT'] == '444': + if request.user.is_authenticated(): + # They're already authenticated --- go ahead and redirect + if 'redirect_field_name' in kwargs: + redirect_field_name = kwargs['redirect_field_names'] + else: + from django.contrib.auth import REDIRECT_FIELD_NAME + redirect_field_name = REDIRECT_FIELD_NAME + redirect_to = request.REQUEST.get(redirect_field_name, '') + if not redirect_to or '//' in redirect_to or ' ' in redirect_to: + redirect_to = settings.LOGIN_REDIRECT_URL + return HttpResponseRedirect(redirect_to) + else: + return login(request, **kwargs) + else: + # Move to port 444 + host = request.META['HTTP_HOST'].split(':')[0] + redirect_to = "https://%s:444%s" % (host, request.META['REQUEST_URI'], ) + return HttpResponseRedirect(redirect_to) -- 2.44.0