X-Git-Url: https://snippets.scripts.mit.edu/gitweb.cgi/Scripts/git/.git/blobdiff_plain/34b7ba86375b9ca215d3e8b9e48e420fc801c606..48dbd9dcb855a14ce5a45d95aafe2d6c73ca9feb:/django/mit/__init__.py

diff --git a/django/mit/__init__.py b/django/mit/__init__.py
index e496dbd..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
@@ -17,3 +18,26 @@ class ScriptsRemoteUserBackend(RemoteUserBackend):
             return name
         else:
             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("", "")
+        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)
+        if len(result) == 1:
+            user.first_name = result[0][1]['givenName'][0]
+            user.last_name = result[0][1]['sn'][0]
+            user.email = result[0][1]['mail'][0]
+            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