dashboard: handle keyerrors in saml attributes

circle/dashboard/models.py

@@ -165,7 +165,8 @@ if hasattr(settings, 'SAML_ORG_ID_ATTRIBUTE'):
             logger.debug("org_id of %s already added to user %s's profile",
             logger.debug("org_id of %s already added to user %s's profile",
                          value, sender.username)
                          value, sender.username)
         memberatrs = getattr(settings, 'SAML_GROUP_ATTRIBUTES', [])
         memberatrs = getattr(settings, 'SAML_GROUP_ATTRIBUTES', [])
-        for group in chain(*[attributes[i] for i in memberatrs]):
+        for group in chain(*[attributes[i]
+                             for i in memberatrs if i in attributes]):
             try:
             try:
                 g = GroupProfile.search(group)
                 g = GroupProfile.search(group)
             except Group.DoesNotExist:
             except Group.DoesNotExist:
@@ -176,7 +177,8 @@ if hasattr(settings, 'SAML_ORG_ID_ATTRIBUTE'):
                 g.user_set.add(sender)
                 g.user_set.add(sender)
 
 
         owneratrs = getattr(settings, 'SAML_GROUP_OWNER_ATTRIBUTES', [])
         owneratrs = getattr(settings, 'SAML_GROUP_OWNER_ATTRIBUTES', [])
-        for group in chain(*[attributes[i] for i in owneratrs]):
+        for group in chain(*[attributes[i]
+                             for i in owneratrs if i in attributes]):
             try:
             try:
                 g = GroupProfile.search(group)
                 g = GroupProfile.search(group)
             except Group.DoesNotExist:
             except Group.DoesNotExist: