Skip to content

Commit 752cf7c

Browse files
gaurpulkitgaurpulkit
committed
remove extra allow
1 parent ba68036 commit 752cf7c

File tree

6 files changed

+9
-128
lines changed

6 files changed

+9
-128
lines changed

‎src/datapilot/core/insights/schema.py‎

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22
from typing import Dict
33

44
from pydantic import BaseModel
5-
from pydantic import ConfigDict
65

76

87
class Severity(Enum):
@@ -12,8 +11,6 @@ class Severity(Enum):
1211

1312

1413
class InsightResult(BaseModel):
15-
model_config = ConfigDict(extra="allow")
16-
1714
name: str
1815
type: str
1916
message: str
@@ -23,7 +20,5 @@ class InsightResult(BaseModel):
2320

2421

2522
class InsightResponse(BaseModel):
26-
model_config = ConfigDict(extra="allow")
27-
2823
insight: InsightResult
2924
severity: Severity = Severity.ERROR

‎src/datapilot/core/platforms/dbt/schemas/catalog.py‎

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,6 @@
1313

1414

1515
class AltimateCatalogMetadata(BaseModel):
16-
model_config = ConfigDict(extra="allow")
17-
1816
dbt_schema_version: Optional[str] = "https://schemas.getdbt.com/dbt/catalog/v1.json"
1917
dbt_version: Optional[str] = "0.19.0"
2018
generated_at: Optional[datetime] = "2021-02-10T04:42:33.680487Z"
@@ -23,8 +21,6 @@ class AltimateCatalogMetadata(BaseModel):
2321

2422

2523
class AltimateCatalogTableMetadata(BaseModel):
26-
model_config = ConfigDict(extra="allow")
27-
2824
type: str
2925
database: Optional[Optional[str]] = None
3026
schema_name: str
@@ -34,17 +30,13 @@ class AltimateCatalogTableMetadata(BaseModel):
3430

3531

3632
class AltimateCatalogColumnMetadata(BaseModel):
37-
model_config = ConfigDict(extra="allow")
38-
3933
type: str
4034
comment: Optional[Optional[str]] = None
4135
index: int
4236
name: str
4337

4438

4539
class AltimateCatalogStatsItem(BaseModel):
46-
model_config = ConfigDict(extra="allow")
47-
4840
id: str
4941
label: str
5042
value: Optional[Optional[Union[bool, str, float]]] = None
@@ -53,17 +45,13 @@ class AltimateCatalogStatsItem(BaseModel):
5345

5446

5547
class AltimateCatalogTable(BaseModel):
56-
model_config = ConfigDict(extra="allow")
57-
5848
metadata: AltimateCatalogTableMetadata
5949
columns: Dict[str, AltimateCatalogColumnMetadata]
6050
stats: Dict[str, AltimateCatalogStatsItem]
6151
unique_id: Optional[Optional[str]] = None
6252

6353

6454
class AltimateCatalogCatalogV1(BaseModel):
65-
model_config = ConfigDict(extra="allow")
66-
6755
metadata: AltimateCatalogMetadata
6856
nodes: Dict[str, AltimateCatalogTable]
6957
sources: Dict[str, AltimateCatalogTable]

‎src/datapilot/core/platforms/dbt/schemas/manifest.py‎

Lines changed: 0 additions & 55 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
from typing import Union
77

88
from pydantic import BaseModel
9-
from pydantic import ConfigDict
109

1110
from vendor.dbt_artifacts_parser.parsers.manifest.manifest_v1 import ManifestV1
1211
from vendor.dbt_artifacts_parser.parsers.manifest.manifest_v2 import ManifestV2
@@ -24,8 +23,6 @@
2423

2524

2625
class DBTVersion(BaseModel):
27-
model_config = ConfigDict(extra="allow")
28-
2926
MAJOR: int
3027
MINOR: int
3128
PATCH: Optional[int] = None
@@ -48,22 +45,16 @@ class DBTVersion(BaseModel):
4845

4946

5047
class AltimateDocs(BaseModel):
51-
model_config = ConfigDict(extra="allow")
52-
5348
show: Optional[bool] = True
5449
node_color: Optional[Optional[str]] = None
5550

5651

5752
class AltimateDependsOn(BaseModel):
58-
model_config = ConfigDict(extra="allow")
59-
6053
nodes: Optional[List[str]] = None
6154
macros: Optional[List[str]] = None
6255

6356

6457
class AltimateManifestColumnInfo(BaseModel):
65-
model_config = ConfigDict(extra="allow")
66-
6758
name: str
6859
description: Optional[str] = ""
6960
meta: Optional[Dict[str, Any]] = {}
@@ -73,8 +64,6 @@ class AltimateManifestColumnInfo(BaseModel):
7364

7465

7566
class AltimateFileHash(BaseModel):
76-
model_config = ConfigDict(extra="allow")
77-
7867
name: Optional[str] = None
7968
checksum: Optional[str] = None
8069

@@ -105,25 +94,19 @@ class AltimateAccess(Enum):
10594

10695

10796
class AltimateDBTContract(BaseModel):
108-
model_config = ConfigDict(extra="allow")
109-
11097
enforced: Optional[bool] = False
11198
alias_types: Optional[bool] = True
11299
checksum: Optional[Optional[str]] = None
113100

114101

115102
class AltimateHook(BaseModel):
116-
model_config = ConfigDict(extra="allow")
117-
118103
sql: str
119104
transaction: Optional[bool] = True
120105
index: Optional[Optional[int]] = None
121106

122107

123108
# TODO: Need to add the rest of the fields
124109
class AltimateNodeConfig(BaseModel):
125-
model_config = ConfigDict(extra="allow")
126-
127110
_extra: Optional[Dict[str, Any]] = None
128111
enabled: Optional[bool] = True
129112
alias: Optional[Optional[str]] = None
@@ -145,8 +128,6 @@ class AltimateNodeConfig(BaseModel):
145128

146129

147130
class AltimateManifestNode(BaseModel):
148-
model_config = ConfigDict(extra="allow")
149-
150131
database: Optional[str] = None
151132
resource_type: AltimateResourceType
152133
schema_name: str
@@ -177,34 +158,26 @@ class AltimateManifestNode(BaseModel):
177158

178159

179160
class AltimateQuoting(BaseModel):
180-
model_config = ConfigDict(extra="allow")
181-
182161
database: Optional[Optional[bool]] = None
183162
schema_: Optional[Optional[bool]] = None
184163
identifier: Optional[Optional[bool]] = None
185164
column: Optional[Optional[bool]] = None
186165

187166

188167
class AltimateFreshnessThreshold(BaseModel):
189-
model_config = ConfigDict(extra="allow")
190-
191168
warn_after: Optional[Dict] = None
192169
error_after: Optional[Dict] = None
193170
filter: Optional[str] = None
194171

195172

196173
class AltimateExternalPartition(BaseModel):
197-
model_config = ConfigDict(extra="allow")
198-
199174
name: Optional[str] = ""
200175
description: Optional[str] = ""
201176
data_type: Optional[str] = ""
202177
meta: Optional[Dict[str, Any]] = {}
203178

204179

205180
class AltimateExternalTable(BaseModel):
206-
model_config = ConfigDict(extra="allow")
207-
208181
location: Optional[Optional[str]] = None
209182
file_format: Optional[Optional[str]] = None
210183
row_format: Optional[Optional[str]] = None
@@ -213,23 +186,17 @@ class AltimateExternalTable(BaseModel):
213186

214187

215188
class AltimateSourceConfig(BaseModel):
216-
model_config = ConfigDict(extra="allow")
217-
218189
enabled: Optional[bool] = True
219190

220191

221192
class AltimateDeferRelation(BaseModel):
222-
model_config = ConfigDict(extra="allow")
223-
224193
database: Optional[str] = None
225194
schema_name: str
226195
alias: str
227196
relation_name: Optional[str] = None
228197

229198

230199
class AltimateSeedConfig(BaseModel):
231-
model_config = ConfigDict(extra="allow")
232-
233200
_extra: Optional[Dict[str, Any]] = None
234201
enabled: Optional[bool] = True
235202
alias: Optional[Optional[str]] = None
@@ -258,8 +225,6 @@ class AltimateSeedConfig(BaseModel):
258225

259226

260227
class AltimateSeedNode(BaseModel):
261-
model_config = ConfigDict(extra="allow")
262-
263228
database: Optional[str] = None
264229
schema_name: str
265230
name: str
@@ -292,8 +257,6 @@ class AltimateSeedNode(BaseModel):
292257

293258

294259
class AltimateManifestSourceNode(BaseModel):
295-
model_config = ConfigDict(extra="allow")
296-
297260
database: Optional[str] = None
298261
resource_type: AltimateResourceType
299262
schema_name: str
@@ -332,8 +295,6 @@ class AltimateExposureType(Enum):
332295

333296

334297
class AltimateOwner(BaseModel):
335-
model_config = ConfigDict(extra="allow")
336-
337298
_extra: Optional[Dict[str, Any]] = None
338299
email: Optional[Optional[str]] = None
339300
name: Optional[Optional[str]] = None
@@ -346,23 +307,17 @@ class AltimateMaturityEnum(Enum):
346307

347308

348309
class AltimateRefArgs(BaseModel):
349-
model_config = ConfigDict(extra="allow")
350-
351310
name: str
352311
package: Optional[Optional[str]] = None
353312
version: Optional[Optional[Union[str, float]]] = None
354313

355314

356315
class AltimateExposureConfig(BaseModel):
357-
model_config = ConfigDict(extra="allow")
358-
359316
_extra: Optional[Dict[str, Any]] = None
360317
enabled: Optional[bool] = True
361318

362319

363320
class AltimateManifestExposureNode(BaseModel):
364-
model_config = ConfigDict(extra="allow")
365-
366321
name: str
367322
resource_type: AltimateResourceType
368323
package_name: str
@@ -388,16 +343,12 @@ class AltimateManifestExposureNode(BaseModel):
388343

389344

390345
class AltimateTestMetadata(BaseModel):
391-
model_config = ConfigDict(extra="allow")
392-
393346
name: str
394347
kwargs: Optional[Dict[str, Any]] = None
395348
namespace: Optional[Optional[str]] = None
396349

397350

398351
class AltimateTestConfig(BaseModel):
399-
model_config = ConfigDict(extra="allow")
400-
401352
_extra: Optional[Dict[str, Any]] = None
402353
enabled: Optional[bool] = True
403354
alias: Optional[Optional[str]] = None
@@ -418,8 +369,6 @@ class AltimateTestConfig(BaseModel):
418369

419370

420371
class AltimateManifestTestNode(BaseModel):
421-
model_config = ConfigDict(extra="allow")
422-
423372
test_metadata: Optional[AltimateTestMetadata] = None
424373
test_type: Optional[str] = None
425374
name: str
@@ -450,8 +399,6 @@ class AltimateManifestTestNode(BaseModel):
450399

451400

452401
class AltimateMacroArgument(BaseModel):
453-
model_config = ConfigDict(extra="allow")
454-
455402
name: str
456403
type: Optional[Optional[str]] = None
457404
description: Optional[Optional[str]] = ""
@@ -461,8 +408,6 @@ class AltimateMacroArgument(BaseModel):
461408

462409

463410
class AltimateManifestMacroNode(BaseModel):
464-
model_config = ConfigDict(extra="allow")
465-
466411
name: str
467412
resource_type: AltimateResourceType
468413
package_name: str

‎src/datapilot/core/platforms/dbt/schemas/run_results.py‎

Lines changed: 6 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -1,38 +1,11 @@
11
from typing import Union
22

3-
from pydantic import ConfigDict
4-
5-
from vendor.dbt_artifacts_parser.parsers.run_results.run_results_v1 import RunResultsV1 as BaseRunResultsV1
6-
from vendor.dbt_artifacts_parser.parsers.run_results.run_results_v2 import RunResultsV2 as BaseRunResultsV2
7-
from vendor.dbt_artifacts_parser.parsers.run_results.run_results_v3 import RunResultsV3 as BaseRunResultsV3
8-
from vendor.dbt_artifacts_parser.parsers.run_results.run_results_v4 import RunResultsV4 as BaseRunResultsV4
9-
from vendor.dbt_artifacts_parser.parsers.run_results.run_results_v5 import RunResultsV5 as BaseRunResultsV5
10-
from vendor.dbt_artifacts_parser.parsers.run_results.run_results_v6 import RunResultsV6 as BaseRunResultsV6
11-
12-
13-
class RunResultsV1(BaseRunResultsV1):
14-
model_config = ConfigDict(extra="allow")
15-
16-
17-
class RunResultsV2(BaseRunResultsV2):
18-
model_config = ConfigDict(extra="allow")
19-
20-
21-
class RunResultsV3(BaseRunResultsV3):
22-
model_config = ConfigDict(extra="allow")
23-
24-
25-
class RunResultsV4(BaseRunResultsV4):
26-
model_config = ConfigDict(extra="allow")
27-
28-
29-
class RunResultsV5(BaseRunResultsV5):
30-
model_config = ConfigDict(extra="allow")
31-
32-
33-
class RunResultsV6(BaseRunResultsV6):
34-
model_config = ConfigDict(extra="allow")
35-
3+
from vendor.dbt_artifacts_parser.parsers.run_results.run_results_v1 import RunResultsV1
4+
from vendor.dbt_artifacts_parser.parsers.run_results.run_results_v2 import RunResultsV2
5+
from vendor.dbt_artifacts_parser.parsers.run_results.run_results_v3 import RunResultsV3
6+
from vendor.dbt_artifacts_parser.parsers.run_results.run_results_v4 import RunResultsV4
7+
from vendor.dbt_artifacts_parser.parsers.run_results.run_results_v5 import RunResultsV5
8+
from vendor.dbt_artifacts_parser.parsers.run_results.run_results_v6 import RunResultsV6
369

3710
RunResults = Union[
3811
RunResultsV6,

‎src/datapilot/core/platforms/dbt/schemas/sources.py‎

Lines changed: 3 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -1,23 +1,8 @@
11
from typing import Union
22

3-
from pydantic import ConfigDict
4-
5-
from vendor.dbt_artifacts_parser.parsers.sources.sources_v1 import SourcesV1 as BaseSourcesV1
6-
from vendor.dbt_artifacts_parser.parsers.sources.sources_v2 import SourcesV2 as BaseSourcesV2
7-
from vendor.dbt_artifacts_parser.parsers.sources.sources_v3 import SourcesV3 as BaseSourcesV3
8-
9-
10-
class SourcesV1(BaseSourcesV1):
11-
model_config = ConfigDict(extra="allow")
12-
13-
14-
class SourcesV2(BaseSourcesV2):
15-
model_config = ConfigDict(extra="allow")
16-
17-
18-
class SourcesV3(BaseSourcesV3):
19-
model_config = ConfigDict(extra="allow")
20-
3+
from vendor.dbt_artifacts_parser.parsers.sources.sources_v1 import SourcesV1
4+
from vendor.dbt_artifacts_parser.parsers.sources.sources_v2 import SourcesV2
5+
from vendor.dbt_artifacts_parser.parsers.sources.sources_v3 import SourcesV3
216

227
Sources = Union[
238
SourcesV3,

0 commit comments

Comments
 (0)