Advertisement
Guest User

Untitled

a guest
Aug 26th, 2019
93
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.52 KB | None | 0 0
  1. from rest_framework.filters import BaseFilterBackend
  2.  
  3. class BookView(ListAPIView):
  4. queryset = Book.objects.all() # Book.author_id
  5.  
  6. filter_backend = [RequestUserFilterBackend]
  7. user_pk_lookup = 'author_id' # queryset.filter(author_id=request.user.pk)
  8.  
  9. class RequestUserFilterBackend(BaseFilterBackend):
  10. def filter_queryset(self, request, queryset, view):
  11. assert hasattr(view, 'user_pk_lookup') and isinstance(view.user_pk_lookup, str)
  12.  
  13. return queryset.filter(**{view.user_pk_lookup: request.user.pk})
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement