@@ -1795,6 +1795,13 @@ def validate(self, data):
17951795
17961796 return data
17971797
1798+ def validate_severity (self , value : str ) -> str :
1799+ if value not in SEVERITIES :
1800+ raise serializers .ValidationError (
1801+ f"Severity must be one of the following: { SEVERITIES } "
1802+ )
1803+ return value
1804+
17981805 def build_relational_field (self , field_name , relation_info ):
17991806 if field_name == "notes" :
18001807 return NoteSerializer , {"many" : True , "read_only" : True }
@@ -1922,6 +1929,13 @@ def validate(self, data):
19221929
19231930 return data
19241931
1932+ def validate_severity (self , value : str ) -> str :
1933+ if value not in SEVERITIES :
1934+ raise serializers .ValidationError (
1935+ f"Severity must be one of the following: { SEVERITIES } "
1936+ )
1937+ return value
1938+
19251939
19261940class VulnerabilityIdTemplateSerializer (serializers .ModelSerializer ):
19271941 class Meta :
@@ -1998,6 +2012,13 @@ class Meta:
19982012 model = Stub_Finding
19992013 fields = "__all__"
20002014
2015+ def validate_severity (self , value : str ) -> str :
2016+ if value not in SEVERITIES :
2017+ raise serializers .ValidationError (
2018+ f"Severity must be one of the following: { SEVERITIES } "
2019+ )
2020+ return value
2021+
20012022
20022023class StubFindingCreateSerializer (serializers .ModelSerializer ):
20032024 test = serializers .PrimaryKeyRelatedField (queryset = Test .objects .all ())
@@ -2009,6 +2030,13 @@ class Meta:
20092030 "reporter" : {"default" : serializers .CurrentUserDefault ()},
20102031 }
20112032
2033+ def validate_severity (self , value : str ) -> str :
2034+ if value not in SEVERITIES :
2035+ raise serializers .ValidationError (
2036+ f"Severity must be one of the following: { SEVERITIES } "
2037+ )
2038+ return value
2039+
20122040
20132041class ProductSerializer (TaggitSerializer , serializers .ModelSerializer ):
20142042 findings_count = serializers .SerializerMethodField ()
0 commit comments