[PATCHv2 10/10] REST: Allow projects to be retrieved by linkname

[Andy Doan]

Building a user-friendly CLI becomes difficult when project-ids are
required. It also makes it almost impossible to work with the current
format of the .pwclientrc file.

Signed-off-by: Andy Doan <andy.doan at linaro.org>
---
 patchwork/tests/test_rest_api.py |  5 +++++
 patchwork/views/rest_api.py      | 16 ++++++++++++++++
 2 files changed, 21 insertions(+)

diff --git a/patchwork/tests/test_rest_api.py b/patchwork/tests/test_rest_api.py
index 6f1a375..0b180e6 100644
--- a/patchwork/tests/test_rest_api.py
+++ b/patchwork/tests/test_rest_api.py
@@ -59,6 +59,11 @@ class TestProjectAPI(APITestCase):
         self.assertEqual(status.HTTP_200_OK, resp.status_code)
         self.assertEqual(defaults.project.name, resp.data['name'])
 
+        # make sure we can look up by linkname
+        resp = self.client.get(self.api_url(resp.data['linkname']))
+        self.assertEqual(status.HTTP_200_OK, resp.status_code)
+        self.assertEqual(defaults.project.name, resp.data['name'])
+
     def test_anonymous_writes(self):
         """Ensure anonymous "write" operations are rejected."""
         defaults.project.save()
diff --git a/patchwork/views/rest_api.py b/patchwork/views/rest_api.py
index e5a642a..e125bc1 100644
--- a/patchwork/views/rest_api.py
+++ b/patchwork/views/rest_api.py
@@ -113,6 +113,22 @@ class ProjectViewSet(PatchworkViewSet):
     permission_classes = (PatchworkPermission,)
     serializer_class = ProjectSerializer
 
+    def retrieve(self, request, pk=None):
+        try:
+            int(pk)
+        except ValueError:
+            self.kwargs = {'linkname': pk}  # try and lookup by linkname
+            self.lookup_field = 'linkname'
+        return super(ProjectViewSet, self).retrieve(request, pk)
+
+    def partial_update(self, request, pk=None):
+        try:
+            int(pk)
+        except ValueError:
+            self.kwargs = {'linkname': pk}  # try and lookup by linkname
+            self.lookup_field = 'linkname'
+        return super(ProjectViewSet, self).partial_update(request, pk)
+
 
 class ChecksViewSet(PatchworkViewSet):
     serializer_class = ChecksSerializer
-- 
2.7.4