dataset.Column obj: change checks attr data type

NYCPlanning · Dec 19, 2024 · f158ce4 · f158ce4
1 parent a041b06
commit f158ce4
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/dcpy/models/dataset.py b/dcpy/models/dataset.py
@@ -1,5 +1,5 @@
 from dcpy.models.base import SortedSerializedBase
-from typing import Literal
+from typing import Literal, Any
 
 COLUMN_TYPES = Literal[
     "text",
@@ -14,10 +14,10 @@
 ]
 
 
-# TODO: extend/modify Checks model
-class Checks(SortedSerializedBase):
-    is_primary_key: bool | None = None
-    non_nullable: bool | None = None
+# TODO: DELETE
+# class Checks(SortedSerializedBase):
+#     is_primary_key: bool | None = None
+#     non_nullable: bool | None = None
 
 
 class Column(SortedSerializedBase, extra="forbid"):
@@ -29,4 +29,4 @@ class Column(SortedSerializedBase, extra="forbid"):
     data_type: COLUMN_TYPES | None = None
     description: str | None = None
     is_required: bool = True
-    checks: Checks | None = None
+    checks: list[str | dict[str, Any]] | None = None
diff --git a/dcpy/models/lifecycle/ingest.py b/dcpy/models/lifecycle/ingest.py
@@ -93,6 +93,7 @@ class Template(BaseModel, extra="forbid"):
     attributes: DatasetAttributes
     ingestion: Ingestion
     columns: list[Column] = []
+    checks: list[str | dict[str, Any]] | None = None
 
     @property
     def has_geom(self):