database:column - allow to ignore field in database

aarhusstadsarkiv · Oct 30, 2023 · aa55639 · aa55639
1 parent 5065d2d
commit aa55639
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/acacore/database/column.py b/acacore/database/column.py
@@ -40,7 +40,10 @@
 }
 
 
-def _schema_to_column(name: str, schema: dict, defs: Optional[dict[str, dict]] = None) -> "Column":
+def _schema_to_column(name: str, schema: dict, defs: Optional[dict[str, dict]] = None) -> Optional["Column"]:
+    if schema.get("ignore"):
+        return None
+
     defs = defs or {}
     if schema.get("$ref"):
         schema.update(defs[schema.get("$ref", "").removeprefix("#/$defs/")])
@@ -86,7 +89,8 @@ def _schema_to_column(name: str, schema: dict, defs: Optional[dict[str, dict]] =
 
 def model_to_columns(model: Type[BaseModel]) -> list["Column"]:
     schema: dict = model.model_json_schema()
-    return [_schema_to_column(p, s, schema.get("$defs")) for p, s in schema["properties"].items()]
+    columns = [_schema_to_column(p, s, schema.get("$defs")) for p, s in schema["properties"].items()]
+    return [c for c in columns if c]
 
 
 class Column(Generic[T]):