teradataml 20.0.0.3__py3-none-any.whl → 20.0.0.5__py3-none-any.whl

teradataml/context/context.py

@@ -13,32 +13,33 @@ A teradataml context functions provide interface to Teradata Vantage. Provides f
 context which can be used by other analytical functions to get the Teradata Vantage connection.
 
 """
+import atexit
 import ipaddress
+import os
+import socket
+import sys
+import threading
+import warnings
 from pathlib import Path
+
+from dotenv import dotenv_values
 from sqlalchemy import create_engine
-from teradataml.telemetry_utils.queryband import collect_queryband
-from teradataml.common.deprecations import argument_deprecation
+from sqlalchemy.engine.base import Engine
+from sqlalchemy.engine.url import URL
+
+from teradataml.common.constants import Query, SQLConstants, TeradataConstants
 from teradataml.common.exceptions import TeradataMlException
-from teradataml.common.warnings import TeradataMlRuntimeWarning
-from teradataml.common.messages import Messages
+from teradataml.common.garbagecollector import GarbageCollector
 from teradataml.common.messagecodes import MessageCodes
+from teradataml.common.messages import Messages
 from teradataml.common.sqlbundle import SQLBundle
-from teradataml.common.constants import SQLConstants
-from teradataml.common.garbagecollector import GarbageCollector
+from teradataml.common.warnings import TeradataMlRuntimeWarning
 from teradataml.context.aed_context import AEDContext
-from teradataml.common.constants import TeradataConstants, Query
 from teradataml.options.configure import configure
+from teradataml.telemetry_utils.queryband import collect_queryband
+from teradataml.utils.internal_buffer import _InternalBuffer
 from teradataml.utils.utils import execute_sql
 from teradataml.utils.validators import _Validators
-from teradataml.utils.internal_buffer import _InternalBuffer
-from sqlalchemy.engine.base import Engine
-from sqlalchemy.engine.url import URL
-import os
-import warnings
-import atexit
-import socket
-import threading
-import urllib.parse
 
 # Store a global Teradata Vantage Connection.
 # Right now user can only provide a single Vantage connection at any point of time.
@@ -47,6 +48,8 @@ td_sqlalchemy_engine = None
 temporary_database_name = None
 user_specified_connection = False
 python_packages_installed = False
+python_version_vantage = None
+python_version_local = None
 td_user = None
 
 function_alias_mappings = {}
@@ -133,7 +136,7 @@ def __cleanup_garbage_collection():
     GarbageCollector._cleanup_garbage_collector()
 
 
-def _get_other_connection_parameters(logmech = None, logdata = None, database = None, **kwargs):
+def _get_other_connection_parameters(logmech=None, logdata=None, database=None, **kwargs):
     """
     DESCRIPTION:
         Internal function to return the connection parameters.
@@ -194,16 +197,26 @@ def _get_other_connection_parameters(logmech = None, logdata = None, database =
 
     return result
 
-@argument_deprecation("future", ["url_encode"], False, None)
+
 @collect_queryband(queryband='CrtCxt')
-def create_context(host = None, username = None, password = None, tdsqlengine = None, temp_database_name = None,
-                   logmech = None, logdata = None, database = None, url_encode = True, **kwargs):
+def create_context(host=None, username=None, password=None, tdsqlengine=None, temp_database_name=None,
+                   logmech=None, logdata=None, database=None, **kwargs):
     """
     DESCRIPTION:
         Creates a connection to the Teradata Vantage using the teradatasql + teradatasqlalchemy DBAPI and dialect
-        combination. Users can pass all required parameters (host, username, password) for establishing a connection to
-        Vantage, or pass a sqlalchemy engine to the tdsqlengine parameter to override the default DBAPI and dialect
         combination.
+        Users can create a connection by passing the connection parameters using any one of the following methods:
+            1. Pass all required parameters (host, username, password) directly to the function.
+            2. Set the connection parameters in a configuration file (.cfg or .env) and
+               pass the configuration file.
+            3. Set the connection parameters in environment variables and create_context() reads from
+               environment variables.
+
+        Alternatively, users can pass a SQLAlchemy engine object to the `tdsqlengine` parameter to override the default DBAPI
+        and dialect combination.
+
+        Function also enables user to set the authentication token which is required to access services running
+        on Teradata Vantage.
 
         Note:
             1. teradataml requires that the user has certain permissions on the user's default database or the initial
@@ -241,6 +254,39 @@ def create_context(host = None, username = None, password = None, tdsqlengine =
                |                                                      | default database.                           |
                +------------------------------------------------------+---------------------------------------------+
 
+            3. The function prioritizes parameters in the following order:
+                1. Explicitly passed arguments (host, username, password).
+                2. Environment variables (TD_HOST, TD_USERNAME, TD_PASSWORD, etc.).
+                   Note:
+                       * The environment variables should start with 'TD_' and all must be in upper case.
+                         Example:
+                                os.environ['TD_HOST'] = 'tdhost'
+                                os.environ['TD_USERNAME'] = 'tduser'
+                                os.environ['TD_PASSWORD'] = 'tdpassword'
+                3. A configuration file if provided (such as a .env file).
+
+            4. Points to note when user sets authentication token with create_context():
+                * The username provided in create_context() is not case-sensitive. For example,
+                  if a user is created with the username xyz, create_context() still establishes
+                  a connection if user passes the username as XyZ. However, authentication token
+                  generation requires the username to be in the same case as when it was created.
+                  Therefore, Teradata recommends to pass the username with the same case as when
+                  it was created.
+                * User must have a privilege to login with a NULL password to use set_auth_token().
+                  Refer to GRANT LOGON section in Teradata Documentation for more details.
+                * When "auth_mech" is not specified, arguments are used in the following combination
+                  to derive authentication mechanism.
+                    * If "base_url" and "client_id" are specified then token generation is done through OAuth.
+                    * If "base_url", "pat_token", "pem_file" are specified then token generation is done using PAT.
+                    * If "base_url" and "auth_token" are specified then value provided for "auth_token" is used.
+                    * If only "base_url" is specified then token generation is done through OAuth.
+                * If Basic authentication mechanism is to be used then user must specify argument
+                  "auth_mech" as "BASIC" along with "username" and "password".
+                * Refresh token works only for OAuth authentication.
+                * Use the argument "kid" only when key used during the pem file generation is different
+                  from pem file name. For example, if you use the key as 'key1' while generating pem file
+                  and the name of the pem file is `key1(1).pem`, then pass value 'key1' to the argument "kid".
+
     PARAMETERS:
         host:
             Optional Argument.
@@ -254,39 +300,15 @@ def create_context(host = None, username = None, password = None, tdsqlengine =
         
         password:
             Optional Argument.
-            Specifies the password required for the username.
+            Specifies the password required for the "username".
             Types: str
             Note:
                 * Encrypted passwords can also be passed to this argument, using Stored Password Protection feature.
                   Examples section below demonstrates passing encrypted password to 'create_context'.
                   More details on Stored Password Protection and how to generate key and encrypted password file
                   can be found at https://pypi.org/project/teradatasql/#StoredPasswordProtection
-                * Special characters that may be used in the password are encoded by default, except '~'.
-                  The encoding is done using urllib.parse library and can be
-                  disabled by setting "url_encode" = False.
-                  For example, if the password is: "kx%jj5/g", then this password is encoded as below:
-                  "kx%25jj5%2Fg"
-                   where,
-                      %25 represents the '%' character and
-                      %2F represents the '/' character
-                  The details of how the special characters are replaced can be found in the below link:
-                  "https://docs.microfocus.com/OMi/10.62/Content/OMi/ExtGuide/ExtApps/URL_encoding.htm".
-                  Note:
-                      When password contains a space:
-                        * "url_encode" must be set to False.
-                        * In addition to space, other special characters in this password should be manually encoded,
-                          following the link: https://docs.microfocus.com/OMi/10.62/Content/OMi/ExtGuide/ExtApps/URL_encoding.htm
-                          Refer Example 16 in examples section for a detailed demonstration.
-                      
-                      When password contains unreserved characters like tilde("~"), dot("."), underscore("_"), hyphen("-"):
-                        * Character is not URL encoded by default.
-                        * If unreserved character in passed needs to be encoded then,
-                          it must be encoded manually and encoded password must be passed, along with "url_encode" set to False.
-                        * The unreserved characters differ from one Python version to another Python version.
-                          For example, The encoding standards for Python version 3.7 found in the below link:
-                          "https://docs.python.org/3/library/urllib.parse.html#url-quoting".
-                      
-            
+                * Special characters used in the password are encoded by default.
+
         tdsqlengine:
             Optional Argument.
             Specifies Teradata Vantage sqlalchemy engine object that should be used to establish a Teradata Vantage
@@ -324,12 +346,90 @@ def create_context(host = None, username = None, password = None, tdsqlengine =
             Types: str
 
         kwargs:
-            Specifies the keyword-value pairs of connection parameters that are passed to Teradata SQL Driver for
-            Python. Please refer to https://github.com/Teradata/python-driver#ConnectionParameters to get information
-            on connection parameters of the driver.
-            Note: When the type of a connection parameter is integer or boolean (eg: log, lob_support etc,.), pass
-                  integer or boolean value, instead of quoted integer or quoted boolean as suggested in the
-                  documentation. Please check the examples for usage.
+            Specifies optional keyword arguments accepted by create_context().
+            Below are the supported keyword arguments:
+
+            Connection parameters for Teradata SQL Driver:
+                Specifies the keyword-value pairs of connection parameters that are passed to Teradata SQL Driver for
+                Python. Please refer to https://github.com/Teradata/python-driver#ConnectionParameters to get information
+                on connection parameters of the driver.
+                Note:
+                    * When type of a connection parameter is integer or boolean (eg: log, lob_support etc,.), pass
+                      integer or boolean value, instead of quoted integer or quoted boolean as suggested in the
+                      documentation. Please check the examples for usage.
+                    * "sql_timeout" represents "request_timeout" connection parameter.
+
+            config_file:
+                Specifies the name of the configuration file to read the connection parameters.
+                Notes:
+                    * If user does not specify full path of file, then file look up is done at current working directory.
+                    * The content of the file must be in '.env' format.
+                    * Use parameters of create_context() as key in the configuration file.
+                      Example:
+                          host=tdhost
+                          username=tduser
+                          password=tdpassword
+                          temp_database_name=tdtemp_database_name
+                          logmech=tdlogmech
+                          logdata=tdlogdata
+                          database=tddatabase
+                    * For more information please refer examples section.
+                Default Value : td_properties.cfg
+                Types: str
+
+            base_url:
+                Specifies the endpoint URL for a given environment on Teradata Vantage.
+                Types: str
+
+            client_id:
+                Specifies the id of the application that requests the access token from
+                VantageCloud Lake.
+                Types: str
+
+            pat_token:
+                Specifies the PAT token generated from VantageCloud Lake Console.
+                Types: str
+
+            pem_file:
+                Specifies the path to private key file which is generated from VantageCloud Lake Console.
+                Types: str
+
+            auth_token:
+                Specifies the authentication token required to access services running
+                on Teradata Vantage.
+
+            expiration_time:
+                Specifies the expiration time of the token in seconds. After expiry time JWT token expires and
+                UserEnv methods does not work, user should regenerate the token.
+                Note:
+                    This option is used only for PAT and not for OAuth.
+                Default Value: 31536000
+                Types: int
+
+            kid:
+                Specifies the name of the key which is used while generating 'pem_file'.
+                Note:
+                    * Use the argument "kid" only when key used during the pem file generation is different
+                      from pem file name. For example, if you use the key as 'key1' while generating pem file
+                      and the name of the pem file is `key1(1).pem`, then pass value 'key1' to the argument "kid".
+                Types: str
+
+            auth_mech:
+                Specifies the mechanism to be used for generating authentication token.
+                Notes:
+                    * User must use this argument if Basic authentication is to be used.
+                    * When "auth_mech" is provided, other arguments are used in the following
+                      combination as per value of "auth_mech":
+                        * OAuth: Token generation is done through OAuth by using client id
+                                 which can be sepcified by user in "client_id" argument or
+                                 can be derived internally from "base_url".
+                        * PAT  : Token generation is done using "pat_token" and "pem_file".
+                        * BASIC: Authentication is done via Basic authentication mechanism
+                                 using user credentials passed in "username" and "password"
+                                 arguments.
+                        * JWT  : Readily available token in "auth_token" argument is used.
+                Permitted Values: "OAuth", "PAT", "BASIC", "JWT".
+                Types: str
 
     RETURNS:
         A Teradata sqlalchemy engine object.
@@ -341,7 +441,7 @@ def create_context(host = None, username = None, password = None, tdsqlengine =
         >>> from teradataml.context.context import *
 
         # Example 1: Create context using hostname, username and password
-        >>> td_context = create_context(host = 'tdhost', username='tduser', password = 'tdpassword')
+        >>> td_context = create_context(host = 'tdhost', username = 'tduser', password = 'tdpassword')
 
         # Example 2: Create context using already created sqlalchemy engine
         >>> from sqlalchemy import create_engine
@@ -349,56 +449,141 @@ def create_context(host = None, username = None, password = None, tdsqlengine =
         >>> td_context = create_context(tdsqlengine = sqlalchemy_engine)
 
         # Example 3: Creating context for Vantage with default logmech 'TD2'
-        >>> td_context = create_context(host = 'tdhost', username='tduser', password = 'tdpassword', logmech='TD2')
+        >>> td_context = create_context(host = 'tdhost', username = 'tduser', password = 'tdpassword', logmech = 'TD2')
 
        # Example 4: Creating context for Vantage with logmech as 'TDNEGO'
-        >>> td_context = create_context(host = 'tdhost', username='tduser', password = 'tdpassword', logmech='TDNEGO')
+        >>> td_context = create_context(host = 'tdhost', username = 'tduser', password = 'tdpassword', logmech = 'TDNEGO')
 
         # Example 5: Creating context for Vantage with logmech as 'LDAP'
-        >>> td_context = create_context(host = 'tdhost', username='tduser', password = 'tdpassword', logmech='LDAP')
+        >>> td_context = create_context(host = 'tdhost', username = 'tduser', password = 'tdpassword', logmech = 'LDAP')
 
         # Example 6: Creating context for Vantage with logmech as 'KRB5'
-        >>> td_context = create_context(host = 'tdhost', logmech='KRB5')
+        >>> td_context = create_context(host = 'tdhost', logmech = 'KRB5')
 
         # Example 7: Creating context for Vantage with logmech as 'JWT'
-        >>> td_context = create_context(host = 'tdhost', logmech='JWT', logdata='token=eyJpc...h8dA')
+        >>> td_context = create_context(host = 'tdhost', logmech = 'JWT', logdata = 'token=eyJpc...h8dA')
 
         # Example 8: Create context using encrypted password and key passed to 'password' parameter.
-        # The password should be specified in the format mentioned below:
-        # ENCRYPTED_PASSWORD(file:<PasswordEncryptionKeyFileName>, file:<EncryptedPasswordFileName>)
-        # The PasswordEncryptionKeyFileName specifies the name of a file that contains the password encryption key
-        # and associated information.
-        # The EncryptedPasswordFileName specifies the name of a file that contains the encrypted password and
-        # associated information.
-        # Each filename must be preceded by the 'file:' prefix. The PasswordEncryptionKeyFileName must be separated
-        # from the EncryptedPasswordFileName by a single comma.
+        #            The password should be specified in the format mentioned below:
+        #            ENCRYPTED_PASSWORD(file:<PasswordEncryptionKeyFileName>, file:<EncryptedPasswordFileName>)
+        #            The PasswordEncryptionKeyFileName specifies the name of a file that contains the password encryption key
+        #            and associated information.
+        #            The EncryptedPasswordFileName specifies the name of a file that contains the encrypted password and
+        #            associated information.
+        #            Each filename must be preceded by the 'file:' prefix. The PasswordEncryptionKeyFileName must be separated
+        #            from the EncryptedPasswordFileName by a single comma.
         >>> encrypted_password = "ENCRYPTED_PASSWORD(file:PassKey.properties, file:EncPass.properties)"
-        >>> td_context = create_context(host = 'tdhost', username='tduser', password = encrypted_password)
+        >>> td_context = create_context(host = 'tdhost', username = 'tduser', password = encrypted_password)
 
         # Example 9: Create context using encrypted password in LDAP logon mechanism.
-        >>> td_context = create_context(host = 'tdhost', username='tduser', password = encrypted_password,
-                                        logmech='LDAP')
+        >>> td_context = create_context(host = 'tdhost', username = 'tduser', password = encrypted_password,
+        ...                             logmech = 'LDAP')
 
        # Example 10: Create context using hostname, username, password and database parameters, and connect to a
        # different initial database by setting the database parameter.
-        >>> td_context = create_context(host = 'tdhost', username='tduser', password = 'tdpassword', database =
-                                        'database_name')
+        >>> td_context = create_context(host = 'tdhost', username = 'tduser', password = 'tdpassword', database =
+        ...                            'database_name')
 
         # Example 11: Create context using already created sqlalchemy engine, and connect to a different initial
         # database by setting the database parameter.
         >>> from sqlalchemy import create_engine
         >>> sqlalchemy_engine = create_engine('teradatasql://'+ tduser +':' + tdpassword + '@'+tdhost +
-                                               '/?DATABASE=database_name')
+        ...                                   '/?DATABASE=database_name')
         >>> td_context = create_context(tdsqlengine = sqlalchemy_engine)
 
         # Example 12: Create context for Vantage with logmech as 'LDAP', and connect to a different initial
         # database by setting the database parameter.
-        >>> td_context = create_context(host = 'tdhost', username='tduser', password = 'tdpassword', logmech='LDAP',
-                                        database = 'database_name')
+        >>> td_context = create_context(host = 'tdhost', username = 'tduser', password = 'tdpassword', logmech = 'LDAP',
+        ...                             database = 'database_name')
 
         # Example 13: Create context using 'tera' mode with log value set to 8 and lob_support disabled.
-        >>> td_context = create_context(host = 'tdhost', username='tduser', password = 'tdpassword', tmode = 'tera',
-                                        log = 8, lob_support = False)
+        >>> td_context = create_context(host = 'tdhost', username = 'tduser', password = 'tdpassword', tmode = 'tera',
+        ...                             log = 8, lob_support = False)
+
+        # Example 14: Create context from config file with name 'td_properties.cfg'
+        #             available under current working directory.
+        #             td_properties.cfg content:
+        #                 host=tdhost
+        #                 username=tduser
+        #                 password=tdpassword
+        #                 temp_database_name=tdtemp_database_name
+        #                 logmech=tdlogmech
+        #                 logdata=tdlogdata
+        #                 database=tddatabase
+        >>> td_context = create_context()
+
+        # Example 15: Create context using the file specified in user's home directory
+        #             with name user_td_properties.cfg.
+        #             user_td_properties.cfg content:
+        #                 host=tdhost
+        #                 username=tduser
+        #                 password=tdpassword
+        #                 temp_database_name=tdtemp_database_name
+        #                 logmech=tdlogmech
+        #                 logdata=tdlogdata
+        #                 database=tddatabase
+        >>> td_context = create_context(config_file = "user_td_properties.cfg")
+
+        # Example 16: Create context using environment variables.
+        #             Set these using os.environ and then run the example:
+        #                  os.environ['TD_HOST'] = 'tdhost'
+        #                  os.environ['TD_USERNAME'] = 'tduser'
+        #                  os.environ['TD_PASSWORD'] = 'tdpassword'
+        #                  os.environ['TD_TEMP_DATABASE_NAME'] = 'tdtemp_database_name'
+        #                  os.environ['TD_LOGMECH'] = 'tdlogmech'
+        #                  os.environ['TD_LOGDATA'] = 'tdlogdata'
+        #                  os.environ['TD_DATABASE'] = 'tddatabase'
+        >>> td_context = create_context()
+
+        # Example 17: Create a context by providing username and password. Along with it,
+        #             set authentication token by providing the pem file and pat token.
+        >>> td_context = create_context(host = 'tdhost', username = 'tduser', password = 'tdpassword',
+        ...                             base_url = 'base_url', pat_token = 'pat_token', pem_file = 'pem_file')
+
+        # Example 18: Create a context by providing username and password. Along with it,
+        #             generate authentication token by providing the client id.
+        >>> td_context = create_context(host = 'tdhost', username = 'tduser', password = 'tdpassword',
+        ...                             base_url = 'base_url', client_id = 'client_id')
+
+        # Example 19: Create context and set authentication token by providing all the details in a config file.
+        #             td_properties.cfg content:
+        #                 host=tdhost
+        #                 username=tduser
+        #                 password=tdpassword
+        #                 base_url=base_url
+        #                 pat_token=pat_token
+        #                 pem_file=pem_file
+        >>> td_context = create_context()
+
+        # Example 20: Create context and set authentication token by providing all the details in environment variables.
+        #             Set these using os.environ and then run the example:
+        #                  os.environ['TD_HOST'] = 'tdhost'
+        #                  os.environ['TD_USERNAME'] = 'tduser'
+        #                  os.environ['TD_PASSWORD'] = 'tdpassword'
+        #                  os.environ['TD_BASE_URL'] = 'base_url'
+        #                  os.environ['TD_PAT_TOKEN'] = 'pat_token'
+        #                  os.environ['TD_PEM_FILE'] = 'pem_file'
+        >>> td_context = create_context()
+
+        # Example 21: Create context with sql_timeout set to 3.
+        >>> td_context = create_context(host = 'tdhost', username = 'tduser', password = 'tdpassword', sql_timeout = 3)
+
+        # Example 22: Create context and set authentication token via Basic authentication mechanism
+        #             using username and password.
+        >>> import getpass
+        >>> create_context(host="host",
+        ...                username=getpass.getpass("username : "),
+        ...                password=getpass.getpass("password : "),
+        ...                base_url=getpass.getpass("base_url : "),
+        ...                auth_mech="BASIC")
+
+        # Example 23: Create context and set authentication token by providing auth_mech argument.
+        >>> import getpass
+        >>> create_context(host="vcl_host",
+        ...                username=getpass.getpass("username : "),
+        ...                password=getpass.getpass("password : "),
+        ...                base_url=getpass.getpass("base_url : "),
+        ...                auth_mech="OAuth")
 
     """
     global td_connection
@@ -406,7 +591,16 @@ def create_context(host = None, username = None, password = None, tdsqlengine =
     global temporary_database_name
     global user_specified_connection
     global python_packages_installed
+    global python_version_vantage
+    global python_version_local
     global td_user
+
+    # Check if the user has provided the connection parameters or tdsqlengine.
+    # If not, check if the user has provided the connection parameters in the environment variables.
+    # If not, check if the user has provided the connection parameters in the config file.
+    if not (host or tdsqlengine) and host != "":
+        return _load_context_from_env_config(kwargs.pop('config_file', 'td_properties.cfg'))
+
     awu_matrix = []
     awu_matrix.append(["host", host, True, (str), True])
     awu_matrix.append(["username", username, True, (str), True])
@@ -415,13 +609,23 @@ def create_context(host = None, username = None, password = None, tdsqlengine =
     awu_matrix.append(["logmech", logmech, True, (str), True])
     awu_matrix.append(["logdata", logdata, True, (str), True])
     awu_matrix.append(["database", database, True, (str), True])
+    # set_auth_token parameters
+    _set_auth_token_params = {}
+    auth_mech = kwargs.get('auth_mech', None)
+    for param in ['base_url', 'pat_token', 'pem_file', 'client_id', 'auth_token', 'expiration_time', 'kid', 'auth_mech']:
+        if kwargs.get(param):
+            _set_auth_token_params[param] = kwargs.pop(param)
+
+    # Set the "sql_timeout" parameter to "request_timeout" which is consumed by teradatasql.
+    if kwargs.get('sql_timeout'):
+        awu_matrix.append(["sql_timeout", kwargs.get('sql_timeout'), True, (int), True])
+        kwargs['request_timeout'] = kwargs.pop('sql_timeout')
 
     awu = _Validators()
     awu._validate_function_arguments(awu_matrix)
 
     # Clearing the internal buffer.
     _InternalBuffer.clean()
-
     if logmech == "JWT" and not logdata:
         raise TeradataMlException(Messages.get_message(MessageCodes.DEPENDENT_ARG_MISSING,
                                                        'logdata',
@@ -456,7 +660,6 @@ def create_context(host = None, username = None, password = None, tdsqlengine =
             host_value = '{}:@{}'.format(username, host)
         else:
             host_value = '{}:{}@{}'.format(username, password, host)
-
         url_object = URL.create(
             "teradatasql",
             username=username,
@@ -470,7 +673,7 @@ def create_context(host = None, username = None, password = None, tdsqlengine =
             td_connection = td_sqlalchemy_engine.connect()
             td_user = username.upper()
 
-            # Masking senstive information - password, logmech and logdata.
+            # Masking sensitive information - password, logmech and logdata.
             if password:
                 try:
                     # Below statement raises an AttributeError with SQLAlchemy
@@ -497,6 +700,8 @@ def create_context(host = None, username = None, password = None, tdsqlengine =
     _load_function_aliases()
 
     python_packages_installed = False
+    python_version_vantage = None
+    python_version_local = sys.version.split(" ")[0].strip()
 
     # Assign the tempdatabase name to global
     if temp_database_name is None:
@@ -520,7 +725,19 @@ def create_context(host = None, username = None, password = None, tdsqlengine =
     # Process Analytic functions.
     from teradataml.analytics import _process_analytic_functions
     _process_analytic_functions()
-    
+
+    if _set_auth_token_params.get('base_url'):
+        from teradataml.scriptmgmt.lls_utils import set_auth_token
+        try:
+            if auth_mech and auth_mech.lower() == 'basic' and password:
+                _set_auth_token_params['password'] = password
+            set_auth_token(**_set_auth_token_params)
+        except Exception as err:
+            print("Connection to Vantage established successfully.")
+            mssg = f"Failed to set authentication token. Rerun \"set_auth_token()\" again to set the authentication token." \
+                   f" Reason for failure: {err.args[0]}"
+            warnings.warn(mssg, stacklevel=2)
+
     # Add global lock to internal buffer
     _InternalBuffer.add(global_lock=threading.Lock())
 
@@ -528,6 +745,36 @@ def create_context(host = None, username = None, password = None, tdsqlengine =
     return td_sqlalchemy_engine
 
 
+def _load_context_from_env_config(config_file=None):
+    """
+    DESCRIPTION:
+        Reads the connection parameters from the configuration file or environment variables.
+
+    PARAMETERS:
+        config_file:
+            Optional Argument.
+            Specifies the name of the configuration file to read the connection parameters.
+            Types: str
+
+    RETURNS:
+        A Teradata sqlalchemy engine object.
+
+    RAISES:
+        TeradataMlException
+    """
+    host = os.environ.get('TD_HOST')
+    if host:
+        connection_params_from_env = {key[3:].lower(): value for key, value in os.environ.items()
+                                      if key.startswith('TD_')}
+        return create_context(**connection_params_from_env)
+    elif config_file is not None:
+        connection_params_from_file = dotenv_values(config_file)
+        if connection_params_from_file.get('host'):
+            return create_context(**connection_params_from_file)
+    raise TeradataMlException(Messages.get_message(MessageCodes.CONNECTION_PARAMS),
+                              MessageCodes.MISSING_ARGS)
+
+
 def _mask_logmech_logdata():
     """
     Masks sensitive connection information LOGMECH, LOGDATA exposed by sqlalchemy engine object
@@ -625,6 +872,8 @@ def set_context(tdsqlengine, temp_database_name=None):
     global temporary_database_name
     global user_specified_connection
     global python_packages_installed
+    global python_version_local
+    global python_version_vantage
     if td_connection is not None:
         # Clearing the internal buffer.
         _InternalBuffer.clean()
@@ -654,10 +903,12 @@ def set_context(tdsqlengine, temp_database_name=None):
     _load_function_aliases()
 
     python_packages_installed = False
+    python_version_vantage = None
+    python_version_local = sys.version.split(" ")[0].strip()
 
     # Initialise Dag
     __initalise_dag()
-    
+
     # Add global lock to internal buffer
     _InternalBuffer.add(global_lock=threading.Lock())
 
@@ -687,11 +938,16 @@ def remove_context():
     global td_sqlalchemy_engine
     global user_specified_connection
     global python_packages_installed
+    global python_version_vantage
+    global python_version_local
     global td_user
 
     # Initiate the garbage collection
     __cleanup_garbage_collection()
 
+    # Clearing the internal buffer.
+    _InternalBuffer.clean()
+
     # Check if connection is established or not.
     if user_specified_connection is not True:
         try:
@@ -706,10 +962,13 @@ def remove_context():
     td_connection = None
     td_sqlalchemy_engine = None
     python_packages_installed = False
+    python_version_local = None
+    python_version_vantage = None
     td_user = None
     configure._current_database_name = None
     configure._database_username = None
     configure.database_version = None
+    configure.indb_install_location = ''
 
     # Closing Dag
     __close_dag()
@@ -993,6 +1252,7 @@ def _get_user():
         td_user = _get_database_username()
     return td_user
 
+
 def _get_host():
     """
     DESCRIPTION:
@@ -1015,6 +1275,7 @@ def _get_host():
     else:
         return td_sqlalchemy_engine.url.host
 
+
 def _get_host_ip():
     """
     DESCRIPTION:
@@ -1035,7 +1296,7 @@ def _get_host_ip():
     # Return None if connection is not established.
     if td_connection is None:
         return None
-    
+
     host = _get_host()
     try:
         # Validate if host_ip is a valid IP address (IPv4 or IPv6)
@@ -1046,7 +1307,7 @@ def _get_host_ip():
         dns_host_ip = _InternalBuffer.get('dns_host_ip')
         if dns_host_ip:
             return dns_host_ip
-        
+
         # If DNS host ip not found, resolve the host name to get the IP address.
         # If there is issue in resolving the host name, it will proceed with DNS host as it is.
         try:
@@ -1061,6 +1322,7 @@ def _get_host_ip():
             host_ip = host
     return host_ip
 
+
 class ContextUtilFuncs():
     @staticmethod
     def _check_alias_config_file_exists(vantage_version, alias_config_file):