From e859a1d59ae913cae51548d57806d7f17da43369 Mon Sep 17 00:00:00 2001 From: Satish Ch Date: Fri, 9 Feb 2024 19:21:28 +0530 Subject: [PATCH] Update teradata.py --- airflow/providers/teradata/hooks/teradata.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/airflow/providers/teradata/hooks/teradata.py b/airflow/providers/teradata/hooks/teradata.py index 48c7930a266e58..528f8653465dd0 100644 --- a/airflow/providers/teradata/hooks/teradata.py +++ b/airflow/providers/teradata/hooks/teradata.py @@ -18,6 +18,11 @@ """An Airflow Hook for interacting with Teradata SQL Server.""" from __future__ import annotations +from typing import TYPE_CHECKING + +import teradatasql +from teradatasql import TeradataConnection + from airflow.providers.common.sql.hooks.sql import DbApiHook @@ -69,3 +74,14 @@ def __init__( **kwargs, ) -> None: super().__init__(*args, schema=database, **kwargs) + + def get_conn(self) -> TeradataConnection: + """Creates and returns a Teradata Connection object using teradatasql client. + + Establishes connection to a Teradata SQL database using config corresponding to teradata_conn_id. + + :return: a Teradata connection object + """ + teradata_conn_config: dict = self._get_conn_config_teradatasql() + teradata_conn = teradatasql.connect(**teradata_conn_config) + return teradata_conn