From 358136f95f0bebfe9e09099bf683fb7663abe67c Mon Sep 17 00:00:00 2001 From: Alex Dehnert Date: Fri, 2 Apr 2010 11:21:11 -0400 Subject: [PATCH] Auto-apply 'mit' and 'autocreated' groups This resolves the "we need rope" part of Trac #20. --- django/mit/__init__.py | 9 +++++++++ 1 file changed, 9 insertions(+) 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 -- 2.45.0