fixed 500 when invalid workflow is presented

ch40s · Nov 7, 2019 · 24dd7ae · 24dd7ae
1 parent 996732d
commit 24dd7ae
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/api/server/db/workflow.py b/api/server/db/workflow.py
@@ -2,7 +2,9 @@
 from typing import List
 from uuid import UUID
 
+from fastapi.exceptions import RequestValidationError
 import jsonschema
+import pydantic
 from pydantic import BaseModel
 
 from api.server.db import IDBaseModel
@@ -46,8 +48,11 @@ class WorkflowModel(IDBaseModel):
     start: UUID
 
     def __init__(self, **kwargs):
-        super(WorkflowModel, self).__init__(**kwargs)
-        self.validate_workflow()
+        try:
+            super(WorkflowModel, self).__init__(**kwargs)
+            self.validate_workflow()
+        except pydantic.error_wrappers.ValidationError as e:
+            raise RequestValidationError(e.raw_errors)
 
     def validate_workflow(self):
         """Validates the object"""