feat: Issue339 ldap logmech

data_validation/cli_tools.py

@@ -64,6 +64,7 @@
         ["port", "Teradata port to connect on"],
         ["user_name", "User used to connect"],
         ["password", "Password for supplied user"],
+        ["logmech", "Log on mechanism"],
     ],
     "Oracle": [
         ["host", "Desired Oracle host"],

docs/connections.md

@@ -129,6 +129,7 @@ via `pip install teradatasql` if you have a license.
     # Connection Details
     "host": "127.0.0.1",
     "port":1025,
+    "logmech":"TD2",
     "user_name":"my-user",
     "password":"my-password"
 }

third_party/ibis/ibis_teradata/api.py

@@ -38,7 +38,7 @@ def verify(expr, params=None):
 
 
 def connect(
-    host: str, user_name: str, password: str, port: Optional[int] = 1025
+    host: str, user_name: str, password: str, port: Optional[int] = 1025, logmech: Optional[str]='TD2'
 ) -> TeradataClient:
     """ Create a TeradataClient for use with Ibis.
     Parameters
@@ -49,11 +49,13 @@ def connect(
         A Database username to connect with
     password : str
         Password for supplied username
+    logmech : Optional[str]
+        Logmech flag to select with (default. TD2)
     port : Optional[int]
         The database port to connect to (default. 1025)
     Returns
     -------
     TeradataClient
     """
 
-    return TeradataClient(host, user_name, password, port)
+    return TeradataClient(host, user_name, password, logmech, port)

third_party/ibis/ibis_teradata/client.py

@@ -66,7 +66,7 @@ class TeradataClient(SQLClient):
     table_class = TeradataTable
     dialect = compiler.TeradataDialect
 
-    def __init__(self, host, user_name, password, port=1025, use_no_lock_tables=False):
+    def __init__(self, host, user_name, password, port=1025, logmech="TD2", use_no_lock_tables=False):
         """Construct a TeradataClient.
 
         Parameters
@@ -85,6 +85,7 @@ def __init__(self, host, user_name, password, port=1025, use_no_lock_tables=Fals
             "user": user_name,
             "password": password,
             "dbs_port": port,
+            "logmech": logmech,
         }
 
         self.client = teradatasql.connect(**self.teradata_config)