Coverage for backend/core/plots/PlotViewSet.py: 80%

1from .Plot import Plot, Series 

2from core.viewset import ModelViewSet 

3from drf_spectacular.utils import extend_schema, OpenApiParameter, OpenApiTypes 

4from .PlotSerializer import PlotSerializer, SeriesSerializer 

5class PlotViewSet(ModelViewSet): 

6 serializer_class = PlotSerializer 

7 

8 def get_queryset(self): 

9 return Plot.objects.all() 

10 

11 @extend_schema( 

12 parameters=[ 

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

14 ] 

15 ) 

16 def list(self, request): 

17 return super().list(request) 

18 

19 

20class SeriesViewSet(ModelViewSet): 

21 serializer_class = SeriesSerializer 

22 

23 def get_queryset(self): 

24 return Series.objects.all()