Coverage for backend/core/auxiliary/viewsets/NoteViewSet.py: 62%

1from drf_spectacular.utils import extend_schema, OpenApiParameter, OpenApiTypes 

2from core.viewset import ModelViewSet 

3from core.auxiliary.models.Note import Note 

4from core.auxiliary.serializers.NoteSerializer import NoteSerializer 

5class NoteViewSet(ModelViewSet): 

6 serializer_class = NoteSerializer 

7 

8 def get_queryset(self, **kwargs): 

9 simulationObjectId = self.request.query_params.get('simulationObject', None) 

10 return Note.objects.filter(simulationObject=simulationObjectId) 

11 

12 @extend_schema( 

13 parameters=[ 

14 OpenApiParameter(name="simulationObject", required=True, type=OpenApiTypes.INT), 

15 ] 

16 ) 

17 def list(self, request): 

18 return super().list(request) 

19 

20 def update(self, request, *args, **kwargs): 

21 # Update the savedDate field 

22 instance = self.get_object() 

23 instance.set_saved_date() 

24 return super().update(request, *args, **kwargs)