override _extract function in SAMLProvider without creating a new Provider

[mgogoulos]

Hi,

I would like to customize _extract function of SAMLProvider, in order to perform the following two actions:

1. get complete list of an attribute, not just the first key, as it happens now
2. provide the attributes mapping through a custom Django model, so that it is editable through the Django admin

This can work if I copy the entire saml directory as something like custom_saml, and then add this on the INSTALLED_APPS, but I'm wondering whether there a better way. Or maybe it makse sense to add _extract to be easily overriden, as the rest in https://docs.allauth.org/en/dev/account/adapter.html ?

Thanks and regards

[pennersr]

It's undocumented, but, you can override the provider class like so:

SOCIALACCOUNT_PROVIDERS = {
    "google": {
        "APP": {
           ...
        },
        "provider_class": "your.own.package.CustomGoogleProvider",
    },

Then, your custom provider class can simply inherit from the original one and tweak where needed.

[mgogoulos]

Thanks for this suggestion! Is there a way for this to work if I'm adding the provider through Django admin?

[pennersr]

That should work, just leave out the APP section.

[mgogoulos]

Confirm that this works, only thing I have to do is add this:

SOCIALACCOUNT_PROVIDERS = {
    "saml": {
        "provider_class": "custom_saml.provider.CustomSAMLProvider"
    }
}

Then I can override _extract and make the changes

# custom_saml.provider.CustomSAMLProvider
from allauth.socialaccount.providers.saml.provider import SAMLProvider

class CustomSAMLProvider(SAMLProvider):
    def _extract(self, data):

and this is more maintanable than creating a new provider.

Thanks a lot @pennersr !