Add exception handling for type error

CentreForDigitalHumanities · Mar 19, 2024 · 135bcaf · 135bcaf
1 parent a4b5eb6
commit 135bcaf
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/humitifier/models.py b/humitifier/models.py
@@ -36,7 +36,19 @@ def from_sql_rows(cls, rows):
             except Exception as e:
                 logger.info(f"Error parsing {name}: {e}")
                 create_args[name] = FactError(**data)
-        return cls(**create_args)
+        try:
+            return cls(**create_args)
+        except TypeError:
+            not_collected_fact_err = FactError(
+                stdout="",
+                stderr="",
+                exception="",
+                exit_code=0,
+                py_excpetion="Fact Not Collected",
+            )
+            err_args_base = {k: not_collected_fact_err for k in cls.__annotations__}
+            err_args = {**err_args_base, **create_args}
+            return cls(**err_args)
 
 
 @dataclass