fnpdjango/auth_backends.py

   1 from django.conf import settings
   2 from django_cas.backends import CASBackend
   3
   4 attr_map = getattr(settings, 'CAS_USER_ATTRS_MAP', {
   5     'email': 'email',
   6     'firstname': 'first_name',
   7     'lastname': 'last_name',
   8 })
   9
  10 class AttrCASBackend(CASBackend):
  11     def authenticate(self, ticket, service, request):
  12         user = super(AttrCASBackend, self).authenticate(ticket, service, request)
  13         dirty = False
  14         for attr, value in request.session.get('attributes', {}).items():
  15             try:
  16                 local_attr = attr_map[attr]
  17             except KeyError:
  18                 pass
  19             else:
  20                 if getattr(user, local_attr, None) != value:
  21                     setattr(user, local_attr, value)
  22                     dirty = True
  23         if dirty:
  24             user.save()
  25         return user
  26