encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2024-11-26 19:43:59 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add view_name argument to HyperlinkedIdentityField

Browse Source
This commit is contained in:
Tom Christie 2012-10-05 15:44:08 +01:00
parent 3e77090b21
commit 484ee8cc27
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
rest_framework/fields.py
View File

@ -378,9 +378,16 @@ class HyperlinkedIdentityField(Field):
""" """
A field that represents the model's identity using a hyperlink. A field that represents the model's identity using a hyperlink.
""" """
def __init__(self, *args, **kwargs):
try:
self.view_name = kwargs.pop('view_name')
except:
raise ValueError("Hyperlinked field requires 'view_name' kwarg")
super(HyperlinkedRelatedField, self).__init__(*args, **kwargs)
def field_to_native(self, obj, field_name): def field_to_native(self, obj, field_name):
request = self.context.get('request', None) request = self.context.get('request', None)
view_name = self.parent.opts.view_name view_name = self.view_name or self.parent.opts.view_name
view_kwargs = {'pk': obj.pk} view_kwargs = {'pk': obj.pk}
return reverse(view_name, kwargs=view_kwargs, request=request) return reverse(view_name, kwargs=view_kwargs, request=request)