experimental: Build pydantic from schema

equinor · Jan 10, 2024 · fee1649 · fee1649
1 parent 0027e1d
commit fee1649
Show file tree

Hide file tree

Showing 3 changed files with 515 additions and 0 deletions.
diff --git a/gen.sh b/gen.sh
@@ -0,0 +1,13 @@
+datamodel-codegen \
+  --input $1 \
+  --input-file-type "jsonschema" \
+  --output-model-type "pydantic_v2.BaseModel" \
+  --target-python-version "3.8" \
+  --use-default-kwarg \
+  --use-double-quotes \
+  --use-schema-description \
+  --use-standard-collections \
+  --use-subclass-enum \
+  --use-title-as-name \
+  --use-union-operator \
+  --use-unique-items-as-set
diff --git a/pyproject.toml b/pyproject.toml
@@ -34,6 +34,7 @@ dependencies = [
     "numpy",
     "pandas",
     "pyarrow",
+    "pydantic>=2.0.0",
     "PyYAML",
     "xtgeo>=2.16",
 ]