Add Serializers

uktrade · Sep 28, 2021 · 2bb59b4 · 2bb59b4
1 parent c59f81a
commit 2bb59b4
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/api/cases/serializers.py b/api/cases/serializers.py
@@ -28,6 +28,7 @@
     GoodCountryDecision,
     CaseType,
     CaseReviewDate,
+    DepartmentSLA,
 )
 from api.compliance.models import ComplianceSiteCase, ComplianceVisitCase
 from api.compliance.serializers.ComplianceSiteCaseSerializers import ComplianceSiteViewSerializer
@@ -93,6 +94,12 @@ class Meta:
         fields = "__all__"
 
 
+class CaseDepartmentSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = DepartmentSLA
+        fields = "__all__"
+
+
 class QueueCaseAssignmentUserSerializer(serializers.ModelSerializer):
     class Meta:
         model = GovUser

diff --git a/api/teams/serializers.py b/api/teams/serializers.py
@@ -2,7 +2,7 @@
 from rest_framework import serializers
 from rest_framework.validators import UniqueValidator
 
-from api.teams.models import Team
+from api.teams.models import Team, Department
 
 
 class TeamReadOnlySerializer(serializers.Serializer):
@@ -48,3 +48,9 @@ def validate(self, attrs):
             raise serializers.ValidationError({"is_ogd": "Select yes if the team is an OGD"})
 
         return validated_data
+
+
+class DepartmentSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Department
+        fields = "__all__"