Coverage for backend/django/core/auxiliary/viewsets/ControlValueViewSet.py: 94%

1from drf_spectacular.openapi import OpenApiParameter 

2from drf_spectacular.types import OpenApiTypes 

3from drf_spectacular.utils import extend_schema 

4 

5from flowsheetInternals.unitops.models.FlowsheetEditOperation import ( 

6 FlowsheetEditOperation, 

7) 

8from flowsheetInternals.unitops.services.edit_operations.mixins import ( 

9 FlowsheetMutationViewSet, 

10) 

11 

12from ..models.ControlValue import ControlValue 

13from ..serializers.ControlValueSerializer import ControlValueSerializer 

14 

15 

16class ControlValueViewSet(FlowsheetMutationViewSet): 

17 """Manage control replacements through the standard history-aware CRUD base.""" 

18 

19 serializer_class = ControlValueSerializer 

20 history_kind = FlowsheetEditOperation.Kind.FieldPatch 

21 history_label = "Change parameter replacement" 

22 history_create_label = "Replace parameter" 

23 history_destroy_label = "Remove parameter replacement" 

24 

25 def get_queryset(self): 

26 return ControlValue.objects.all() 

27 

28 @extend_schema( 

29 parameters=[ 

30 OpenApiParameter(name="flowsheet", required=True, type=OpenApiTypes.INT), 

31 ] 

32 ) 

33 def list(self, request): 

34 return super().list(request)