Coverage for backend/core/auxiliary/serializers/FlowsheetSerializer.py: 100%

1from rest_framework import serializers 

2from core.auxiliary.models.Flowsheet import Flowsheet 

3 

4 

5class FlowsheetSerializer(serializers.ModelSerializer): 

6 

7 class Meta: 

8 model = Flowsheet 

9 fields = "__all__" 

10 

11 def create(self, validated_data): 

12 context = self.context 

13 request = context["request"] 

14 owner = request.user 

15 

16 validated_data["owner"] = owner 

17 

18 # Return primary key of Flowsheet 

19 return Flowsheet.create(**validated_data) 

20 

21 def validate(self, attrs): 

22 context = self.context 

23 request = context["request"] 

24 owner = request.user 

25 

26 if owner is None: 

27 raise serializers.ValidationError("Flowsheet must have an owner") 

28 

29 return super().validate(attrs)