From: Alex Dehnert <adehnert@mit.edu>
Date: Fri, 2 Apr 2010 15:21:11 +0000 (-0400)
Subject: Auto-apply 'mit' and 'autocreated' groups
X-Git-Url: https://sipb.mit.edu/gitweb.cgi/snippets/.git/commitdiff_plain/358136f95f0bebfe9e09099bf683fb7663abe67c

Auto-apply 'mit' and 'autocreated' groups

This resolves the "we need rope" part of Trac #20.
---

diff --git a/django/mit/__init__.py b/django/mit/__init__.py
index fd8d452..ff296a8 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
@@ -29,5 +30,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]
+            try:
+                user.groups.add(auth.models.Group.objects.get(name='mit'))
+            except ObjectDoesNotExist:
+                print "Failed to retrieve mit group"
             user.save()
+        try:
+            user.groups.add(auth.models.Group.objects.get(name='autocreated'))
+        except ObjectDoesNotExist:
+            print "Failed to retrieve autocreated group"
         return user