Support manifest v9 with dbt-core 1.5rc1 (#38)

yu-iskw · Apr 19, 2023 · 656b2da · 656b2da
1 parent 2de9eba
commit 656b2da
Show file tree

Hide file tree

Showing 7 changed files with 6,310 additions and 11 deletions.
diff --git a/dbt_artifacts_parser/parser.py b/dbt_artifacts_parser/parser.py
@@ -16,8 +16,6 @@
 #
 from typing import Union
 
-from dbt_artifacts_parser.parsers.utils import get_dbt_schema_version
-
 from dbt_artifacts_parser.parsers.catalog.catalog_v1 import CatalogV1
 from dbt_artifacts_parser.parsers.manifest.manifest_v1 import ManifestV1
 from dbt_artifacts_parser.parsers.manifest.manifest_v2 import ManifestV2
@@ -27,14 +25,15 @@
 from dbt_artifacts_parser.parsers.manifest.manifest_v6 import ManifestV6
 from dbt_artifacts_parser.parsers.manifest.manifest_v7 import ManifestV7
 from dbt_artifacts_parser.parsers.manifest.manifest_v8 import ManifestV8
+from dbt_artifacts_parser.parsers.manifest.manifest_v9 import ManifestV9
 from dbt_artifacts_parser.parsers.run_results.run_results_v1 import RunResultsV1
 from dbt_artifacts_parser.parsers.run_results.run_results_v2 import RunResultsV2
 from dbt_artifacts_parser.parsers.run_results.run_results_v3 import RunResultsV3
 from dbt_artifacts_parser.parsers.run_results.run_results_v4 import RunResultsV4
 from dbt_artifacts_parser.parsers.sources.sources_v1 import SourcesV1
 from dbt_artifacts_parser.parsers.sources.sources_v2 import SourcesV2
 from dbt_artifacts_parser.parsers.sources.sources_v3 import SourcesV3
-
+from dbt_artifacts_parser.parsers.utils import get_dbt_schema_version
 from dbt_artifacts_parser.parsers.version_map import ArtifactTypes
 
 
@@ -96,6 +95,8 @@ def parse_manifest(
         return ManifestV7(**manifest)
     elif dbt_schema_version == ArtifactTypes.MANIFEST_V8.value.dbt_schema_version:
         return ManifestV8(**manifest)
+    elif dbt_schema_version == ArtifactTypes.MANIFEST_V9.value.dbt_schema_version:
+        return ManifestV9(**manifest)
     raise ValueError("Not a soft of manifest.json")
 
 
@@ -163,6 +164,14 @@ def parse_manifest_v8(manifest: dict) -> ManifestV6:
     raise ValueError("Not a manifest.json v8")
 
 
+def parse_manifest_v9(manifest: dict) -> ManifestV6:
+    """Parse manifest.json ver.9"""
+    dbt_schema_version = get_dbt_schema_version(artifact_json=manifest)
+    if dbt_schema_version == ArtifactTypes.MANIFEST_V9.value.dbt_schema_version:
+        return ManifestV9(**manifest)
+    raise ValueError("Not a manifest.json v9")
+
+
 #
 # run-results
 #