X-Git-Url: https://snippets.scripts.mit.edu/gitweb.cgi/Scripts/git/.git/blobdiff_plain/9c504e485be1758ee01e7f577b9649633daae6ca..48dbd9dcb855a14ce5a45d95aafe2d6c73ca9feb:/django/mit/__init__.py diff --git a/django/mit/__init__.py b/django/mit/__init__.py index fd8d452..acdb674 100644 --- a/django/mit/__init__.py +++ b/django/mit/__init__.py @@ -1,6 +1,7 @@ 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 @@ -19,6 +20,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 +31,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