Allow overriding the lookup field, so that it can be used with views that are
not based on BaseMessageViewSet.  The next patch will use it to include the URI
for results.

Signed-off-by: Paolo Bonzini <pbonzini@redhat.com>
---
 api/rest.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/api/rest.py b/api/rest.py
index 0bb3c30..7df7618 100644
--- a/api/rest.py
+++ b/api/rest.py
@@ -101,8 +101,9 @@ class ProjectsViewSet(viewsets.ModelViewSet):
 # Common classes for series and messages
 
 class HyperlinkedMessageField(HyperlinkedIdentityField):
+    lookup_field = 'message_id'
     def get_url(self, obj, view_name, request, format):
-        kwargs = {'projects_pk': obj.project_id, 'message_id': obj.message_id}
+        kwargs = {'projects_pk': obj.project_id, self.lookup_field: obj.message_id}
         return self.reverse(view_name, kwargs=kwargs, request=request, format=format)
 
 class BaseMessageSerializer(serializers.ModelSerializer):
-- 
2.16.2


_______________________________________________
Patchew-devel mailing list
Patchew-devel@redhat.com
https://www.redhat.com/mailman/listinfo/patchew-devel