[Patchew-devel] [PATCH 2/8] rest: use lookup_field in HyperlinkedMessageField

[Paolo Bonzini]

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 at redhat.com>
---
 api/rest.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/api/rest.py b/api/rest.py
index 8c11050..0af1f30 100644
--- a/api/rest.py
+++ b/api/rest.py
@@ -99,8 +99,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