型の判定

Author: yuji38kwmt

annofabapi/pydantic_models/full_annotation_data_classification.py

@@ -46,7 +46,10 @@ def to_json(self) -> str:
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
         """Create an instance of FullAnnotationDataClassification from a JSON string"""
-        return cls.from_dict(json.loads(json_str))
+        result = cls.from_dict(json.loads(json_str))
+        if result.type != "Classification":
+            raise ValueError("Invalid type")
+        return result
 
     def to_dict(self) -> Dict[str, Any]:
         """Return the dictionary representation of the model using alias.

annofabapi/pydantic_models/full_annotation_data_segmentation.py

@@ -49,7 +49,10 @@ def to_json(self) -> str:
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
         """Create an instance of FullAnnotationDataSegmentation from a JSON string"""
-        return cls.from_dict(json.loads(json_str))
+        result = cls.from_dict(json.loads(json_str))
+        if result.type != "Segmentation":
+            raise ValueError("Invalid type")
+        return result
 
     def to_dict(self) -> Dict[str, Any]:
         """Return the dictionary representation of the model using alias.

annofabapi/pydantic_models/full_annotation_data_segmentation_v2.py

@@ -49,7 +49,10 @@ def to_json(self) -> str:
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
         """Create an instance of FullAnnotationDataSegmentationV2 from a JSON string"""
-        return cls.from_dict(json.loads(json_str))
+        result = cls.from_dict(json.loads(json_str))
+        if result.type != "SegmentationV2":
+            raise ValueError("Invalid type")
+        return result
 
     def to_dict(self) -> Dict[str, Any]:
         """Return the dictionary representation of the model using alias.