dbt-artifacts-parser/dbt_artifacts_parser/parsers/catalog/catalog_v1.py at main · yu-iskw/dbt-artifacts-parser · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
# generated by datamodel-codegen:
#   filename:  catalog_v1.json

from __future__ import annotations

from typing import Any

from pydantic import ConfigDict, Field

from dbt_artifacts_parser.parsers.base import BaseParserModel


class Metadata(BaseParserModel):
    model_config = ConfigDict(
        extra="forbid",
    )
    dbt_schema_version: str | None = None
    dbt_version: str | None = "1.12.0a1"
    generated_at: str | None = None
    invocation_id: str | None = None
    invocation_started_at: str | None = None
    env: dict[str, str] | None = None


class Metadata1(BaseParserModel):
    model_config = ConfigDict(
        extra="forbid",
    )
    type: str
    schema_: str = Field(..., alias="schema")
    name: str
    database: str | None = None
    comment: str | None = None
    owner: str | None = None


class Columns(BaseParserModel):
    model_config = ConfigDict(
        extra="forbid",
    )
    type: str
    index: int
    name: str
    comment: str | None = None


class Stats(BaseParserModel):
    model_config = ConfigDict(
        extra="forbid",
    )
    id: str
    label: str
    value: bool | str | float | None
    include: bool
    description: str | None = None


class Nodes(BaseParserModel):
    model_config = ConfigDict(
        extra="forbid",
    )
    metadata: Metadata1 = Field(..., title="TableMetadata")
    columns: dict[str, Columns]
    stats: dict[str, Stats]
    unique_id: str | None = None


class Sources(BaseParserModel):
    model_config = ConfigDict(
        extra="forbid",
    )
    metadata: Metadata1 = Field(..., title="TableMetadata")
    columns: dict[str, Columns]
    stats: dict[str, Stats]
    unique_id: str | None = None


class CatalogV1(BaseParserModel):
    model_config = ConfigDict(
        extra="forbid",
    )
    metadata: Metadata = Field(..., title="CatalogMetadata")
    nodes: dict[str, Nodes]
    sources: dict[str, Sources]
    errors: list[str] | None = None
    field_compile_results: Any = Field(None, alias="_compile_results")