Endpoint to retrieve all the connected social accounts

Author: humitos

readthedocs/restapi/serializers.py

@@ -3,6 +3,8 @@
 from __future__ import absolute_import
 
 from builtins import object
+
+from allauth.socialaccount.models import SocialAccount
 from rest_framework import serializers
 
 from readthedocs.builds.models import Build, BuildCommandResult, Version
@@ -157,3 +159,26 @@ def get_matches(self, obj):
         request = self.context['request']
         if request.user is not None and request.user.is_authenticated():
             return obj.matches(request.user)
+
+
+class ProviderSerializer(serializers.Serializer):
+
+    name = serializers.CharField(max_length=20)
+
+
+class SocialAccountSerializer(serializers.ModelSerializer):
+
+    username = serializers.SerializerMethodField()
+    avatar_url = serializers.URLField(source='get_avatar_url')
+    provider = ProviderSerializer(source='get_provider')
+
+    class Meta(object):
+        model = SocialAccount
+        exclude = ('extra_data',)
+
+    def get_username(self, obj):
+        return (
+            obj.extra_data.get('username') or
+            obj.extra_data.get('login')
+            # FIXME: which one is GitLab?
+        )

readthedocs/restapi/urls.py

@@ -17,7 +17,8 @@
                                 ProjectViewSet, NotificationViewSet,
                                 VersionViewSet, DomainViewSet,
                                 RemoteOrganizationViewSet,
-                                RemoteRepositoryViewSet)
+                                RemoteRepositoryViewSet,
+                                SocialAccountViewSet)
 
 router = routers.DefaultRouter()
 router.register(r'build', BuildViewSet, base_name='build')
@@ -30,6 +31,8 @@
     r'remote/org', RemoteOrganizationViewSet, base_name='remoteorganization')
 router.register(
     r'remote/repo', RemoteRepositoryViewSet, base_name='remoterepository')
+router.register(
+    r'remote/account', SocialAccountViewSet, base_name='remoteaccount')
 router.register(r'comments', CommentViewSet, base_name="comments")
 
 urlpatterns = [

readthedocs/restapi/views/model_views.py

@@ -3,6 +3,7 @@
 from __future__ import absolute_import
 import logging
 
+from allauth.socialaccount.models import SocialAccount
 from django.shortcuts import get_object_or_404
 from rest_framework import decorators, permissions, viewsets, status
 from rest_framework.decorators import detail_route
@@ -26,7 +27,7 @@
                            ProjectSerializer, ProjectAdminSerializer,
                            VersionSerializer, VersionAdminSerializer,
                            DomainSerializer, RemoteOrganizationSerializer,
-                           RemoteRepositorySerializer)
+                           RemoteRepositorySerializer, SocialAccountSerializer)
 from .. import utils as api_utils
 
 log = logging.getLogger(__name__)
@@ -274,3 +275,13 @@ def get_queryset(self):
 
     def get_paginate_by(self):
         return self.request.query_params.get('page_size', 25)
+
+
+class SocialAccountViewSet(viewsets.ReadOnlyModelViewSet):
+    permission_classes = [IsOwner]
+    renderer_classes = (JSONRenderer,)
+    serializer_class = SocialAccountSerializer
+    model = SocialAccount
+
+    def get_queryset(self):
+        return self.model.objects.filter(user=self.request.user.pk)