Skip to content

Commit

Permalink
Update teradata.py
Browse files Browse the repository at this point in the history
  • Loading branch information
satish-chinthanippu committed Feb 9, 2024
1 parent d1c08e1 commit e859a1d
Showing 1 changed file with 16 additions and 0 deletions.
16 changes: 16 additions & 0 deletions airflow/providers/teradata/hooks/teradata.py
Original file line number Diff line number Diff line change
Expand Up @@ -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


Expand Down Expand Up @@ -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

0 comments on commit e859a1d

Please sign in to comment.