diff --git a/django3_saml2_nbplugin/backends.py b/django3_saml2_nbplugin/backends.py index de0eed8..364b11d 100644 --- a/django3_saml2_nbplugin/backends.py +++ b/django3_saml2_nbplugin/backends.py @@ -171,8 +171,10 @@ def configure_user(self, request: WSGIRequest, user: User) -> User: pass if "FLAGS_BY_GROUP" in be_settings and "GROUP_ATTR" in be_settings: - for flag, group_name in be_settings["FLAGS_BY_GROUP"].items(): - if group_name in ident_groups: + for flag, group_names in be_settings["FLAGS_BY_GROUP"].items(): + if not isinstance(group_names, list): + group_names = [group_names] + if any(group_name in ident_groups for group_name in group_names): setattr(user, flag, True) else: setattr(user, flag, False)