Add and apply custom ViewSet classes

addon_service/authorized_storage_account/views.py

@@ -3,7 +3,8 @@
 from .models import AuthorizedStorageAccount
 from .serializers import AuthorizedStorageAccountSerializer
 
+from addon_service.common.viewsets import RetrieveWriteViewSet
 
-class AuthorizedStorageAccountViewSet(ModelViewSet):
+class AuthorizedStorageAccountViewSet(RetrieveWriteViewSet):
     queryset = AuthorizedStorageAccount.objects.all()
     serializer_class = AuthorizedStorageAccountSerializer

addon_service/common/viewsets.py

@@ -0,0 +1,21 @@
+from rest_framework.viewsets import GenericViewSet
+from rest_framework import mixins as drf_mixins
+from rest_framework_json_api.views import AutoPrefetchMixin, PreloadIncludesMixin, RelatedMixin
+
+
+class _DrfJsonApiHelpers(AutoPrefetchMixin, PreloadIncludesMixin, RelatedMixin):
+    pass
+
+class RetrieveOnlyViewSet(_DrfJsonApiHelpers, drf_mixins.RetrieveModelMixin, GenericViewSet):
+    http_method_names = ["get", "head", "options"]
+
+
+class RetrieveWriteViewSet(
+    _DrfJsonApiHelpers,
+    drf_mixins.CreateModelMixin,
+    drf_mixins.RetrieveModelMixin,
+    drf_mixins.UpdateModelMixin,
+    drf_mixins.DestroyModelMixin,
+    GenericViewSet
+):
+    http_method_names = ["get", "post", "patch", "delete", "head", "options"]

addon_service/configured_storage_addon/views.py

@@ -1,9 +1,8 @@
-from rest_framework_json_api.views import ModelViewSet
-
 from .models import ConfiguredStorageAddon
 from .serializers import ConfiguredStorageAddonSerializer
 
+from addon_service.common.viewsets import RetrieveWriteViewSet
 
-class ConfiguredStorageAddonViewSet(ModelViewSet):
+class ConfiguredStorageAddonViewSet(RetrieveWriteViewSet):
     queryset = ConfiguredStorageAddon.objects.all()
     serializer_class = ConfiguredStorageAddonSerializer

addon_service/external_storage_service/views.py

@@ -1,10 +1,10 @@
-from rest_framework_json_api.views import ModelViewSet
+from rest_framework_json_api.views import ReadOnlyViewSet
 
 from .models import ExternalStorageService
 from .serializers import ExternalStorageServiceSerializer
 
 
-class ExternalStorageServiceViewSet(ModelViewSet):
+class ExternalStorageServiceViewSet(ReadOnlyViewSet):
     queryset = ExternalStorageService.objects.all()
     serializer_class = ExternalStorageServiceSerializer
     # TODO: permissions_classes

addon_service/resource_reference/views.py

@@ -1,10 +1,10 @@
-from rest_framework_json_api.views import ReadOnlyModelViewSet
-
 from .models import ResourceReference
 from .serializers import ResourceReferenceSerializer
 
+from addon_service.common.viewsets import RetrieveOnlyViewSet
 
-class ResourceReferenceViewSet(ReadOnlyModelViewSet):
+class ResourceReferenceViewSet(RetrieveOnlyViewSet):
     queryset = ResourceReference.objects.all()
     serializer_class = ResourceReferenceSerializer
     # TODO: permissions_classes
+

addon_service/user_reference/views.py

@@ -1,10 +1,9 @@
-from rest_framework_json_api.views import ReadOnlyModelViewSet
-
 from .models import UserReference
 from .serializers import UserReferenceSerializer
 
+from addon_service.common.viewsets import RetrieveOnlyViewSet
 
-class UserReferenceViewSet(ReadOnlyModelViewSet):
+class UserReferenceViewSet(RetrieveOnlyViewSet):
     queryset = UserReference.objects.all()
     serializer_class = UserReferenceSerializer
     # TODO: permissions_classes