X-Git-Url: https://snippets.scripts.mit.edu/gitweb.cgi/Scripts/git/.git/blobdiff_plain/9c504e485be1758ee01e7f577b9649633daae6ca..2a644ab1cfed458a1afe7d0277afdfefa63f6248:/django/mit/__init__.py diff --git a/django/mit/__init__.py b/django/mit/__init__.py index fd8d452..78ba4de 100644 --- a/django/mit/__init__.py +++ b/django/mit/__init__.py @@ -1,10 +1,16 @@ +import subprocess + from django.contrib.auth.middleware import RemoteUserMiddleware from django.contrib.auth.backends import RemoteUserBackend from django.contrib import auth +from django.core.exceptions import ObjectDoesNotExist 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, )) + 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' @@ -19,6 +25,7 @@ class ScriptsRemoteUserBackend(RemoteUserBackend): return username def configure_user(self, user, ): username = user.username + user.password = "ScriptsSSLAuth" import ldap con = ldap.open('ldap.mit.edu') con.simple_bind_s("", "") @@ -29,5 +36,13 @@ class ScriptsRemoteUserBackend(RemoteUserBackend): user.first_name = result[0][1]['givenName'][0] user.last_name = result[0][1]['sn'][0] user.email = result[0][1]['mail'][0] - user.save() + try: + user.groups.add(auth.models.Group.objects.get(name='mit')) + except ObjectDoesNotExist: + print "Failed to retrieve mit group" + try: + user.groups.add(auth.models.Group.objects.get(name='autocreated')) + except ObjectDoesNotExist: + print "Failed to retrieve autocreated group" + user.save() return user