diff --git a/app/api/urls_v2.py b/app/api/urls_v2.py index 37f78027..524a140f 100644 --- a/app/api/urls_v2.py +++ b/app/api/urls_v2.py @@ -75,6 +75,7 @@ from itim.viewsets import ( index as itim_v2, change, cluster as cluster_v2, + cluster_notes, cluster_type as cluster_type_v2, incident, port as port_v2, @@ -165,7 +166,7 @@ router.register('itim', itim_v2.Index, basename='_api_v2_itim_home') router.register('itim/ticket/change', change.ViewSet, basename='_api_v2_ticket_change') router.register('itim/cluster', cluster_v2.ViewSet, basename='_api_v2_cluster') router.register('itim/cluster/(?P[0-9]+)/service', service_cluster.ViewSet, basename='_api_v2_service_cluster') -# router.register('itim/cluster/(?P[0-9]+)/notes', notes_v2.ViewSet, basename='_api_v2_cluster_notes') +router.register('itim/cluster/(?P[0-9]+)/notes', cluster_notes.ViewSet, basename='_api_v2_cluster_note') router.register('itim/ticket/incident', incident.ViewSet, basename='_api_v2_ticket_incident') router.register('itim/ticket/problem', problem.ViewSet, basename='_api_v2_ticket_problem') router.register('itim/service', service_v2.ViewSet, basename='_api_v2_service') diff --git a/app/itim/serializers/cluster.py b/app/itim/serializers/cluster.py index 96ad047a..04386e72 100644 --- a/app/itim/serializers/cluster.py +++ b/app/itim/serializers/cluster.py @@ -71,7 +71,13 @@ class ClusterModelSerializer( 'model_pk': item.pk } ), - 'notes': reverse("v2:_api_v2_cluster_notes-list", request=self._context['view'].request, kwargs={'cluster_id': item.pk}), + 'notes': reverse( + "v2:_api_v2_cluster_note-list", + request=self._context['view'].request, + kwargs={ + 'model_id': item.pk + } + ), 'service': reverse("v2:_api_v2_service_cluster-list", request=self._context['view'].request, kwargs={'cluster_id': item.pk}), 'tickets': reverse( "v2:_api_v2_item_tickets-list", diff --git a/app/itim/viewsets/cluster_notes.py b/app/itim/viewsets/cluster_notes.py new file mode 100644 index 00000000..907c5ac1 --- /dev/null +++ b/app/itim/viewsets/cluster_notes.py @@ -0,0 +1,66 @@ +from drf_spectacular.utils import extend_schema, extend_schema_view, OpenApiResponse + +from core.viewsets.model_notes import ModelNoteViewSet + +from itim.serializers.cluster_notes import ( + ClusterNotes, + ClusterNoteModelSerializer, + ClusterNoteViewSerializer +) + + + + +@extend_schema_view( + create=extend_schema( + summary = 'Add a note to a Cluster', + description = '', + responses = { + 201: OpenApiResponse(description='created', response=ClusterNoteViewSerializer), + 400: OpenApiResponse(description='Validation failed.'), + 403: OpenApiResponse(description='User is missing create permissions'), + } + ), + destroy = extend_schema( + summary = 'Delete a Cluster note', + description = '' + ), + list = extend_schema( + summary = 'Fetch all Cluster notes', + description='', + ), + retrieve = extend_schema( + summary = 'Fetch a single Cluster note', + description='', + ), + update = extend_schema(exclude = True), + partial_update = extend_schema( + summary = 'Update a Cluster note', + description = '' + ), +) +class ViewSet(ModelNoteViewSet): + + model = ClusterNotes + + + def get_serializer_class(self): + + if self.serializer_class is not None: + + return self.serializer_class + + + if ( + self.action == 'list' + or self.action == 'retrieve' + ): + + self.serializer_class = ClusterNoteViewSerializer + + + else: + + self.serializer_class = ClusterNoteModelSerializer + + return self.serializer_class