[WIP] Add Swagger

freemusicninja/settings.py

@@ -50,6 +50,7 @@
     'raven.contrib.django.raven_compat',
     'rest_framework',
     'rest_framework.authtoken',
+    'rest_framework_swagger',
 
     'artists',
     'echonest',

freemusicninja/urls.py

@@ -16,5 +16,6 @@
                                namespace='rest_framework')),
     url(r'^api-token-auth/',
         'rest_framework.authtoken.views.obtain_auth_token'),
+    url(r'^docs/', include('rest_framework_swagger.urls')),
     url(r'^admin/', include(admin.site.urls)),
 ]

requirements.txt

@@ -5,6 +5,7 @@ django-cors-headers==0.13
 django-filter==0.8
 django-jsonfield==0.9.13
 django-model-utils==2.2.0
+django-rest-swagger==0.2.0
 djangorestframework==2.4.3
 psycopg2==2.5.4
 purl==1.0

users/tests/test_api.py

@@ -12,7 +12,6 @@ class UserTest(APITestCase):
     def test_list_users_logged_out(self):
         url = reverse('user-list')
         response = self.client.get(url, format='json')
-        print(response.data)
         assert response.status_code == status.HTTP_200_OK
         assert (response.data == [{
             'id': 1,

users/views.py

@@ -12,17 +12,16 @@ class UserViewSet(viewsets.ModelViewSet):
 
     queryset = User.objects.all()
     permission_classes = (IsUserOrReadOnly,)
+    serializer_class = UserSerializer
 
-    def get_serializer_class(self):
-        try:
-            obj = self.get_object()
-        except ImproperlyConfigured:  # assume list view, use non-authenticated
-            pass
+    def get_serializer(self, instance=None, *args, **kwargs):
+        context = self.get_serializer_context()
+        if (self.request.user.is_authenticated()
+                and instance == self.request.user):
+            serializer_class = AuthenticatedUserSerializer
         else:
-            if (self.request.user.is_authenticated()
-                    and obj == self.request.user):
-                return AuthenticatedUserSerializer
-        return UserSerializer
+            serializer_class = self.serializer_class
+        return serializer_class(instance, context=context, *args, **kwargs)
 
     def initial(self, request, *args, **kwargs):
         """Retrieve given user or current user if ``pk`` is "me"."""