Add an user to a group automatically after SAML authentication

[tkloda]

Hello,

I'm trying to figure out a solution how to add an user to a group after this user is authenticated via SAML.
I was reading through the documentation and it seems overwriting this method might work:

from djangosaml2.backends import Saml2Backend

class MySaml2Backend(Saml2Backend):
    def save_user(self, user, *args, **kwargs):
        # Add custom logic here
        return super().save_user(user, *args, **kwargs)

My question is where should I add this piece of code in my Django project to get it working? I was trying to add it to my settings.py file, but I got an error: django.core.exceptions.AppRegistryNotReady: Apps aren't loaded yet.

Or is there any better solution how to add users automatically to Django groups after authentication?

Thanks in advance.

[tkloda]

I was able to figure it out. I added the modified class to my authentication.py file where I'm storing my authentication backends.

class ModifiedSaml2Backend(Saml2Backend):
    def save_user(self, user, *args, **kwargs):
        user.save()
        user_group = Group.objects.get(name='Default') 
        user.groups.add(user_group)
        return super().save_user(user, *args, **kwargs)

Then I just referenced the modified Saml2Backend in settings.py file:

AUTHENTICATION_BACKENDS = [
    'netbox.authentication.ModifiedSaml2Backend',
]

Works like a charm.

[peppelinux]

@tkloda
take a look to documentation, find the place that you think it would be better to extend it with the experience you shared here, do a PR (please to dev branch), I'll merge it asap

thank you for contribution, we're community

[peppelinux]

Thank you @tkloda
c9ac2e7