Coverage for backend/django/core/auxiliary/viewsets/DataCellViewSet.py: 73%

1from core.viewset import ModelViewSet 

2from core.auxiliary.models.DataCell import DataCell 

3from core.auxiliary.serializers.DataCellSerializer import DataCellSerializer 

4from core.auxiliary.services.mss_input_data import mark_mss_input_rows_changed 

5 

6class DataCellViewSet(ModelViewSet): 

7 

8 serializer_class = DataCellSerializer 

9 

10 def get_queryset(self): 

11 queryset = DataCell.objects.all() 

12 return queryset 

13 

14 def list(self, request): 

15 return super().list(request) 

16 

17 def perform_create(self, serializer): 

18 instance = serializer.save() 

19 mark_mss_input_rows_changed(data_row_ids=[instance.data_row_id]) 

20 

21 def perform_update(self, serializer): 

22 previous_data_row_id = serializer.instance.data_row_id 

23 instance = serializer.save() 

24 mark_mss_input_rows_changed(data_row_ids=[previous_data_row_id, instance.data_row_id]) 

25 

26 def perform_destroy(self, instance): 

27 data_row_id = instance.data_row_id 

28 super().perform_destroy(instance) 

29 mark_mss_input_rows_changed(data_row_ids=[data_row_id])