From 2622b6dd0f70e06e9d96fe53524774a2cd691ed9 Mon Sep 17 00:00:00 2001 From: Pankaj Date: Sat, 17 Aug 2024 18:40:43 +0530 Subject: [PATCH] Add init --- cosmos/operators/aws_eks.py | 24 ++++++++++++++++++++ cosmos/operators/azure_container_instance.py | 24 ++++++++++++++++++-- cosmos/operators/docker.py | 21 +++++++++++++++++ cosmos/operators/kubernetes.py | 21 +++++++++++++++++ cosmos/operators/local.py | 21 +++++++++++++++++ 5 files changed, 109 insertions(+), 2 deletions(-) diff --git a/cosmos/operators/aws_eks.py b/cosmos/operators/aws_eks.py index 0d98b2c63..ad268e310 100644 --- a/cosmos/operators/aws_eks.py +++ b/cosmos/operators/aws_eks.py @@ -79,12 +79,18 @@ class DbtBuildAwsEksOperator(DbtAwsEksBaseOperator, DbtBuildKubernetesOperator): DbtAwsEksBaseOperator.template_fields + DbtBuildKubernetesOperator.template_fields # type: ignore[operator] ) + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtLSAwsEksOperator(DbtAwsEksBaseOperator, DbtLSKubernetesOperator): """ Executes a dbt core ls command. """ + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtSeedAwsEksOperator(DbtAwsEksBaseOperator, DbtSeedKubernetesOperator): """ @@ -95,18 +101,27 @@ class DbtSeedAwsEksOperator(DbtAwsEksBaseOperator, DbtSeedKubernetesOperator): DbtAwsEksBaseOperator.template_fields + DbtSeedKubernetesOperator.template_fields # type: ignore[operator] ) + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtSnapshotAwsEksOperator(DbtAwsEksBaseOperator, DbtSnapshotKubernetesOperator): """ Executes a dbt core snapshot command. """ + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtSourceAzureContainerInstanceOperator(DbtAwsEksBaseOperator, DbtSourceKubernetesOperator): """ Executes a dbt source freshness command. """ + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtRunAwsEksOperator(DbtAwsEksBaseOperator, DbtRunKubernetesOperator): """ @@ -117,6 +132,9 @@ class DbtRunAwsEksOperator(DbtAwsEksBaseOperator, DbtRunKubernetesOperator): DbtAwsEksBaseOperator.template_fields + DbtRunKubernetesOperator.template_fields # type: ignore[operator] ) + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtTestAwsEksOperator(DbtAwsEksBaseOperator, DbtTestKubernetesOperator): """ @@ -127,6 +145,9 @@ class DbtTestAwsEksOperator(DbtAwsEksBaseOperator, DbtTestKubernetesOperator): DbtAwsEksBaseOperator.template_fields + DbtTestKubernetesOperator.template_fields # type: ignore[operator] ) + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtRunOperationAwsEksOperator(DbtAwsEksBaseOperator, DbtRunOperationKubernetesOperator): """ @@ -136,3 +157,6 @@ class DbtRunOperationAwsEksOperator(DbtAwsEksBaseOperator, DbtRunOperationKubern template_fields: Sequence[str] = ( DbtAwsEksBaseOperator.template_fields + DbtRunOperationKubernetesOperator.template_fields # type: ignore[operator] ) + + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) diff --git a/cosmos/operators/azure_container_instance.py b/cosmos/operators/azure_container_instance.py index 3b6ae3b7d..ddd0bce97 100644 --- a/cosmos/operators/azure_container_instance.py +++ b/cosmos/operators/azure_container_instance.py @@ -82,6 +82,9 @@ class DbtLSAzureContainerInstanceOperator(DbtLSMixin, DbtAzureContainerInstanceB Executes a dbt core ls command. """ + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtSeedAzureContainerInstanceOperator(DbtSeedMixin, DbtAzureContainerInstanceBaseOperator): # type: ignore """ @@ -92,6 +95,9 @@ class DbtSeedAzureContainerInstanceOperator(DbtSeedMixin, DbtAzureContainerInsta template_fields: Sequence[str] = DbtAzureContainerInstanceBaseOperator.template_fields + DbtRunMixin.template_fields # type: ignore[operator] + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtSnapshotAzureContainerInstanceOperator(DbtSnapshotMixin, DbtAzureContainerInstanceBaseOperator): # type: ignore """ @@ -99,12 +105,18 @@ class DbtSnapshotAzureContainerInstanceOperator(DbtSnapshotMixin, DbtAzureContai """ + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtSourceAzureContainerInstanceOperator(DbtSourceMixin, DbtAzureContainerInstanceBaseOperator): """ Executes a dbt source freshness command. """ + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtRunAzureContainerInstanceOperator(DbtRunMixin, DbtAzureContainerInstanceBaseOperator): # type: ignore """ @@ -113,6 +125,9 @@ class DbtRunAzureContainerInstanceOperator(DbtRunMixin, DbtAzureContainerInstanc template_fields: Sequence[str] = DbtAzureContainerInstanceBaseOperator.template_fields + DbtRunMixin.template_fields # type: ignore[operator] + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtTestAzureContainerInstanceOperator(DbtTestMixin, DbtAzureContainerInstanceBaseOperator): # type: ignore """ @@ -125,7 +140,7 @@ def __init__(self, on_warning_callback: Callable[..., Any] | None = None, **kwar self.on_warning_callback = on_warning_callback -class DbtRunOperationAzureContainerInstanceOperator(DbtRunOperationMixin, DbtAzureContainerInstanceBaseOperator): # type: ignore +class DbtRunOperationAzureContainerInstanceOperator(DbtRunOperationMixin, DbtAzureContainerInstanceBaseOperator): """ Executes a dbt core run-operation command. @@ -134,4 +149,9 @@ class DbtRunOperationAzureContainerInstanceOperator(DbtRunOperationMixin, DbtAzu selected macro. """ - template_fields: Sequence[str] = DbtAzureContainerInstanceBaseOperator.template_fields + DbtRunOperationMixin.template_fields # type: ignore[operator] + template_fields: Sequence[str] = ( + DbtAzureContainerInstanceBaseOperator.template_fields + DbtRunOperationMixin.template_fields # type: ignore[operator] + ) + + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) diff --git a/cosmos/operators/docker.py b/cosmos/operators/docker.py index 91d13fe72..ace9a6739 100644 --- a/cosmos/operators/docker.py +++ b/cosmos/operators/docker.py @@ -69,12 +69,18 @@ class DbtBuildDockerOperator(DbtBuildMixin, DbtDockerBaseOperator): template_fields: Sequence[str] = DbtDockerBaseOperator.template_fields + DbtBuildMixin.template_fields # type: ignore[operator] + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtLSDockerOperator(DbtLSMixin, DbtDockerBaseOperator): """ Executes a dbt core ls command. """ + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtSeedDockerOperator(DbtSeedMixin, DbtDockerBaseOperator): """ @@ -85,18 +91,27 @@ class DbtSeedDockerOperator(DbtSeedMixin, DbtDockerBaseOperator): template_fields: Sequence[str] = DbtDockerBaseOperator.template_fields + DbtSeedMixin.template_fields # type: ignore[operator] + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtSnapshotDockerOperator(DbtSnapshotMixin, DbtDockerBaseOperator): """ Executes a dbt core snapshot command. """ + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtSourceDockerOperator(DbtSourceMixin, DbtDockerBaseOperator): """ Executes a dbt source freshness command. """ + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtRunDockerOperator(DbtRunMixin, DbtDockerBaseOperator): """ @@ -105,6 +120,9 @@ class DbtRunDockerOperator(DbtRunMixin, DbtDockerBaseOperator): template_fields: Sequence[str] = DbtDockerBaseOperator.template_fields + DbtRunMixin.template_fields # type: ignore[operator] + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtTestDockerOperator(DbtTestMixin, DbtDockerBaseOperator): """ @@ -127,3 +145,6 @@ class DbtRunOperationDockerOperator(DbtRunOperationMixin, DbtDockerBaseOperator) """ template_fields: Sequence[str] = DbtDockerBaseOperator.template_fields + DbtRunOperationMixin.template_fields # type: ignore[operator] + + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) diff --git a/cosmos/operators/kubernetes.py b/cosmos/operators/kubernetes.py index 29a90a3ff..6186242b0 100644 --- a/cosmos/operators/kubernetes.py +++ b/cosmos/operators/kubernetes.py @@ -102,12 +102,18 @@ class DbtBuildKubernetesOperator(DbtBuildMixin, DbtKubernetesBaseOperator): template_fields: Sequence[str] = DbtKubernetesBaseOperator.template_fields + DbtBuildMixin.template_fields # type: ignore[operator] + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtLSKubernetesOperator(DbtLSMixin, DbtKubernetesBaseOperator): """ Executes a dbt core ls command. """ + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtSeedKubernetesOperator(DbtSeedMixin, DbtKubernetesBaseOperator): """ @@ -116,18 +122,27 @@ class DbtSeedKubernetesOperator(DbtSeedMixin, DbtKubernetesBaseOperator): template_fields: Sequence[str] = DbtKubernetesBaseOperator.template_fields + DbtSeedMixin.template_fields # type: ignore[operator] + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtSnapshotKubernetesOperator(DbtSnapshotMixin, DbtKubernetesBaseOperator): """ Executes a dbt core snapshot command. """ + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtSourceKubernetesOperator(DbtSourceMixin, DbtKubernetesBaseOperator): """ Executes a dbt source freshness command. """ + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtRunKubernetesOperator(DbtRunMixin, DbtKubernetesBaseOperator): """ @@ -136,6 +151,9 @@ class DbtRunKubernetesOperator(DbtRunMixin, DbtKubernetesBaseOperator): template_fields: Sequence[str] = DbtKubernetesBaseOperator.template_fields + DbtRunMixin.template_fields # type: ignore[operator] + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtTestKubernetesOperator(DbtTestMixin, DbtKubernetesBaseOperator): """ @@ -239,3 +257,6 @@ class DbtRunOperationKubernetesOperator(DbtRunOperationMixin, DbtKubernetesBaseO """ template_fields: Sequence[str] = DbtKubernetesBaseOperator.template_fields + DbtRunOperationMixin.template_fields # type: ignore[operator] + + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) diff --git a/cosmos/operators/local.py b/cosmos/operators/local.py index 1a88edb62..cc7b6e68f 100644 --- a/cosmos/operators/local.py +++ b/cosmos/operators/local.py @@ -545,12 +545,18 @@ class DbtBuildLocalOperator(DbtBuildMixin, DbtLocalBaseOperator): template_fields: Sequence[str] = DbtLocalBaseOperator.template_fields + DbtBuildMixin.template_fields # type: ignore[operator] + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtLSLocalOperator(DbtLSMixin, DbtLocalBaseOperator): """ Executes a dbt core ls command. """ + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtSeedLocalOperator(DbtSeedMixin, DbtLocalBaseOperator): """ @@ -559,18 +565,27 @@ class DbtSeedLocalOperator(DbtSeedMixin, DbtLocalBaseOperator): template_fields: Sequence[str] = DbtLocalBaseOperator.template_fields + DbtSeedMixin.template_fields # type: ignore[operator] + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtSnapshotLocalOperator(DbtSnapshotMixin, DbtLocalBaseOperator): """ Executes a dbt core snapshot command. """ + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtSourceLocalOperator(DbtSourceMixin, DbtLocalBaseOperator): """ Executes a dbt source freshness command. """ + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtRunLocalOperator(DbtRunMixin, DbtLocalBaseOperator): """ @@ -579,6 +594,9 @@ class DbtRunLocalOperator(DbtRunMixin, DbtLocalBaseOperator): template_fields: Sequence[str] = DbtLocalBaseOperator.template_fields + DbtRunMixin.template_fields # type: ignore[operator] + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtTestLocalOperator(DbtTestMixin, DbtLocalBaseOperator): """ @@ -641,6 +659,9 @@ class DbtRunOperationLocalOperator(DbtRunOperationMixin, DbtLocalBaseOperator): template_fields: Sequence[str] = DbtLocalBaseOperator.template_fields + DbtRunOperationMixin.template_fields # type: ignore[operator] + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + class DbtDocsLocalOperator(DbtLocalBaseOperator): """