Coverage for backend/django/core/auxiliary/viewsets/MonitoringTableViewSet.py: 90%

1from core.viewset import ModelViewSet 

2from drf_spectacular.utils import extend_schema 

3from drf_spectacular.types import OpenApiTypes 

4from drf_spectacular.openapi import OpenApiParameter 

5from ..models.MonitoringTable import MonitoringTable, MonitoringTableProperty 

6from ..serializers.MonitoringTableSerializer import MonitoringTableSerializer, MonitoringTablePropertySerializer 

7 

8class MonitoringTableViewSet(ModelViewSet): 

9 serializer_class = MonitoringTableSerializer 

10 

11 def get_queryset(self): 

12 return MonitoringTable.objects.all() 

13 

14 @extend_schema( 

15 parameters=[ 

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

17 ] 

18 ) 

19 def list(self, request): 

20 return super().list(request) 

21 

22 

23class MonitoringTablePropertyViewSet(ModelViewSet): 

24 serializer_class = MonitoringTablePropertySerializer 

25 

26 def get_queryset(self): 

27 return MonitoringTableProperty.objects.all() 

28 

29 @extend_schema( 

30 parameters=[ 

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

32 ] 

33 ) 

34 def list(self, request): 

35 return super().list(request)