restapi: decouple ProjectViewSet from the pk

We may want to hook the very same viewset under a different
router in order to lookup by slug and not pk. If we use
lookup_field instead of hardcoding it in the views, it becomes
easier. While at it reuse get_queryset() method for filtering
the queryset.

Author: xrmx

readthedocs/restapi/views/model_views.py

@@ -94,6 +94,11 @@ class ProjectViewSet(UserSelectViewSet):
     filter_fields = ('slug',)  # django-filter<2.0.0
     filterset_fields = ('slug',)
 
+    def get_project_for_user_or_404(self, lookup_value):
+        lookup_query = {self.lookup_field: lookup_value}
+        qs = self.get_queryset()
+        return get_object_or_404(qs, **lookup_query)
+
     @decorators.action(detail=True)
     def translations(self, *_, **__):
         translations = self.get_object().translations.all()
@@ -103,8 +108,8 @@ def translations(self, *_, **__):
 
     @decorators.action(detail=True)
     def subprojects(self, request, **kwargs):
-        project = get_object_or_404(
-            Project.objects.api(request.user), pk=kwargs['pk'])
+        project = self.get_project_for_user_or_404(
+            kwargs[self.lookup_field])
         rels = project.subprojects.all()
         children = [rel.child for rel in rels]
         return Response({
@@ -113,26 +118,26 @@ def subprojects(self, request, **kwargs):
 
     @decorators.action(detail=True)
     def active_versions(self, request, **kwargs):
-        project = get_object_or_404(
-            Project.objects.api(request.user), pk=kwargs['pk'])
+        project = self.get_project_for_user_or_404(
+            kwargs[self.lookup_field])
         versions = project.versions.filter(active=True)
         return Response({
             'versions': VersionSerializer(versions, many=True).data,
         })
 
     @decorators.action(detail=True, permission_classes=[permissions.IsAdminUser])
     def token(self, request, **kwargs):
-        project = get_object_or_404(
-            Project.objects.api(request.user), pk=kwargs['pk'])
+        project = self.get_project_for_user_or_404(
+            kwargs[self.lookup_field])
         token = GitHubService.get_token_for_project(project, force_local=True)
         return Response({
             'token': token,
         })
 
     @decorators.action(detail=True)
     def canonical_url(self, request, **kwargs):
-        project = get_object_or_404(
-            Project.objects.api(request.user), pk=kwargs['pk'])
+        project = self.get_project_for_user_or_404(
+            kwargs[self.lookup_field])
         return Response({
             'url': project.get_docs_url(),
         })
@@ -149,8 +154,8 @@ def sync_versions(self, request, **kwargs):  # noqa: D205
 
         :returns: the identifiers for the versions that have been deleted.
         """
-        project = get_object_or_404(
-            Project.objects.api(request.user), pk=kwargs['pk'])
+        project = self.get_project_for_user_or_404(
+            kwargs[self.lookup_field])
 
         # If the currently highest non-prerelease version is active, then make
         # the new latest version active as well.