def get_object(self):
"""Lookup a username by pk else use lookup_field"""
queryset = self.filter_queryset(self.get_queryset())
lookup = self.kwargs.get(self.lookup_field)
filter_kwargs = {self.lookup_field: lookup}
try:
pk = int(lookup)
except ValueError:
pass
else:
filter_kwargs = {'pk': pk}
obj = get_object_or_404(queryset, **filter_kwargs)
# May raise a permission denied
self.check_object_permissions(self.request, obj)
return obj
评论列表
文章目录