added request_timeout and restriction for teradatasql

Teradata · Sep 21, 2023 · ddc38d3 · ddc38d3
1 parent fedc6ae
commit ddc38d3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/dbt/adapters/teradata/connections.py b/dbt/adapters/teradata/connections.py
@@ -55,6 +55,7 @@ class TeradataCredentials(Credentials):
     logon_timeout: Optional[int] = None
     https_port: Optional[int] = None
     connect_timeout: Optional[int] = None
+    request_timeout: Optional[int] = None
 
     _ALIASES = {
         "UID": "username",
@@ -136,7 +137,8 @@ def _connection_keys(self):
             "runstartup",
             "logon_timeout",
             "https_port",
-            "connect_timeout"
+            "connect_timeout",
+            "request_timeout"
         )
 
 
@@ -235,6 +237,8 @@ def open(cls, connection):
            kwargs["https_port"]=credentials.https_port
         if credentials.connect_timeout:
            kwargs["connect_timeout"]=credentials.connect_timeout
+        if credentials.request_timeout:
+           kwargs["request_timeout"]=credentials.request_timeout
 
         # Save the transaction mode
         cls.TMODE = credentials.tmode

diff --git a/requirements_dev.txt b/requirements_dev.txt
@@ -5,7 +5,7 @@ pytest~=7.0
 tox~=3.2
 dbt-tests-adapter~=1.5.4
 pylava~=0.3.0
-teradatasql>=16.20.0.0
+teradatasql>=17.20.0.30
 dbt-core==1.5.4
 MarkupSafe==2.0.1
 pytest-dotenv