Mail

venv/lib/python3.12/site-packages/allauth/socialaccount/providers/lemonldap/provider.py

@@ -0,0 +1,35 @@
+# -*- coding: utf-8 -*-
+from allauth.socialaccount.providers.base import ProviderAccount
+from allauth.socialaccount.providers.oauth2.provider import OAuth2Provider
+
+
+class LemonLDAPAccount(ProviderAccount):
+    def get_avatar_url(self):
+        return self.account.extra_data.get("picture")
+
+    def to_str(self):
+        dflt = super(LemonLDAPAccount, self).to_str()
+        return self.account.extra_data.get("name", dflt)
+
+
+class LemonLDAPProvider(OAuth2Provider):
+    id = "lemonldap"
+    name = "LemonLDAP::NG"
+    account_class = LemonLDAPAccount
+
+    def get_default_scope(self):
+        return ["openid", "profile", "email"]
+
+    def extract_uid(self, data):
+        return str(data["id"])
+
+    def extract_common_fields(self, data):
+        return dict(
+            email=data.get("email"),
+            username=data.get("preferred_username"),
+            name=data.get("name"),
+            picture=data.get("picture"),
+        )
+
+
+provider_classes = [LemonLDAPProvider]

venv/lib/python3.12/site-packages/allauth/socialaccount/providers/lemonldap/tests.py

@@ -0,0 +1,23 @@
+# -*- coding: utf-8 -*-
+from allauth.socialaccount.providers.lemonldap.provider import (
+    LemonLDAPProvider,
+)
+from allauth.socialaccount.tests import OAuth2TestsMixin
+from allauth.tests import MockedResponse, TestCase
+
+
+class LemonLDAPTests(OAuth2TestsMixin, TestCase):
+    provider_id = LemonLDAPProvider.id
+
+    def get_mocked_response(self):
+        return MockedResponse(
+            200,
+            """
+            {
+                "email": "dwho@example.com",
+                "sub": "dwho",
+                "preferred_username": "dwho",
+                "name": "Doctor Who"
+            }
+        """,
+        )

venv/lib/python3.12/site-packages/allauth/socialaccount/providers/lemonldap/urls.py

@@ -0,0 +1,8 @@
+# -*- coding: utf-8 -*-
+from allauth.socialaccount.providers.lemonldap.provider import (
+    LemonLDAPProvider,
+)
+from allauth.socialaccount.providers.oauth2.urls import default_urlpatterns
+
+
+urlpatterns = default_urlpatterns(LemonLDAPProvider)

venv/lib/python3.12/site-packages/allauth/socialaccount/providers/lemonldap/views.py

@@ -0,0 +1,39 @@
+# -*- coding: utf-8 -*-
+import requests
+
+from allauth.socialaccount import app_settings
+from allauth.socialaccount.providers.lemonldap.provider import (
+    LemonLDAPProvider,
+)
+from allauth.socialaccount.providers.oauth2.views import (
+    OAuth2Adapter,
+    OAuth2CallbackView,
+    OAuth2LoginView,
+)
+
+
+class LemonLDAPOAuth2Adapter(OAuth2Adapter):
+    provider_id = LemonLDAPProvider.id
+    supports_state = True
+
+    settings = app_settings.PROVIDERS.get(provider_id, {})
+    provider_base_url = settings.get("LEMONLDAP_URL")
+
+    access_token_url = "{0}/oauth2/token".format(provider_base_url)
+    authorize_url = "{0}/oauth2/authorize".format(provider_base_url)
+    profile_url = "{0}/oauth2/userinfo".format(provider_base_url)
+
+    def complete_login(self, request, app, token, response):
+        response = requests.post(
+            self.profile_url, headers={"Authorization": "Bearer " + str(token)}
+        )
+        response.raise_for_status()
+        extra_data = response.json()
+        extra_data["id"] = extra_data["sub"]
+        del extra_data["sub"]
+
+        return self.get_provider().sociallogin_from_response(request, extra_data)
+
+
+oauth2_login = OAuth2LoginView.adapter_view(LemonLDAPOAuth2Adapter)
+oauth2_callback = OAuth2CallbackView.adapter_view(LemonLDAPOAuth2Adapter)