apache-airflow-providers-teradata 2.1.1rc1__tar.gz → 2.2.0rc1__tar.gz

{apache_airflow_providers_teradata-2.1.1rc1 → apache_airflow_providers_teradata-2.2.0rc1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: apache-airflow-providers-teradata
-Version: 2.1.1rc1
+Version: 2.2.0rc1
 Summary: Provider package apache-airflow-providers-teradata for Apache Airflow
 Keywords: airflow-provider,teradata,airflow,integration
 Author-email: Apache Software Foundation <dev@airflow.apache.org>
@@ -25,15 +25,19 @@ Requires-Dist: apache-airflow-providers-common-sql>=1.3.1rc0
 Requires-Dist: apache-airflow>=2.7.0rc0
 Requires-Dist: teradatasql>=17.20.0.28
 Requires-Dist: teradatasqlalchemy>=17.20.0.0
+Requires-Dist: apache-airflow-providers-amazon ; extra == "amazon"
 Requires-Dist: apache-airflow-providers-common-sql ; extra == "common.sql"
+Requires-Dist: apache-airflow-providers-microsoft-azure ; extra == "microsoft.azure"
 Project-URL: Bug Tracker, https://github.com/apache/airflow/issues
-Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-teradata/2.1.1/changelog.html
-Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-teradata/2.1.1
+Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-teradata/2.2.0/changelog.html
+Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-teradata/2.2.0
 Project-URL: Slack Chat, https://s.apache.org/airflow-slack
 Project-URL: Source Code, https://github.com/apache/airflow
 Project-URL: Twitter, https://twitter.com/ApacheAirflow
 Project-URL: YouTube, https://www.youtube.com/channel/UCSXwxpWZQ7XZ1WL3wqevChA/
+Provides-Extra: amazon
 Provides-Extra: common.sql
+Provides-Extra: microsoft.azure
 
 
 .. Licensed to the Apache Software Foundation (ASF) under one
@@ -79,7 +83,7 @@ Provides-Extra: common.sql
 
 Package ``apache-airflow-providers-teradata``
 
-Release: ``2.1.1.rc1``
+Release: ``2.2.0.rc1``
 
 
 `Teradata <https://www.teradata.com/>`__
@@ -92,7 +96,7 @@ This is a provider package for ``teradata`` provider. All classes for this provi
 are in ``airflow.providers.teradata`` python package.
 
 You can find package information and changelog for the provider
-in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-teradata/2.1.1/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-teradata/2.2.0/>`_.
 
 Installation
 ------------
@@ -125,14 +129,16 @@ You can install such cross-provider dependencies when installing from PyPI. For
 
 .. code-block:: bash
 
-    pip install apache-airflow-providers-teradata[common.sql]
+    pip install apache-airflow-providers-teradata[amazon]
 
 
-============================================================================================================  ==============
-Dependent package                                                                                             Extra
-============================================================================================================  ==============
-`apache-airflow-providers-common-sql <https://airflow.apache.org/docs/apache-airflow-providers-common-sql>`_  ``common.sql``
-============================================================================================================  ==============
+======================================================================================================================  ===================
+Dependent package                                                                                                       Extra
+======================================================================================================================  ===================
+`apache-airflow-providers-amazon <https://airflow.apache.org/docs/apache-airflow-providers-amazon>`_                    ``amazon``
+`apache-airflow-providers-common-sql <https://airflow.apache.org/docs/apache-airflow-providers-common-sql>`_            ``common.sql``
+`apache-airflow-providers-microsoft-azure <https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure>`_  ``microsoft.azure``
+======================================================================================================================  ===================
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-teradata/2.1.1/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-teradata/2.2.0/changelog.html>`_.

{apache_airflow_providers_teradata-2.1.1rc1 → apache_airflow_providers_teradata-2.2.0rc1}/README.rst

@@ -42,7 +42,7 @@
 
 Package ``apache-airflow-providers-teradata``
 
-Release: ``2.1.1.rc1``
+Release: ``2.2.0.rc1``
 
 
 `Teradata <https://www.teradata.com/>`__
@@ -55,7 +55,7 @@ This is a provider package for ``teradata`` provider. All classes for this provi
 are in ``airflow.providers.teradata`` python package.
 
 You can find package information and changelog for the provider
-in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-teradata/2.1.1/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-teradata/2.2.0/>`_.
 
 Installation
 ------------
@@ -88,14 +88,16 @@ You can install such cross-provider dependencies when installing from PyPI. For
 
 .. code-block:: bash
 
-    pip install apache-airflow-providers-teradata[common.sql]
+    pip install apache-airflow-providers-teradata[amazon]
 
 
-============================================================================================================  ==============
-Dependent package                                                                                             Extra
-============================================================================================================  ==============
-`apache-airflow-providers-common-sql <https://airflow.apache.org/docs/apache-airflow-providers-common-sql>`_  ``common.sql``
-============================================================================================================  ==============
+======================================================================================================================  ===================
+Dependent package                                                                                                       Extra
+======================================================================================================================  ===================
+`apache-airflow-providers-amazon <https://airflow.apache.org/docs/apache-airflow-providers-amazon>`_                    ``amazon``
+`apache-airflow-providers-common-sql <https://airflow.apache.org/docs/apache-airflow-providers-common-sql>`_            ``common.sql``
+`apache-airflow-providers-microsoft-azure <https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure>`_  ``microsoft.azure``
+======================================================================================================================  ===================
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-teradata/2.1.1/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-teradata/2.2.0/changelog.html>`_.

{apache_airflow_providers_teradata-2.1.1rc1 → apache_airflow_providers_teradata-2.2.0rc1}/airflow/providers/teradata/__init__.py

@@ -29,7 +29,7 @@ from airflow import __version__ as airflow_version
 
 __all__ = ["__version__"]
 
-__version__ = "2.1.1"
+__version__ = "2.2.0"
 
 if packaging.version.parse(packaging.version.parse(airflow_version).base_version) < packaging.version.parse(
     "2.7.0"

{apache_airflow_providers_teradata-2.1.1rc1 → apache_airflow_providers_teradata-2.2.0rc1}/airflow/providers/teradata/get_provider_info.py

@@ -28,14 +28,18 @@ def get_provider_info():
         "name": "Teradata",
         "description": "`Teradata <https://www.teradata.com/>`__\n",
         "state": "ready",
-        "source-date-epoch": 1716289161,
-        "versions": ["2.1.1", "2.1.0", "2.0.0"],
+        "source-date-epoch": 1717051506,
+        "versions": ["2.2.0", "2.1.1", "2.1.0", "2.0.0"],
         "dependencies": [
             "apache-airflow>=2.7.0",
             "apache-airflow-providers-common-sql>=1.3.1",
             "teradatasqlalchemy>=17.20.0.0",
             "teradatasql>=17.20.0.28",
         ],
+        "additional-extras": [
+            {"name": "microsoft.azure", "dependencies": ["apache-airflow-providers-microsoft-azure"]},
+            {"name": "amazon", "dependencies": ["apache-airflow-providers-amazon"]},
+        ],
         "integrations": [
             {
                 "integration-name": "Teradata",
@@ -60,7 +64,19 @@ def get_provider_info():
                 "target-integration-name": "Teradata",
                 "python-module": "airflow.providers.teradata.transfers.teradata_to_teradata",
                 "how-to-guide": "/docs/apache-airflow-providers-teradata/operators/teradata_to_teradata.rst",
-            }
+            },
+            {
+                "source-integration-name": "Microsoft Azure Blob Storage",
+                "target-integration-name": "Teradata",
+                "python-module": "airflow.providers.teradata.transfers.azure_blob_to_teradata",
+                "how-to-guide": "/docs/apache-airflow-providers-teradata/operators/azure_blob_to_teradata.rst",
+            },
+            {
+                "source-integration-name": "Amazon Simple Storage Service (S3)",
+                "target-integration-name": "Teradata",
+                "python-module": "airflow.providers.teradata.transfers.s3_to_teradata",
+                "how-to-guide": "/docs/apache-airflow-providers-teradata/operators/s3_to_teradata.rst",
+            },
         ],
         "connection-types": [
             {

{apache_airflow_providers_teradata-2.1.1rc1 → apache_airflow_providers_teradata-2.2.0rc1}/airflow/providers/teradata/hooks/teradata.py

@@ -32,6 +32,17 @@ from airflow.providers.common.sql.hooks.sql import DbApiHook
 if TYPE_CHECKING:
     from airflow.models.connection import Connection
 
+PARAM_TYPES = {bool, float, int, str}
+
+
+def _map_param(value):
+    if value in PARAM_TYPES:
+        # In this branch, value is a Python type; calling it produces
+        # an instance of the type which is understood by the Teradata driver
+        # in the out parameter mapping mechanism.
+        value = value()
+    return value
+
 
 class TeradataHook(DbApiHook):
     """General hook for interacting with Teradata SQL Database.
@@ -187,3 +198,58 @@ class TeradataHook(DbApiHook):
                 "password": "dbc",
             },
         }
+
+    def callproc(
+        self,
+        identifier: str,
+        autocommit: bool = False,
+        parameters: list | dict | None = None,
+    ) -> list | dict | tuple | None:
+        """
+        Call the stored procedure identified by the provided string.
+
+        Any OUT parameters must be provided with a value of either the
+        expected Python type (e.g., `int`) or an instance of that type.
+
+        :param identifier: stored procedure name
+        :param autocommit: What to set the connection's autocommit setting to
+            before executing the query.
+        :param parameters: The `IN`, `OUT` and `INOUT` parameters for Teradata
+            stored procedure
+
+        The return value is a list or mapping that includes parameters in
+        both directions; the actual return type depends on the type of the
+        provided `parameters` argument.
+
+        """
+        if parameters is None:
+            parameters = []
+
+        args = ",".join("?" for name in parameters)
+
+        sql = f"{{CALL {identifier}({(args)})}}"
+
+        def handler(cursor):
+            records = cursor.fetchall()
+
+            if records is None:
+                return
+            if isinstance(records, list):
+                return [row for row in records]
+
+            if isinstance(records, dict):
+                return {n: v for (n, v) in records.items()}
+            raise TypeError(f"Unexpected results: {records}")
+
+        result = self.run(
+            sql,
+            autocommit=autocommit,
+            parameters=(
+                [_map_param(value) for (name, value) in parameters.items()]
+                if isinstance(parameters, dict)
+                else [_map_param(value) for value in parameters]
+            ),
+            handler=handler,
+        )
+
+        return result

{apache_airflow_providers_teradata-2.1.1rc1 → apache_airflow_providers_teradata-2.2.0rc1}/airflow/providers/teradata/operators/teradata.py

@@ -17,11 +17,15 @@
 # under the License.
 from __future__ import annotations
 
-from typing import Sequence
+from typing import TYPE_CHECKING, Sequence
 
+from airflow.models import BaseOperator
 from airflow.providers.common.sql.operators.sql import SQLExecuteQueryOperator
 from airflow.providers.teradata.hooks.teradata import TeradataHook
 
+if TYPE_CHECKING:
+    from airflow.utils.context import Context
+
 
 class TeradataOperator(SQLExecuteQueryOperator):
     """
@@ -34,15 +38,15 @@ class TeradataOperator(SQLExecuteQueryOperator):
         :ref:`howto/operator:TeradataOperator`
 
     :param sql: the SQL query to be executed as a single string, or a list of str (sql statements)
-    :param conn_id: reference to a predefined database
+    :param teradata_conn_id: reference to a predefined database
     :param autocommit: if True, each command is automatically committed.(default value: False)
     :param parameters: (optional) the parameters to render the SQL query with.
     :param schema: The Teradata database to connect to.
     """
 
     template_fields: Sequence[str] = (
-        "parameters",
         "sql",
+        "parameters",
     )
     template_ext: Sequence[str] = (".sql",)
     template_fields_renderers = {"sql": "sql"}
@@ -50,7 +54,7 @@ class TeradataOperator(SQLExecuteQueryOperator):
 
     def __init__(
         self,
-        conn_id: str = TeradataHook.default_conn_name,
+        teradata_conn_id: str = TeradataHook.default_conn_name,
         schema: str | None = None,
         **kwargs,
     ) -> None:
@@ -61,4 +65,39 @@ class TeradataOperator(SQLExecuteQueryOperator):
                 **hook_params,
             }
         super().__init__(**kwargs)
-        self.conn_id = conn_id
+        self.conn_id = teradata_conn_id
+
+
+class TeradataStoredProcedureOperator(BaseOperator):
+    """
+    Executes stored procedure in a specific Teradata database.
+
+    :param procedure: name of stored procedure to call (templated)
+    :param teradata_conn_id: The :ref:`Teradata connection id <howto/connection:teradata>`
+        reference to a specific Teradata database.
+    :param parameters: (optional, templated) the parameters provided in the call
+
+    """
+
+    template_fields: Sequence[str] = (
+        "procedure",
+        "parameters",
+    )
+    ui_color = "#ededed"
+
+    def __init__(
+        self,
+        *,
+        procedure: str,
+        teradata_conn_id: str = TeradataHook.default_conn_name,
+        parameters: dict | list | None = None,
+        **kwargs,
+    ) -> None:
+        super().__init__(**kwargs)
+        self.teradata_conn_id = teradata_conn_id
+        self.procedure = procedure
+        self.parameters = parameters
+
+    def execute(self, context: Context):
+        hook = TeradataHook(teradata_conn_id=self.teradata_conn_id)
+        return hook.callproc(self.procedure, autocommit=True, parameters=self.parameters)

apache_airflow_providers_teradata-2.2.0rc1/airflow/providers/teradata/transfers/azure_blob_to_teradata.py

@@ -0,0 +1,103 @@
+#
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+from __future__ import annotations
+
+from textwrap import dedent
+from typing import TYPE_CHECKING, Sequence
+
+from airflow.models import BaseOperator
+
+try:
+    from airflow.providers.microsoft.azure.hooks.wasb import WasbHook
+except ModuleNotFoundError as e:
+    from airflow.exceptions import AirflowOptionalProviderFeatureException
+
+    raise AirflowOptionalProviderFeatureException(e)
+
+from airflow.providers.teradata.hooks.teradata import TeradataHook
+
+if TYPE_CHECKING:
+    from airflow.utils.context import Context
+
+
+class AzureBlobStorageToTeradataOperator(BaseOperator):
+    """
+
+    Loads CSV, JSON and Parquet format data from Azure Blob Storage to Teradata.
+
+    .. seealso::
+        For more information on how to use this operator, take a look at the guide:
+        :ref:`howto/operator:AzureBlobStorageToTeradataOperator`
+
+    :param blob_source_key: The URI format specifying the location of the Azure blob object store.(templated)
+        The URI format is `/az/YOUR-STORAGE-ACCOUNT.blob.core.windows.net/YOUR-CONTAINER/YOUR-BLOB-LOCATION`.
+        Refer to
+        https://docs.teradata.com/search/documents?query=native+object+store&sort=last_update&virtual-field=title_only&content-lang=en-US
+    :param azure_conn_id: The Airflow WASB connection used for azure blob credentials.
+    :param teradata_table: The name of the teradata table to which the data is transferred.(templated)
+    :param teradata_conn_id: The connection ID used to connect to Teradata
+        :ref:`Teradata connection <howto/connection:Teradata>`
+
+    Note that ``blob_source_key`` and ``teradata_table`` are
+    templated, so you can use variables in them if you wish.
+    """
+
+    template_fields: Sequence[str] = ("blob_source_key", "teradata_table")
+    ui_color = "#e07c24"
+
+    def __init__(
+        self,
+        *,
+        blob_source_key: str,
+        azure_conn_id: str = "azure_default",
+        teradata_table: str,
+        teradata_conn_id: str = "teradata_default",
+        **kwargs,
+    ) -> None:
+        super().__init__(**kwargs)
+        self.blob_source_key = blob_source_key
+        self.azure_conn_id = azure_conn_id
+        self.teradata_table = teradata_table
+        self.teradata_conn_id = teradata_conn_id
+
+    def execute(self, context: Context) -> None:
+        self.log.info(
+            "transferring data from %s to teradata table %s...", self.blob_source_key, self.teradata_table
+        )
+        azure_hook = WasbHook(wasb_conn_id=self.azure_conn_id)
+        conn = azure_hook.get_connection(self.azure_conn_id)
+        # Obtaining the Azure client ID and Azure secret in order to access a specified Blob container
+        access_id = conn.login if conn.login is not None else ""
+        access_secret = conn.password if conn.password is not None else ""
+        teradata_hook = TeradataHook(teradata_conn_id=self.teradata_conn_id)
+        sql = dedent(f"""
+                    CREATE MULTISET TABLE {self.teradata_table}  AS
+                    (
+                        SELECT * FROM (
+                            LOCATION = '{self.blob_source_key}'
+                            ACCESS_ID= '{access_id}'
+                            ACCESS_KEY= '{access_secret}'
+                    ) AS d
+                    ) WITH DATA
+                """).rstrip()
+        try:
+            teradata_hook.run(sql, True)
+        except Exception as ex:
+            self.log.error(str(ex))
+            raise
+        self.log.info("The transfer of data from Azure Blob to Teradata was successful")

apache_airflow_providers_teradata-2.2.0rc1/airflow/providers/teradata/transfers/s3_to_teradata.py

@@ -0,0 +1,109 @@
+#
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+from __future__ import annotations
+
+from textwrap import dedent
+from typing import TYPE_CHECKING, Sequence
+
+from airflow.models import BaseOperator
+
+try:
+    from airflow.providers.amazon.aws.hooks.s3 import S3Hook
+except ModuleNotFoundError as e:
+    from airflow.exceptions import AirflowOptionalProviderFeatureException
+
+    raise AirflowOptionalProviderFeatureException(e)
+from airflow.providers.teradata.hooks.teradata import TeradataHook
+
+if TYPE_CHECKING:
+    from airflow.utils.context import Context
+
+
+class S3ToTeradataOperator(BaseOperator):
+    """
+    Loads CSV, JSON and Parquet format data from Amazon S3 to Teradata.
+
+    .. seealso::
+        For more information on how to use this operator, take a look at the guide:
+        :ref:`howto/operator:S3ToTeradataOperator`
+
+    :param s3_source_key: The URI format specifying the location of the S3 bucket.(templated)
+        The URI format is /s3/YOUR-BUCKET.s3.amazonaws.com/YOUR-BUCKET-NAME.
+        Refer to
+        https://docs.teradata.com/search/documents?query=native+object+store&sort=last_update&virtual-field=title_only&content-lang=en-US
+    :param public_bucket: Specifies whether the provided S3 bucket is public. If the bucket is public,
+        it means that anyone can access the objects within it via a URL without requiring authentication.
+        If the bucket is private and authentication is not provided, the operator will throw an exception.
+    :param teradata_table: The name of the teradata table to which the data is transferred.(templated)
+    :param aws_conn_id: The Airflow AWS connection used for AWS credentials.
+    :param teradata_conn_id: The connection ID used to connect to Teradata
+        :ref:`Teradata connection <howto/connection:Teradata>`.
+
+    Note that ``s3_source_key`` and ``teradata_table`` are
+    templated, so you can use variables in them if you wish.
+    """
+
+    template_fields: Sequence[str] = ("s3_source_key", "teradata_table")
+    ui_color = "#e07c24"
+
+    def __init__(
+        self,
+        *,
+        s3_source_key: str,
+        public_bucket: bool = False,
+        teradata_table: str,
+        aws_conn_id: str = "aws_default",
+        teradata_conn_id: str = "teradata_default",
+        **kwargs,
+    ) -> None:
+        super().__init__(**kwargs)
+        self.s3_source_key = s3_source_key
+        self.public_bucket = public_bucket
+        self.teradata_table = teradata_table
+        self.aws_conn_id = aws_conn_id
+        self.teradata_conn_id = teradata_conn_id
+
+    def execute(self, context: Context) -> None:
+        self.log.info(
+            "transferring data from %s to teradata table %s...", self.s3_source_key, self.teradata_table
+        )
+
+        s3_hook = S3Hook(aws_conn_id=self.aws_conn_id)
+        access_key = ""
+        access_secret = ""
+        if not self.public_bucket:
+            credentials = s3_hook.get_credentials()
+            access_key = credentials.access_key
+            access_secret = credentials.secret_key
+        teradata_hook = TeradataHook(teradata_conn_id=self.teradata_conn_id)
+        sql = dedent(f"""
+                        CREATE MULTISET TABLE {self.teradata_table} AS
+                        (
+                            SELECT * FROM (
+                                LOCATION = '{self.s3_source_key}'
+                                ACCESS_ID= '{access_key}'
+                                ACCESS_KEY= '{access_secret}'
+                            ) AS d
+                        ) WITH DATA
+                        """).rstrip()
+        try:
+            teradata_hook.run(sql, True)
+        except Exception as ex:
+            self.log.error(str(ex))
+            raise
+        self.log.info("The transfer of data from S3 to Teradata was successful")

{apache_airflow_providers_teradata-2.1.1rc1 → apache_airflow_providers_teradata-2.2.0rc1}/pyproject.toml

@@ -28,7 +28,7 @@ build-backend = "flit_core.buildapi"
 
 [project]
 name = "apache-airflow-providers-teradata"
-version = "2.1.1.rc1"
+version = "2.2.0.rc1"
 description = "Provider package apache-airflow-providers-teradata for Apache Airflow"
 readme = "README.rst"
 authors = [
@@ -63,8 +63,8 @@ dependencies = [
 ]
 
 [project.urls]
-"Documentation" = "https://airflow.apache.org/docs/apache-airflow-providers-teradata/2.1.1"
-"Changelog" = "https://airflow.apache.org/docs/apache-airflow-providers-teradata/2.1.1/changelog.html"
+"Documentation" = "https://airflow.apache.org/docs/apache-airflow-providers-teradata/2.2.0"
+"Changelog" = "https://airflow.apache.org/docs/apache-airflow-providers-teradata/2.2.0/changelog.html"
 "Bug Tracker" = "https://github.com/apache/airflow/issues"
 "Source Code" = "https://github.com/apache/airflow"
 "Slack Chat" = "https://s.apache.org/airflow-slack"
@@ -74,9 +74,15 @@ dependencies = [
 [project.entry-points."apache_airflow_provider"]
 provider_info = "airflow.providers.teradata.get_provider_info:get_provider_info"
 [project.optional-dependencies]
+"amazon" = [
+    "apache-airflow-providers-amazon",
+]
 "common.sql" = [
     "apache-airflow-providers-common-sql",
 ]
+"microsoft.azure" = [
+    "apache-airflow-providers-microsoft-azure",
+]
 
 [tool.flit.module]
 name = "airflow.providers.teradata"