Skip to content

Commit fa04b9d

Browse files
committed
update create_spec to build migration/transform/validation components
1 parent 45b47f4 commit fa04b9d

File tree

1 file changed

+17
-5
lines changed

1 file changed

+17
-5
lines changed

airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py

Lines changed: 17 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -3465,16 +3465,28 @@ def _get_job_timeout() -> datetime.timedelta:
34653465
parameters=model.parameters or {},
34663466
)
34673467

3468-
@staticmethod
3469-
def create_spec(model: SpecModel, config: Config, **kwargs: Any) -> Spec:
3468+
def create_spec(self, model: SpecModel, config: Config, **kwargs: Any) -> Spec:
3469+
config_migrations = []
3470+
transformations = []
3471+
validations = []
3472+
3473+
for migration in model.config_normalization_rules.config_migrations:
3474+
config_migrations.append(self._create_component_from_model(migration, config))
3475+
3476+
for transformation in model.config_normalization_rules.transformations:
3477+
transformations.append(self._create_component_from_model(transformation, config))
3478+
3479+
for validation in model.config_normalization_rules.validations:
3480+
validations.append(self._create_component_from_model(validation, config))
3481+
34703482
return Spec(
34713483
connection_specification=model.connection_specification,
34723484
documentation_url=model.documentation_url,
34733485
advanced_auth=model.advanced_auth,
34743486
parameters={},
3475-
config_migrations=model.config_normalization_rules.config_migrations,
3476-
transformations=model.config_normalization_rules.transformations,
3477-
validations=model.config_normalization_rules.validations,
3487+
config_migrations=config_migrations,
3488+
transformations=transformations,
3489+
validations=validations,
34783490
)
34793491

34803492
def create_substream_partition_router(

0 commit comments

Comments
 (0)