Mail

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

@@ -0,0 +1,21 @@
+from allauth.socialaccount.providers.base import ProviderAccount
+from allauth.socialaccount.providers.oauth2.provider import OAuth2Provider
+
+
+class DigitalOceanAccount(ProviderAccount):
+    pass
+
+
+class DigitalOceanProvider(OAuth2Provider):
+    id = "digitalocean"
+    name = "DigitalOcean"
+    account_class = DigitalOceanAccount
+
+    def extract_uid(self, data):
+        return str(data["account"]["uuid"])
+
+    def extract_common_fields(self, data):
+        return dict(email=data["account"]["email"])
+
+
+provider_classes = [DigitalOceanProvider]

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

@@ -0,0 +1,41 @@
+from allauth.socialaccount.tests import OAuth2TestsMixin
+from allauth.tests import MockedResponse, TestCase
+
+from .provider import DigitalOceanProvider
+
+
+class DigitalOceanTests(OAuth2TestsMixin, TestCase):
+    provider_id = DigitalOceanProvider.id
+
+    def get_mocked_response(self):
+        return MockedResponse(
+            200,
+            """
+        {
+          "account": {
+            "droplet_limit": 25,
+            "floating_ip_limit": 5,
+            "email": "sammy@example.com",
+            "uuid": "b6fr89dbf6d9156cace5f3c78dc9851d957381ef",
+            "email_verified": true,
+            "status": "active",
+            "status_message": ""
+          }
+        }
+        """,
+        )
+
+    def get_login_response_json(self, with_refresh_token=True):
+        return """
+        {
+          "access_token": "testac",
+          "token_type": "bearer",
+          "expires_in": 2592000,
+          "refresh_token": "00a3aae641658d",
+          "scope": "read write",
+          "info": {
+            "name": "Sammy the Shark",
+            "email":"sammy@example.com",
+            "uuid":"b6fr89dbf6d9156cace5f3c78dc9851d957381ef"
+          }
+        }"""

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

@@ -0,0 +1,6 @@
+from allauth.socialaccount.providers.oauth2.urls import default_urlpatterns
+
+from .provider import DigitalOceanProvider
+
+
+urlpatterns = default_urlpatterns(DigitalOceanProvider)

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

@@ -0,0 +1,26 @@
+import requests
+
+from allauth.socialaccount.providers.oauth2.views import (
+    OAuth2Adapter,
+    OAuth2CallbackView,
+    OAuth2LoginView,
+)
+
+from .provider import DigitalOceanProvider
+
+
+class DigitalOceanOAuth2Adapter(OAuth2Adapter):
+    provider_id = DigitalOceanProvider.id
+    access_token_url = "https://cloud.digitalocean.com/v1/oauth/token"
+    authorize_url = "https://cloud.digitalocean.com/v1/oauth/authorize"
+    profile_url = "https://api.digitalocean.com/v2/account"
+
+    def complete_login(self, request, app, token, **kwargs):
+        headers = {"Authorization": "Bearer {0}".format(token.token)}
+        resp = requests.get(self.profile_url, headers=headers)
+        extra_data = resp.json()
+        return self.get_provider().sociallogin_from_response(request, extra_data)
+
+
+oauth2_login = OAuth2LoginView.adapter_view(DigitalOceanOAuth2Adapter)
+oauth2_callback = OAuth2CallbackView.adapter_view(DigitalOceanOAuth2Adapter)