Repository URL to install this package:
Version:
0.2.0a11 ▾
|
from rest_framework import viewsets
from .querysets import UserAwareQuerySet
class UserAwareViewSetMixin(viewsets.GenericViewSet):
def get_queryset(self):
queryset: UserAwareQuerySet = super().get_queryset()
if self.request.user.is_superuser:
return queryset
return queryset.filter_by_user(self.request.user)