From aec307d9657dbff6bc8ffbf529fdf050c25b726c Mon Sep 17 00:00:00 2001 From: Alex Dehnert Date: Mon, 6 Jun 2011 03:10:15 -0400 Subject: [PATCH] Merge "upstream" fixes from SIPB's snippets --- __init__.py | 22 ++++++++++++++++------ 1 file changed, 16 insertions(+), 6 deletions(-) diff --git a/__init__.py b/__init__.py index 8f785ae..7958beb 100644 --- a/__init__.py +++ b/__init__.py @@ -1,3 +1,7 @@ +import subprocess +import ldap +import ldap.filter + from django.contrib.auth.middleware import RemoteUserMiddleware from django.contrib.auth.backends import RemoteUserBackend from django.contrib.auth.views import login @@ -7,9 +11,12 @@ from django.contrib import auth from django.core.exceptions import ObjectDoesNotExist import settings -def zephyr(msg, clas='remit', instance='log', rcpt='adehnert',): - import os - os.system("zwrite -d -c '%s' -i '%s' '%s' -m '%s'" % (clas, instance, rcpt, msg, )) +def zephyr(msg, clas='message', instance='log', rcpt='nobody',): + proc = subprocess.Popen( + ['zwrite', '-d', '-n', '-c', clas, '-i', instance, rcpt, ], + stdin=subprocess.PIPE, stdout=subprocess.PIPE + ) + proc.communicate(msg) class ScriptsRemoteUserMiddleware(RemoteUserMiddleware): header = 'SSL_CLIENT_S_DN_Email' @@ -24,12 +31,13 @@ class ScriptsRemoteUserBackend(RemoteUserBackend): return username def configure_user(self, user, ): username = user.username - import ldap + user.password = "ScriptsSSLAuth" con = ldap.open('ldap.mit.edu') con.simple_bind_s("", "") dn = "dc=mit,dc=edu" fields = ['cn', 'sn', 'givenName', 'mail', ] - result = con.search_s('dc=mit,dc=edu', ldap.SCOPE_SUBTREE, 'uid=%s'%username, fields) + userfilter = ldap.filter.filter_format('uid=%s', [username]) + result = con.search_s('dc=mit,dc=edu', ldap.SCOPE_SUBTREE, userfilter, fields) if len(result) == 1: user.first_name = result[0][1]['givenName'][0] user.last_name = result[0][1]['sn'][0] @@ -38,11 +46,13 @@ class ScriptsRemoteUserBackend(RemoteUserBackend): user.groups.add(auth.models.Group.objects.get(name='mit')) except ObjectDoesNotExist: print "Failed to retrieve mit group" - user.save() + else: + raise ValueError, ("Could not find user with username '%s' (filter '%s')"%(username, userfilter)) try: user.groups.add(auth.models.Group.objects.get(name='autocreated')) except ObjectDoesNotExist: print "Failed to retrieve autocreated group" + user.save() return user def scripts_login(request, **kwargs): -- 2.44.0