teradataml 17.20.0.7__py3-none-any.whl → 20.0.0.0__py3-none-any.whl

teradataml/context/context.py

@@ -1,999 +1,1071 @@
-# -*- coding: utf-8 -*-
-"""
-Unpublished work.
-Copyright (c) 2018 by Teradata Corporation. All rights reserved.
-TERADATA CORPORATION CONFIDENTIAL AND TRADE SECRET
-
-Primary Owner: rameshchandra.d@teradata.com
-Secondary Owner:
-
-teradataml context
-----------
-A teradataml context functions provide interface to Teradata Vantage. Provides functionality to get and set a global
-context which can be used by other analytical functions to get the Teradata Vantage connection.
-
-"""
-from sqlalchemy import create_engine
-from teradataml.common.exceptions import TeradataMlException
-from teradataml.common.warnings import TeradataMlRuntimeWarning
-from teradataml.common.messages import Messages
-from teradataml.common.messagecodes import MessageCodes
-from teradataml.common.sqlbundle import SQLBundle
-from teradataml.common.constants import SQLConstants
-from teradataml.common.garbagecollector import GarbageCollector
-from teradataml.context.aed_context import AEDContext
-from teradataml.common.constants import TeradataConstants, Query
-from teradataml.common.utils import UtilFuncs
-from teradataml.options.configure import configure
-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
-import os
-import warnings
-import atexit
-import urllib.parse
-
-# Store a global Teradata Vantage Connection.
-# Right now user can only provide a single Vantage connection at any point of time.
-td_connection = None
-td_sqlalchemy_engine = None
-temporary_database_name = None
-user_specified_connection = False
-python_packages_installed = False
-td_user = None
-
-function_alias_mappings = {}
-
-# Current directory is context folder.
-teradataml_folder = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
-config_folder = os.path.join(teradataml_folder, "config")
-
-
-def _get_current_databasename():
-    """
-    Returns the database name associated with the current context.
-
-    PARAMETERS:
-        None.
-
-    RETURNS:
-        Database name associated with the current context
-
-    RAISES:
-        TeradataMlException - If Vantage connection can't be established using the engine.
-
-    EXAMPLES:
-        _get_current_databasename()
-    """
-    if get_connection() is not None:
-        select_user_query = ""
-        try:
-            sqlbundle = SQLBundle()
-            select_user_query = sqlbundle._get_sql_query(SQLConstants.SQL_SELECT_DATABASE)
-            result = execute_sql(select_user_query)
-            return result.fetchall()[0][0]
-        except TeradataMlException:
-            raise
-        except Exception as err:
-            raise TeradataMlException(Messages.get_message(MessageCodes.TDMLDF_EXEC_SQL_FAILED, select_user_query),
-                                      MessageCodes.TDMLDF_EXEC_SQL_FAILED) from err
-    else:
-        return None
-
-
-def _get_database_username():
-    """
-    Function to get the database user name.
-
-    PARAMETERS:
-        None.
-
-    RETURNS:
-        Database user name.
-
-    RAISES:
-        TeradataMlException - If "select user" query fails.
-
-    EXAMPLES:
-        _get_database_username()
-    """
-    if get_connection() is not None:
-        select_query = ""
-        try:
-            sqlbundle = SQLBundle()
-            select_query = sqlbundle._get_sql_query(SQLConstants.SQL_SELECT_USER)
-            result = execute_sql(select_query)
-            return result.fetchall()[0][0]
-        except TeradataMlException:
-            raise
-        except Exception as err:
-            raise TeradataMlException(Messages.get_message(MessageCodes.TDMLDF_EXEC_SQL_FAILED, select_query),
-                                      MessageCodes.TDMLDF_EXEC_SQL_FAILED) from err
-    else:
-        return None
-
-
-def __cleanup_garbage_collection():
-    """initiate the garbage collection."""
-    GarbageCollector._cleanup_garbage_collector()
-
-
-def _get_other_connection_parameters(logmech = None, logdata = None, database = None, **kwargs):
-    """
-    DESCRIPTION:
-        Function to generate and return the string for connection parameters.
-
-    PARAMETERS:
-        logmech:
-            Optional Argument.
-            Specifies the logon mechanism - TD2, LDAP, TDNEGO, KRB5 or JWT, to establish the connection.
-            Types: str
-
-        logdata:
-            Optional Argument.
-            Specifies additional connection information needed for the given logon mechanism.
-            Types: str
-
-        database:
-            Optional Argument.
-            Specifies the initial database to use after logon, instead of the user's default database.
-            Types: str
-
-        kwargs:
-            Optional Argument.
-            Specifies the keyword value pairs of other connection parameters to create the connection string.
-
-    RETURNS:
-        String containing connection parameters, needed to generate engine URL.
-
-    EXAMPLES:
-        __get_other_connection_parameters(logmech = "JWT", logdata = "<jwt_token>", database = "<database_name>",
-                                          kwargs)
-    """
-    # Return empty string if there are no additional connection parameters.
-    if not logmech and not logdata and not database and len(kwargs) == 0:
-        return ""
-
-    if logmech:
-        kwargs['LOGMECH'] = logmech.upper()
-    if logdata:
-        kwargs['LOGDATA'] = logdata
-    if database:
-        kwargs['DATABASE'] = database
-
-    # Create connection parameters string.
-    other_params = []
-    for key, val in kwargs.items():
-        if isinstance(val, str):
-            # Value of TMODE connection parameter should be upper case (as per driver specification) i.e., ansi -> ANSI.
-            # Converting all string values to upper case.
-            if key != "LOGDATA":
-                val = val.upper()
-        else:
-            # Other type values like integer, boolean etc, are converted to string.
-            # For boolean values, the connection string should contain lower case values i.e., True -> true
-            val = str(val).lower()
-        other_params.append("{}={}".format(key.upper(), val))
-
-    return "/?{}".format("&".join(other_params))
-
-
-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):
-    """
-    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.
-
-        Note:
-            1. teradataml requires that the user has certain permissions on the user's default database or the initial
-               default database specified using the database argument, or the temporary database when specified using
-               temp_database_name. These permissions allow the user to:
-                a. Create tables and views to save results of teradataml analytic functions.
-                b. Create views in the background for results of DataFrame APIs such as assign(),
-                   filter(), etc., whenever the result for these APIs are accessed using a print().
-                c. Create view in the background on the query passed to the DataFrame.from_query() API.
-
-               It is expected that the user has the correct permissions to create these objects in the database that
-               will be used.
-               The access to the views created may also require issuing additional GRANT SELECT ... WITH GRANT OPTION
-               permission depending on which database is used and which object the view being created is based on.
-
-            2. The temp_database_name and database parameters play a crucial role in determining which database
-               is used by default to lookup for tables/views while creating teradataml DataFrame using 'DataFrame()'
-               and 'DataFrame.from_table()' and which database is used to create all internal temporary objects.
-               +------------------------------------------------------+---------------------------------------------+
-               |                     Scenario                         |            teradataml behaviour             |
-               +------------------------------------------------------+---------------------------------------------+
-               | Both temp_database_name and database are provided    | Internal temporary objects are created in   |
-               |                                                      | temp_database_name, and database table/view |
-               |                                                      | lookup is done from database.               |
-               +------------------------------------------------------+---------------------------------------------+
-               | database is provided but temp_database_name is not   | Database table/view lookup and internal     |
-               |                                                      | temporary objects are created in database.  |
-               +------------------------------------------------------+---------------------------------------------+
-               | temp_database_name is provided but database is not   | Internal temporary objects are created in   |
-               |                                                      | temp_database_name, database table/view     |
-               |                                                      | lookup from the users default database.     |
-               +------------------------------------------------------+---------------------------------------------+
-               | Neither temp_database_name nor database are provided | Database table/view lookup and internal     |
-               |                                                      | temporary objects are created in users      |
-               |                                                      | default database.                           |
-               +------------------------------------------------------+---------------------------------------------+
-
-    PARAMETERS:
-        host:
-            Optional Argument.
-            Specifies the fully qualified domain name or IP address of the Teradata System.
-            Types: str
-        
-        username:
-            Optional Argument.
-            Specifies the username for logging onto the Teradata Vantage.
-            Types: str
-        
-        password:
-            Optional Argument.
-            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".
-                      
-            
-        tdsqlengine:
-            Optional Argument.
-            Specifies Teradata Vantage sqlalchemy engine object that should be used to establish a Teradata Vantage
-            connection.
-            Types: str
-            
-        temp_database_name:
-            Optional Argument.
-            Specifies the temporary database name where temporary tables, views will be created.
-            Types: str
-            
-        logmech:
-            Optional Argument.
-            Specifies the type of logon mechanism to establish a connection to Teradata Vantage. 
-            Permitted Values: 'TD2', 'TDNEGO', 'LDAP', 'KRB5' & 'JWT'.
-                TD2: 
-                    The Teradata 2 (TD2) mechanism provides authentication using a 
-                    Vantage username and password. This is the default logon mechanism
-                    using which the connection is established to Vantage.
-                TDNEGO:
-                    A security mechanism that automatically determines the actual 
-                    mechanism required, based on policy, without user's involvement.
-                    The actual mechanism is determined by the TDGSS server configuration
-                    and by the security policy's mechanism restrictions.
-                LDAP:
-                    A directory-based user logon to Vantage with a directory username
-                    and password and is authenticated by the directory.
-                KRB5 (Kerberos):
-                    A directory-based user logon to Vantage with a domain username
-                    and password and is authenticated by Kerberos (KRB5 mechanism).
-                    Note: User must have a valid ticket-granting ticket in order to use this logon mechanism.
-                JWT: 
-                    The JSON Web Token (JWT) authentication mechanism enables single 
-                    sign-on (SSO) to the Vantage after the user successfully authenticates
-                    to Teradata UDA User Service.
-                    Note: User must use logdata parameter when using 'JWT' as the logon mechanism.
-            Types: str
-
-            Note:
-                teradataml expects the client environments are already setup with appropriate
-                security mechanisms and are in working conditions.
-                For more information please refer Teradata Vantage™ - Advanced SQL Engine
-                Security Administration at https://www.info.teradata.com/
-        
-        logdata:
-            Optional Argument.
-            Specifies parameters to the LOGMECH command beyond those needed by the logon mechanism, such as 
-            user ID, password and tokens (in case of JWT) to successfully authenticate the user.
-            Types: str
-
-        database:
-            Optional Argument.
-            Specifies the initial database to use after logon, instead of the user's default database.
-            Types: str
-
-        url_encode:
-            Optional Argument.
-            Specifies whether the password should be URL encoded or not.
-            When set to True, passwords are URL encoded, otherwise not.
-            Default Value: True
-            Types: bool
-
-        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.
-
-    RETURNS:
-        A Teradata sqlalchemy engine object.
-
-    RAISES:
-        TeradataMlException
-
-    EXAMPLES:
-        >>> from teradataml.context.context import *
-
-        # Example 1: Create context using hostname, username and password
-        >>> td_context = create_context(host = 'tdhost', username='tduser', password = 'tdpassword')
-
-        # Example 2: Create context using already created sqlalchemy engine
-        >>> from sqlalchemy import create_engine
-        >>> sqlalchemy_engine  = create_engine('teradatasql://'+ tduser +':' + tdpassword + '@'+tdhost)
-        >>> 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')
-
-       # Example 4: Creating context for Vantage with logmech as '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')
-
-        # Example 6: Creating context for Vantage with logmech as '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')
-
-        # 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.
-        >>> encrypted_password = "ENCRYPTED_PASSWORD(file:PassKey.properties, file:EncPass.properties)"
-        >>> 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')
-
-       # 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')
-
-        # 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')
-        >>> 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')
-
-        # 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)
-
-        # Example 14: Create context when password has special characters.
-        # In this example password "alice@pass" is encoded by default.
-        >>> td_context = create_context(host = 'tdhost', username='alice_pass', password = 'alice@pass')
-            UserWarning: Warning: Password is URL encoded.
-
-        # Example 15: Create context when password contains space and special characters.
-        # "url_encode" must be set to False and special characters should be manually encoded in the password.
-        # For example, let's say password is: "kx%jj5/ g"
-        # then password should be manually encoded as: "kx%25jj5%2F g"
-        # where,
-        #     %25 represents the '%' character and
-        #     %2F represents the '/' character
-        #     space is not encoded.
-        >>> td_context = create_context(host = 'tdhost', username='alice_pass', password = 'kx%25jj5%2F g',
-                                        url_encode=False)
-
-    """
-    global td_connection
-    global td_sqlalchemy_engine
-    global temporary_database_name
-    global user_specified_connection
-    global python_packages_installed
-    global td_user
-    logmech_valid_values = ['TD2', 'TDNEGO', 'LDAP', 'KRB5', 'JWT']
-    awu_matrix = []
-    awu_matrix.append(["host", host, True, (str), True])
-    awu_matrix.append(["username", username, True, (str), True])
-    awu_matrix.append(["password", password, True, (str), True])
-    awu_matrix.append(["tdsqlengine", tdsqlengine, True, (Engine)])
-    awu_matrix.append(["logmech", logmech, True, (str), True, logmech_valid_values])
-    awu_matrix.append(["logdata", logdata, True, (str), True])
-    awu_matrix.append(["database", database, True, (str), True])
-    awu_matrix.append(["url_encode", url_encode, True, (bool), True])
-
-    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',
-                                                       'logmech=JWT'),
-                                  MessageCodes.DEPENDENT_ARG_MISSING)
-
-    # Setting the filter to raise warning every time.
-    warnings.simplefilter("always", TeradataMlRuntimeWarning)
-    # Throwing warning and removing context if any.
-    if td_connection is not None:
-        warnings.warn(Messages.get_message(MessageCodes.OVERWRITE_CONTEXT))
-        remove_context()
-
-    # Check if teradata sqlalchemy engine is provided by the user    
-    if tdsqlengine:
-        try:
-            td_connection = tdsqlengine.connect()
-            td_sqlalchemy_engine = tdsqlengine
-            user_specified_connection = True
-        except TeradataMlException:
-            raise
-        except Exception as err:
-            raise TeradataMlException(Messages.get_message(MessageCodes.CONNECTION_FAILURE),
-                                      MessageCodes.CONNECTION_FAILURE) from err
-    # Check if host and username and password are provided
-    elif host:
-        username = '' if username is None else username
-
-        # Encode the password if authentication is through username/password and url_encode is True.
-        if password and url_encode:
-            encoded_password = urllib.parse.quote_plus(password)
-            if encoded_password != password:
-                warnings.warn("Warning: Password is URL encoded.", category=TeradataMlRuntimeWarning)
-            password = encoded_password
-
-        if logmech and logmech.upper() == 'JWT':
-            host_value = host
-        elif logmech and logmech.upper() == 'KRB5':
-            host_value = '{}:@{}'.format(username, host)
-        else:
-            host_value = '{}:{}@{}'.format(username, password, host)
-
-        other_connection_parameters = _get_other_connection_parameters(logmech, logdata, database, **kwargs)
-
-        try:
-            engine_url = "teradatasql://{}{}"
-            td_sqlalchemy_engine = create_engine(engine_url.format(host_value, other_connection_parameters))
-            td_connection = td_sqlalchemy_engine.connect()
-            td_user = username.upper()
-
-            # Masking senstive information - password, logmech and logdata.
-            if password:
-                try:
-                    # Below statement raises an AttributeError with SQLAlchemy
-                    # version 1.4.x
-                    td_sqlalchemy_engine.url.password = "***"
-                except AttributeError:
-                    # Masking the password should be different from above as SQLAlchemy
-                    # converted _URL object to immutable object from version 1.4.x.
-                    new_url = td_sqlalchemy_engine.url.set(password="***")
-                    td_sqlalchemy_engine.url = new_url
-                except Exception:
-                    pass
-            _mask_logmech_logdata()
-
-            user_specified_connection = False
-
-        except TeradataMlException:
-            raise
-        except Exception as err:
-            raise TeradataMlException(Messages.get_message(MessageCodes.CONNECTION_FAILURE),
-                                      MessageCodes.CONNECTION_FAILURE) from err
-
-    # Load function aliases from config.
-    _load_function_aliases()
-
-    python_packages_installed = False
-
-    # Assign the tempdatabase name to global
-    if temp_database_name is None:
-        temporary_database_name = _get_current_databasename()
-    else:
-        temporary_database_name = temp_database_name
-
-    # Connection is established initiate the garbage collection
-    atexit.register(__cleanup_garbage_collection)
-    __cleanup_garbage_collection()
-    # Initialise Dag
-    __initalise_dag()
-
-    # Set database version.
-    configure.database_version = _get_database_version()
-
-    # Process Analytic functions.
-    from teradataml.analytics import _process_analytic_functions
-    _process_analytic_functions()
-
-    # Return the connection by default
-    return td_sqlalchemy_engine
-
-
-def _mask_logmech_logdata():
-    """
-    Masks sensitive connection information LOGMECH, LOGDATA exposed by sqlalchemy engine object
-    """
-    global td_sqlalchemy_engine
-    try:
-        # Below statement raises a TypeError with SQLAlchemy version 1.4.x
-        if ('LOGMECH' in td_sqlalchemy_engine.url.query):
-            td_sqlalchemy_engine.url.query['LOGMECH'] = "***"
-        if ('LOGDATA' in td_sqlalchemy_engine.url.query):
-            td_sqlalchemy_engine.url.query['LOGDATA'] = "***"
-    except TypeError:
-        # Masking the password should be different from above as SQLAlchemy
-        # converted _URL object to immutable object from version 1.4.x.
-        new_url = td_sqlalchemy_engine.url.update_query_dict({"LOGMECH": "***", "LOGDATA": "***"})
-        td_sqlalchemy_engine.url = new_url
-    except Exception:
-        pass
-
-
-def get_context():
-    """
-    DESCRIPTION:
-        Returns the Teradata Vantage connection associated with the current context.
-
-    PARAMETERS:
-        None
-
-    RETURNS:
-        A Teradata sqlalchemy engine object.
-
-    RAISES:
-        None.
-
-    EXAMPLES:
-        td_sqlalchemy_engine = get_context()
-        
-    """
-    global td_sqlalchemy_engine
-    return td_sqlalchemy_engine
-
-
-def get_connection():
-    """
-    DESCRIPTION:
-        Returns the Teradata Vantage connection associated with the current context.
-
-    PARAMETERS:
-        None
-
-    RETURNS:
-        A Teradata dbapi connection object.
-
-    RAISES:
-        None.
-
-    EXAMPLES:
-        tdconnection = get_connection()
-        
-    """
-    global td_connection
-    return td_connection
-
-
-def set_context(tdsqlengine, temp_database_name=None):
-    """
-    DESCRIPTION:
-        Specifies a Teradata Vantage sqlalchemy engine as current context.
-
-    PARAMETERS:
-        tdsqlengine:
-            Required Argument.
-            Specifies Teradata Vantage sqlalchemy engine object that should be used to establish a Teradata Vantage
-            connection.
-            Types: str
-            
-        temp_database_name:
-            Optional Argument.
-            Specifies the temporary database name where temporary tables, views will be created.
-            Types: str
-
-    RETURNS:
-        A Teradata Vantage connection object.
-
-    RAISES:
-        TeradataMlException
-
-    EXAMPLES:
-        set_context(tdsqlengine = td_sqlalchemy_engine)
-        
-    """
-    global td_connection
-    global td_sqlalchemy_engine
-    global temporary_database_name
-    global user_specified_connection
-    global python_packages_installed
-    if td_connection is not None:
-        warnings.warn(Messages.get_message(MessageCodes.OVERWRITE_CONTEXT))
-        remove_context()
-
-    if tdsqlengine:
-        try:
-            td_connection = tdsqlengine.connect()
-            td_sqlalchemy_engine = tdsqlengine
-            # Assign the tempdatabase name to global
-            if temp_database_name is None:
-                temporary_database_name = _get_current_databasename()
-            else:
-                temporary_database_name = temp_database_name
-
-            user_specified_connection = True
-        except TeradataMlException:
-            raise
-        except Exception as err:
-            raise TeradataMlException(Messages.get_message(MessageCodes.CONNECTION_FAILURE),
-                                      MessageCodes.CONNECTION_FAILURE) from err
-    else:
-        return None
-
-    # Load function aliases from config.
-    _load_function_aliases()
-
-    python_packages_installed = False
-
-    # Initialise Dag
-    __initalise_dag()
-
-    return td_connection
-
-
-def remove_context():
-    """
-    DESCRIPTION:
-        Removes the current context associated with the Teradata Vantage connection.
-
-    PARAMETERS:
-        None.
-
-    RETURNS:
-        None.
-
-    RAISES:
-        None.
-
-    EXAMPLES:
-        remove_context()
-        
-    """
-    global td_connection
-    global td_sqlalchemy_engine
-    global user_specified_connection
-    global python_packages_installed
-    global td_user
-
-    # Intiate the garbage collection
-    __cleanup_garbage_collection()
-
-    if user_specified_connection is not True:
-        try:
-            # Close the connection if not user specified connection.
-            td_connection.close()
-            td_sqlalchemy_engine.engine.dispose()
-        except TeradataMlException:
-            raise
-        except Exception as err:
-            raise TeradataMlException(Messages.get_message(MessageCodes.DISCONNECT_FAILURE),
-                                      MessageCodes.DISCONNECT_FAILURE) from err
-    td_connection = None
-    td_sqlalchemy_engine = None
-    python_packages_installed = False
-    td_user = None
-
-    # Closing Dag
-    __close_dag()
-    return True
-
-
-def _get_context_temp_databasename(table_type=TeradataConstants.TERADATA_VIEW):
-    """
-    Returns the temporary database name associated with the current context.
-
-    PARAMETERS:
-        table_type:
-            Optional Argument.
-            Specifies the type of object - table or view.
-            Default value: TeradataConstants.TERADATA_VIEW
-            Types: TeradataConstant
-
-    RETURNS:
-        Database name associated with the current context
-
-    RAISES:
-        None.
-
-    EXAMPLES:
-        _get_context_temp_databasename()
-        _get_context_temp_databasename(table_type=TeradataConstants.TERADATA_TABLE)
-    """
-    global temporary_database_name
-    if table_type == TeradataConstants.TERADATA_TABLE and \
-            configure.temp_table_database is not None:
-        return configure.temp_table_database
-    if table_type == TeradataConstants.TERADATA_VIEW and \
-            configure.temp_view_database is not None:
-        return configure.temp_view_database
-    return temporary_database_name
-
-
-def __initalise_dag():
-    """
-        Intialises the Dag
-
-        PARAMETERS:
-            None.
-
-        RETURNS:
-            None
-
-        RAISES:
-            None.
-
-        EXAMPLES:
-            __initalise_dag()
-    """
-    aed_context = AEDContext()
-    # Closing the Dag if previous instance is still exists.
-    __close_dag()
-    # TODO: Need to add logLevel and log_file functionlaity once AED is implemented these functionalities
-    aed_context._init_dag(_get_database_username(), _get_context_temp_databasename(),
-                          log_level=4, log_file="")
-
-
-def __close_dag():
-    """
-    Closes the Dag
-
-    PARAMETERS:
-        None.
-
-    RETURNS:
-        None
-
-    RAISES:
-        None.
-
-    EXAMPLES:
-        __close_dag()
-    """
-    try:
-        AEDContext()._close_dag()
-    # Ignore if any exception occurs.
-    except TeradataMlException:
-        pass
-
-
-def _load_function_aliases():
-    """
-    Function to load function aliases for analytical functions
-    based on the vantage version from configuration file.
-
-    PARAMETERS:
-        None
-
-    RETURNS:
-        None
-
-    RAISES:
-        TeradataMLException
-
-    EXAMPLES:
-        _load_function_aliases()
-    """
-
-    global function_alias_mappings
-    function_alias_mappings = {}
-
-    supported_engines = TeradataConstants.SUPPORTED_ENGINES.value
-    vantage_versions = TeradataConstants.SUPPORTED_VANTAGE_VERSIONS.value
-
-    __set_vantage_version()
-
-    for vv in vantage_versions.keys():
-        function_alias_mappings_by_engine = {}
-        for engine in supported_engines.keys():
-            alias_config_file = os.path.join(config_folder,
-                                             "{}_{}".format(supported_engines[engine]["file"], vantage_versions[vv]))
-            engine_name = supported_engines[engine]['name']
-            UtilFuncs._check_alias_config_file_exists(vv, alias_config_file)
-            function_alias_mappings_by_engine[engine_name] = \
-                UtilFuncs._get_function_mappings_from_config_file(alias_config_file)
-            function_alias_mappings[vv] = function_alias_mappings_by_engine
-
-
-def _get_vantage_version():
-    """
-    Function to determine the underlying Vantage version.
-
-    PARAMETERS:
-        None
-
-    RETURNS:
-        A string specifying the Vantage version, else None when not able to determine it.
-
-    RAISES:
-        Warning
-
-    EXAMPLES:
-        _get_vantage_version()
-    """
-    if td_connection.dialect.has_table(td_connection, "versionInfo", schema="pm"):
-
-        # BTEQ -- Enter your SQL request or BTEQ command:
-        # select * from pm.versionInfo;
-        #
-        # select * from pm.versionInfo;
-        #
-        # *** Query completed. 2 rows found. 2 columns returned.
-        # *** Total elapsed time was 1 second.
-        #
-        # InfoKey                        InfoData
-        # ------------------------------ --------------------------------------------
-        # BUILD_VERSION                  08.10.00.00-e84ce5f7
-        # RELEASE                        Vantage 1.1 GA
-
-        try:
-            vantage_ver_qry = "select InfoData from pm.versionInfo where InfoKey = 'RELEASE' (NOT CASESPECIFIC)"
-            res = execute_sql(vantage_ver_qry)
-            return res.fetchall()[0][0]
-        except:
-            return None
-    else:
-        # If "pm.versionInfo" does not exist, then vantage version is 1.0
-        return "vantage1.1"
-
-
-def _get_database_version():
-    """
-    DESCRIPTION:
-        An internal function to determine the underlying Vantage Database version.
-
-    PARAMETERS:
-        None
-
-    RETURNS:
-        A string specifying the Vantage Database version, else None when not able to determine it.
-
-    RAISES:
-        None
-
-    EXAMPLES:
-        _get_database_version()
-    """
-
-    # BTEQ -- Enter your SQL request or BTEQ command:
-    # select * from DBC.DBCInfoV;
-    # *** Query completed. 3 rows found. 2 columns returned.
-    # *** Total elapsed time was 1 second.
-    #
-    # InfoKey                        InfoData
-    # ------------------------------ --------------------------------------------
-    # VERSION                        17.05a.00.147
-    # LANGUAGE SUPPORT MODE          Standard
-    # RELEASE                        17.05a.00.147
-
-    try:
-        return execute_sql(Query.VANTAGE_VERSION.value).fetchall()[0][0]
-    except:
-        return None
-
-
-def __set_vantage_version():
-    """
-    Function to set the configuration option vantage_version.
-
-    PARAMETERS:
-        None
-
-    RETURNS:
-        None
-
-    RAISES:
-        TeradataMLException
-
-    EXAMPLES:
-        __set_vantage_version()
-    """
-    vantage_version = _get_vantage_version()
-    if vantage_version is None:
-        # Raise warning here.
-        warnings.warn(Messages.get_message(
-            MessageCodes.UNABLE_TO_GET_VANTAGE_VERSION).format("vantage_version", configure.vantage_version))
-    elif "vantage1.1" in vantage_version.lower().replace(" ", ""):
-        configure.vantage_version = "vantage1.1"
-    elif "mlengine9.0" in vantage_version.lower().replace(" ", ""):
-        configure.vantage_version = "vantage1.3"
-    elif "mlengine08.10" in vantage_version.lower().replace(" ", ""):
-        configure.vantage_version = "vantage2.0"
-    else:
-        # If "pm.versionInfo" does not exist, then vantage version is 1.0
-        configure.vantage_version = "vantage1.0"
-
-
-def _get_function_mappings():
-    """
-    Function to return function aliases for analytical functions.
-
-    PARAMETERS:
-        None
-
-    RETURNS:
-        Dict of function aliases of the format
-        {'mle' : {'func_name': "alias_name", ...},
-        'sqle' : {'func_name': "alias_name", ...}
-        ......
-        }
-
-    RAISES:
-        None
-
-    EXAMPLES:
-        get_function_aliases()
-    """
-    global function_alias_mappings
-    return function_alias_mappings
-
-
-def _get_user():
-    """
-    DESCRIPTION:
-        An internal function to get the database username associated with the current context.
-
-    PARAMETERS:
-        None.
-
-    RETURNS:
-        Database username associated with the current context.
-
-    RAISES:
-        TeradataMlException
-
-    EXAMPLES:
-        _get_user()
-    """
-    global td_user
-    if not td_user:
-        td_user = _get_database_username()
-    return td_user
+# -*- coding: utf-8 -*-
+"""
+Unpublished work.
+Copyright (c) 2018 by Teradata Corporation. All rights reserved.
+TERADATA CORPORATION CONFIDENTIAL AND TRADE SECRET
+
+Primary Owner: rameshchandra.d@teradata.com
+Secondary Owner:
+
+teradataml context
+----------
+A teradataml context functions provide interface to Teradata Vantage. Provides functionality to get and set a global
+context which can be used by other analytical functions to get the Teradata Vantage connection.
+
+"""
+from pathlib import Path
+from sqlalchemy import create_engine
+from teradatasqlalchemy.telemetry.queryband import collect_queryband
+from teradataml.common.deprecations import argument_deprecation
+from teradataml.common.exceptions import TeradataMlException
+from teradataml.common.warnings import TeradataMlRuntimeWarning
+from teradataml.common.messages import Messages
+from teradataml.common.messagecodes import MessageCodes
+from teradataml.common.sqlbundle import SQLBundle
+from teradataml.common.constants import SQLConstants
+from teradataml.common.garbagecollector import GarbageCollector
+from teradataml.context.aed_context import AEDContext
+from teradataml.common.constants import TeradataConstants, Query
+from teradataml.options.configure import configure
+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 urllib.parse
+
+# Store a global Teradata Vantage Connection.
+# Right now user can only provide a single Vantage connection at any point of time.
+td_connection = None
+td_sqlalchemy_engine = None
+temporary_database_name = None
+user_specified_connection = False
+python_packages_installed = False
+td_user = None
+
+function_alias_mappings = {}
+
+# Current directory is context folder.
+teradataml_folder = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+config_folder = os.path.join(teradataml_folder, "config")
+
+
+def _get_current_databasename():
+    """
+    Returns the database name associated with the current context.
+
+    PARAMETERS:
+        None.
+
+    RETURNS:
+        Database name associated with the current context
+
+    RAISES:
+        TeradataMlException - If Vantage connection can't be established using the engine.
+
+    EXAMPLES:
+        _get_current_databasename()
+    """
+    if get_connection() is not None:
+        select_user_query = ""
+        try:
+            sqlbundle = SQLBundle()
+            select_user_query = sqlbundle._get_sql_query(SQLConstants.SQL_SELECT_DATABASE)
+            result = execute_sql(select_user_query)
+            return result.fetchall()[0][0]
+        except TeradataMlException:
+            raise
+        except Exception as err:
+            raise TeradataMlException(Messages.get_message(MessageCodes.TDMLDF_EXEC_SQL_FAILED, select_user_query),
+                                      MessageCodes.TDMLDF_EXEC_SQL_FAILED) from err
+    else:
+        return None
+
+
+def _get_database_username():
+    """
+    Function to get the database user name.
+
+    PARAMETERS:
+        None.
+
+    RETURNS:
+        Database user name.
+
+    RAISES:
+        TeradataMlException - If "select user" query fails.
+
+    EXAMPLES:
+        _get_database_username()
+    """
+    if get_connection() is not None:
+        select_query = ""
+        try:
+            sqlbundle = SQLBundle()
+            select_query = sqlbundle._get_sql_query(SQLConstants.SQL_SELECT_USER)
+            result = execute_sql(select_query)
+            return result.fetchall()[0][0]
+        except TeradataMlException:
+            raise
+        except Exception as err:
+            raise TeradataMlException(Messages.get_message(MessageCodes.TDMLDF_EXEC_SQL_FAILED, select_query),
+                                      MessageCodes.TDMLDF_EXEC_SQL_FAILED) from err
+    else:
+        return None
+
+
+def __cleanup_garbage_collection():
+    """initiate the garbage collection."""
+    GarbageCollector._cleanup_garbage_collector()
+
+
+def _get_other_connection_parameters(logmech = None, logdata = None, database = None, **kwargs):
+    """
+    DESCRIPTION:
+        Internal function to return the connection parameters.
+
+    PARAMETERS:
+        logmech:
+            Optional Argument.
+            Specifies the logon mechanism - TD2, LDAP, TDNEGO, KRB5 or JWT, to establish the connection.
+            Types: str
+
+        logdata:
+            Optional Argument.
+            Specifies additional connection information needed for the given logon mechanism.
+            Types: str
+
+        database:
+            Optional Argument.
+            Specifies the initial database to use after logon, instead of the user's default database.
+            Types: str
+
+        kwargs:
+            Optional Argument.
+            Specifies the keyword value pairs of other connection parameters to create the connection string.
+
+    RETURNS:
+        dict, needed to generate engine URL.
+
+    EXAMPLES:
+        __get_other_connection_parameters(logmech = "JWT", logdata = "<jwt_token>", database = "<database_name>",
+                                          kwargs)
+    """
+    # Return empty string if there are no additional connection parameters.
+    if not logmech and not logdata and not database and len(kwargs) == 0:
+        return ""
+
+    result = {}
+
+    if logmech:
+        result['LOGMECH'] = logmech.upper()
+    if logdata:
+        result['LOGDATA'] = logdata
+    if database:
+        result['DATABASE'] = database
+
+    # Create connection parameters string.
+    other_params = []
+    for key, val in kwargs.items():
+        if isinstance(val, str):
+            # Value of TMODE connection parameter should be upper case (as per driver specification) i.e., ansi -> ANSI.
+            # Converting all string values to upper case.
+            if key != "LOGDATA":
+                val = val.upper()
+        else:
+            # Other type values like integer, boolean etc, are converted to string.
+            # For boolean values, the connection string should contain lower case values i.e., True -> true
+            val = str(val).lower()
+        result[key] = val
+
+    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):
+    """
+    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.
+
+        Note:
+            1. teradataml requires that the user has certain permissions on the user's default database or the initial
+               default database specified using the database argument, or the temporary database when specified using
+               temp_database_name. These permissions allow the user to:
+                a. Create tables and views to save results of teradataml analytic functions.
+                b. Create views in the background for results of DataFrame APIs such as assign(),
+                   filter(), etc., whenever the result for these APIs are accessed using a print().
+                c. Create view in the background on the query passed to the DataFrame.from_query() API.
+
+               It is expected that the user has the correct permissions to create these objects in the database that
+               will be used.
+               The access to the views created may also require issuing additional GRANT SELECT ... WITH GRANT OPTION
+               permission depending on which database is used and which object the view being created is based on.
+
+            2. The temp_database_name and database parameters play a crucial role in determining which database
+               is used by default to lookup for tables/views while creating teradataml DataFrame using 'DataFrame()'
+               and 'DataFrame.from_table()' and which database is used to create all internal temporary objects.
+               +------------------------------------------------------+---------------------------------------------+
+               |                     Scenario                         |            teradataml behaviour             |
+               +------------------------------------------------------+---------------------------------------------+
+               | Both temp_database_name and database are provided    | Internal temporary objects are created in   |
+               |                                                      | temp_database_name, and database table/view |
+               |                                                      | lookup is done from database.               |
+               +------------------------------------------------------+---------------------------------------------+
+               | database is provided but temp_database_name is not   | Database table/view lookup and internal     |
+               |                                                      | temporary objects are created in database.  |
+               +------------------------------------------------------+---------------------------------------------+
+               | temp_database_name is provided but database is not   | Internal temporary objects are created in   |
+               |                                                      | temp_database_name, database table/view     |
+               |                                                      | lookup from the users default database.     |
+               +------------------------------------------------------+---------------------------------------------+
+               | Neither temp_database_name nor database are provided | Database table/view lookup and internal     |
+               |                                                      | temporary objects are created in users      |
+               |                                                      | default database.                           |
+               +------------------------------------------------------+---------------------------------------------+
+
+    PARAMETERS:
+        host:
+            Optional Argument.
+            Specifies the fully qualified domain name or IP address of the Teradata System.
+            Types: str
+        
+        username:
+            Optional Argument.
+            Specifies the username for logging onto the Teradata Vantage.
+            Types: str
+        
+        password:
+            Optional Argument.
+            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".
+                      
+            
+        tdsqlengine:
+            Optional Argument.
+            Specifies Teradata Vantage sqlalchemy engine object that should be used to establish a Teradata Vantage
+            connection.
+            Types: str
+            
+        temp_database_name:
+            Optional Argument.
+            Specifies the temporary database name where temporary tables, views will be created.
+            Types: str
+            
+        logmech:
+            Optional Argument.
+            Specifies the type of logon mechanism to establish a connection to Teradata Vantage. 
+            Permitted Values: As supported by the teradata driver.
+            Notes:
+                1. teradataml expects the client environments are already setup with appropriate
+                   security mechanisms and are in working conditions.
+                2. User must have a valid ticket-granting ticket in order to use KRB5 (Kerberos) logon mechanism.
+                3. User must use logdata parameter when using 'JWT' as the logon mechanism.
+                4. Browser Authentication is supported for Windows and macOS.
+                For more information please refer Teradata Vantage™ - Advanced SQL Engine
+                Security Administration at https://www.info.teradata.com/
+            Types: str
+
+        logdata:
+            Optional Argument.
+            Specifies parameters to the LOGMECH command beyond those needed by the logon mechanism, such as 
+            user ID, password and tokens (in case of JWT) to successfully authenticate the user.
+            Types: str
+
+        database:
+            Optional Argument.
+            Specifies the initial database to use after logon, instead of the user's default database.
+            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.
+
+    RETURNS:
+        A Teradata sqlalchemy engine object.
+
+    RAISES:
+        TeradataMlException
+
+    EXAMPLES:
+        >>> from teradataml.context.context import *
+
+        # Example 1: Create context using hostname, username and password
+        >>> td_context = create_context(host = 'tdhost', username='tduser', password = 'tdpassword')
+
+        # Example 2: Create context using already created sqlalchemy engine
+        >>> from sqlalchemy import create_engine
+        >>> sqlalchemy_engine  = create_engine('teradatasql://'+ tduser +':' + tdpassword + '@'+tdhost)
+        >>> 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')
+
+       # Example 4: Creating context for Vantage with logmech as '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')
+
+        # Example 6: Creating context for Vantage with logmech as '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')
+
+        # 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.
+        >>> encrypted_password = "ENCRYPTED_PASSWORD(file:PassKey.properties, file:EncPass.properties)"
+        >>> 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')
+
+       # 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')
+
+        # 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')
+        >>> 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')
+
+        # 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)
+
+    """
+    global td_connection
+    global td_sqlalchemy_engine
+    global temporary_database_name
+    global user_specified_connection
+    global python_packages_installed
+    global td_user
+    awu_matrix = []
+    awu_matrix.append(["host", host, True, (str), True])
+    awu_matrix.append(["username", username, True, (str), True])
+    awu_matrix.append(["password", password, True, (str), True])
+    awu_matrix.append(["tdsqlengine", tdsqlengine, True, (Engine)])
+    awu_matrix.append(["logmech", logmech, True, (str), True])
+    awu_matrix.append(["logdata", logdata, True, (str), True])
+    awu_matrix.append(["database", database, True, (str), True])
+
+    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',
+                                                       'logmech=JWT'),
+                                  MessageCodes.DEPENDENT_ARG_MISSING)
+
+    # Setting the filter to raise warning every time.
+    warnings.simplefilter("always", TeradataMlRuntimeWarning)
+    # Throwing warning and removing context if any.
+    if td_connection is not None:
+        warnings.warn(Messages.get_message(MessageCodes.OVERWRITE_CONTEXT), stacklevel=2)
+        remove_context()
+
+    # Check if teradata sqlalchemy engine is provided by the user    
+    if tdsqlengine:
+        try:
+            td_connection = tdsqlengine.connect()
+            td_sqlalchemy_engine = tdsqlengine
+            user_specified_connection = True
+        except TeradataMlException:
+            raise
+        except Exception as err:
+            raise TeradataMlException(Messages.get_message(MessageCodes.CONNECTION_FAILURE),
+                                      MessageCodes.CONNECTION_FAILURE) from err
+    # Check if host and username and password are provided
+    elif host:
+        username = '' if username is None else username
+
+        if logmech and logmech.upper() in ['JWT', 'BROWSER']:
+            host_value = host
+        elif logmech and logmech.upper() == 'KRB5':
+            host_value = '{}:@{}'.format(username, host)
+        else:
+            host_value = '{}:{}@{}'.format(username, password, host)
+
+        url_object = URL.create(
+            "teradatasql",
+            username=username,
+            password=password,  # plain (unescaped) text
+            host=host,
+            query=_get_other_connection_parameters(logmech, logdata, database, **kwargs)
+        )
+
+        try:
+            td_sqlalchemy_engine = create_engine(url_object)
+            td_connection = td_sqlalchemy_engine.connect()
+            td_user = username.upper()
+
+            # Masking senstive information - password, logmech and logdata.
+            if password:
+                try:
+                    # Below statement raises an AttributeError with SQLAlchemy
+                    # version 1.4.x
+                    td_sqlalchemy_engine.url.password = "***"
+                except AttributeError:
+                    # Masking the password should be different from above as SQLAlchemy
+                    # converted _URL object to immutable object from version 1.4.x.
+                    new_url = td_sqlalchemy_engine.url.set(password="***")
+                    td_sqlalchemy_engine.url = new_url
+                except Exception:
+                    pass
+            _mask_logmech_logdata()
+
+            user_specified_connection = False
+
+        except TeradataMlException:
+            raise
+        except Exception as err:
+            raise TeradataMlException(Messages.get_message(MessageCodes.CONNECTION_FAILURE),
+                                      MessageCodes.CONNECTION_FAILURE) from err
+
+    # Load function aliases from config.
+    _load_function_aliases()
+
+    python_packages_installed = False
+
+    # Assign the tempdatabase name to global
+    if temp_database_name is None:
+        temporary_database_name = _get_current_databasename()
+    else:
+        temporary_database_name = temp_database_name
+
+    # Connection is established initiate the garbage collection
+    atexit.register(__cleanup_garbage_collection)
+    __cleanup_garbage_collection()
+    # Initialise Dag
+    __initalise_dag()
+
+    # Set database version.
+    configure.database_version = _get_database_version()
+
+    # Process Analytic functions.
+    from teradataml.analytics import _process_analytic_functions
+    _process_analytic_functions()
+
+    # Return the connection by default
+    return td_sqlalchemy_engine
+
+
+def _mask_logmech_logdata():
+    """
+    Masks sensitive connection information LOGMECH, LOGDATA exposed by sqlalchemy engine object
+    """
+    global td_sqlalchemy_engine
+    try:
+        # Below statement raises a TypeError with SQLAlchemy version 1.4.x
+        if ('LOGMECH' in td_sqlalchemy_engine.url.query):
+            td_sqlalchemy_engine.url.query['LOGMECH'] = "***"
+        if ('LOGDATA' in td_sqlalchemy_engine.url.query):
+            td_sqlalchemy_engine.url.query['LOGDATA'] = "***"
+    except TypeError:
+        # Masking the password should be different from above as SQLAlchemy
+        # converted _URL object to immutable object from version 1.4.x.
+        new_url = td_sqlalchemy_engine.url.update_query_dict({"LOGMECH": "***", "LOGDATA": "***"})
+        td_sqlalchemy_engine.url = new_url
+    except Exception:
+        pass
+
+
+def get_context():
+    """
+    DESCRIPTION:
+        Returns the Teradata Vantage connection associated with the current context.
+
+    PARAMETERS:
+        None
+
+    RETURNS:
+        A Teradata sqlalchemy engine object.
+
+    RAISES:
+        None.
+
+    EXAMPLES:
+        td_sqlalchemy_engine = get_context()
+        
+    """
+    global td_sqlalchemy_engine
+    return td_sqlalchemy_engine
+
+
+def get_connection():
+    """
+    DESCRIPTION:
+        Returns the Teradata Vantage connection associated with the current context.
+
+    PARAMETERS:
+        None
+
+    RETURNS:
+        A Teradata dbapi connection object.
+
+    RAISES:
+        None.
+
+    EXAMPLES:
+        tdconnection = get_connection()
+        
+    """
+    global td_connection
+    return td_connection
+
+
+@collect_queryband(queryband='SetCxt')
+def set_context(tdsqlengine, temp_database_name=None):
+    """
+    DESCRIPTION:
+        Specifies a Teradata Vantage sqlalchemy engine as current context.
+
+    PARAMETERS:
+        tdsqlengine:
+            Required Argument.
+            Specifies Teradata Vantage sqlalchemy engine object that should be used to establish a Teradata Vantage
+            connection.
+            Types: str
+            
+        temp_database_name:
+            Optional Argument.
+            Specifies the temporary database name where temporary tables, views will be created.
+            Types: str
+
+    RETURNS:
+        A Teradata Vantage connection object.
+
+    RAISES:
+        TeradataMlException
+
+    EXAMPLES:
+        set_context(tdsqlengine = td_sqlalchemy_engine)
+        
+    """
+    global td_connection
+    global td_sqlalchemy_engine
+    global temporary_database_name
+    global user_specified_connection
+    global python_packages_installed
+    if td_connection is not None:
+        warnings.warn(Messages.get_message(MessageCodes.OVERWRITE_CONTEXT), stacklevel=2)
+        remove_context()
+
+    if tdsqlengine:
+        try:
+            td_connection = tdsqlengine.connect()
+            td_sqlalchemy_engine = tdsqlengine
+            # Assign the tempdatabase name to global
+            if temp_database_name is None:
+                temporary_database_name = _get_current_databasename()
+            else:
+                temporary_database_name = temp_database_name
+
+            user_specified_connection = True
+        except TeradataMlException:
+            raise
+        except Exception as err:
+            raise TeradataMlException(Messages.get_message(MessageCodes.CONNECTION_FAILURE),
+                                      MessageCodes.CONNECTION_FAILURE) from err
+    else:
+        return None
+
+    # Load function aliases from config.
+    _load_function_aliases()
+
+    python_packages_installed = False
+
+    # Initialise Dag
+    __initalise_dag()
+
+    return td_connection
+
+
+@collect_queryband(queryband='RmCxt')
+def remove_context():
+    """
+    DESCRIPTION:
+        Removes the current context associated with the Teradata Vantage connection.
+
+    PARAMETERS:
+        None.
+
+    RETURNS:
+        None.
+
+    RAISES:
+        None.
+
+    EXAMPLES:
+        remove_context()
+        
+    """
+    global td_connection
+    global td_sqlalchemy_engine
+    global user_specified_connection
+    global python_packages_installed
+    global td_user
+
+    # Intiate the garbage collection
+    __cleanup_garbage_collection()
+
+    if user_specified_connection is not True:
+        try:
+            # Close the connection if not user specified connection.
+            td_connection.close()
+            td_sqlalchemy_engine.engine.dispose()
+        except TeradataMlException:
+            raise
+        except Exception as err:
+            raise TeradataMlException(Messages.get_message(MessageCodes.DISCONNECT_FAILURE),
+                                      MessageCodes.DISCONNECT_FAILURE) from err
+    td_connection = None
+    td_sqlalchemy_engine = None
+    python_packages_installed = False
+    td_user = None
+
+    # Closing Dag
+    __close_dag()
+    return True
+
+
+def _get_context_temp_databasename(table_type=TeradataConstants.TERADATA_VIEW):
+    """
+    Returns the temporary database name associated with the current context.
+
+    PARAMETERS:
+        table_type:
+            Optional Argument.
+            Specifies the type of object - table or view.
+            Default value: TeradataConstants.TERADATA_VIEW
+            Types: TeradataConstant
+
+    RETURNS:
+        Database name associated with the current context
+
+    RAISES:
+        None.
+
+    EXAMPLES:
+        _get_context_temp_databasename()
+        _get_context_temp_databasename(table_type=TeradataConstants.TERADATA_TABLE)
+    """
+    global temporary_database_name
+    if table_type == TeradataConstants.TERADATA_TABLE and \
+            configure.temp_table_database is not None:
+        return configure.temp_table_database
+    if table_type == TeradataConstants.TERADATA_VIEW and \
+            configure.temp_view_database is not None:
+        return configure.temp_view_database
+    return temporary_database_name
+
+
+def __initalise_dag():
+    """
+        Intialises the Dag
+
+        PARAMETERS:
+            None.
+
+        RETURNS:
+            None
+
+        RAISES:
+            None.
+
+        EXAMPLES:
+            __initalise_dag()
+    """
+    aed_context = AEDContext()
+    # Closing the Dag if previous instance is still exists.
+    __close_dag()
+    # TODO: Need to add logLevel and log_file functionlaity once AED is implemented these functionalities
+    aed_context._init_dag(_get_database_username(), _get_context_temp_databasename(),
+                          log_level=4, log_file="")
+
+
+def __close_dag():
+    """
+    Closes the Dag
+
+    PARAMETERS:
+        None.
+
+    RETURNS:
+        None
+
+    RAISES:
+        None.
+
+    EXAMPLES:
+        __close_dag()
+    """
+    try:
+        AEDContext()._close_dag()
+    # Ignore if any exception occurs.
+    except TeradataMlException:
+        pass
+
+
+def _load_function_aliases():
+    """
+    Function to load function aliases for analytical functions
+    based on the vantage version from configuration file.
+
+    PARAMETERS:
+        None
+
+    RETURNS:
+        None
+
+    RAISES:
+        TeradataMLException
+
+    EXAMPLES:
+        _load_function_aliases()
+    """
+
+    global function_alias_mappings
+    function_alias_mappings = {}
+
+    supported_engines = TeradataConstants.SUPPORTED_ENGINES.value
+    vantage_versions = TeradataConstants.SUPPORTED_VANTAGE_VERSIONS.value
+
+    __set_vantage_version()
+
+    for vv in vantage_versions.keys():
+        function_alias_mappings_by_engine = {}
+        for engine in supported_engines.keys():
+            alias_config_file = os.path.join(config_folder,
+                                             "{}_{}".format(supported_engines[engine]["file"], vantage_versions[vv]))
+            engine_name = supported_engines[engine]['name']
+            ContextUtilFuncs._check_alias_config_file_exists(vv, alias_config_file)
+            function_alias_mappings_by_engine[engine_name] = \
+                ContextUtilFuncs._get_function_mappings_from_config_file(alias_config_file)
+            function_alias_mappings[vv] = function_alias_mappings_by_engine
+
+
+def _get_vantage_version():
+    """
+    Function to determine the underlying Vantage version.
+
+    PARAMETERS:
+        None
+
+    RETURNS:
+        A string specifying the Vantage version, else None when not able to determine it.
+
+    RAISES:
+        Warning
+
+    EXAMPLES:
+        _get_vantage_version()
+    """
+    if td_connection.dialect.has_table(td_connection, "versionInfo", schema="pm"):
+
+        # BTEQ -- Enter your SQL request or BTEQ command:
+        # select * from pm.versionInfo;
+        #
+        # select * from pm.versionInfo;
+        #
+        # *** Query completed. 2 rows found. 2 columns returned.
+        # *** Total elapsed time was 1 second.
+        #
+        # InfoKey                        InfoData
+        # ------------------------------ --------------------------------------------
+        # BUILD_VERSION                  08.10.00.00-e84ce5f7
+        # RELEASE                        Vantage 1.1 GA
+
+        try:
+            vantage_ver_qry = "select InfoData from pm.versionInfo where InfoKey = 'RELEASE' (NOT CASESPECIFIC)"
+            res = execute_sql(vantage_ver_qry)
+            return res.fetchall()[0][0]
+        except:
+            return None
+    else:
+        # If "pm.versionInfo" does not exist, then vantage version is 1.0
+        return "vantage1.1"
+
+
+def _get_database_version():
+    """
+    DESCRIPTION:
+        An internal function to determine the underlying Vantage Database version.
+
+    PARAMETERS:
+        None
+
+    RETURNS:
+        A string specifying the Vantage Database version, else None when not able to determine it.
+
+    RAISES:
+        None
+
+    EXAMPLES:
+        _get_database_version()
+    """
+
+    # BTEQ -- Enter your SQL request or BTEQ command:
+    # select * from DBC.DBCInfoV;
+    # *** Query completed. 3 rows found. 2 columns returned.
+    # *** Total elapsed time was 1 second.
+    #
+    # InfoKey                        InfoData
+    # ------------------------------ --------------------------------------------
+    # VERSION                        17.05a.00.147
+    # LANGUAGE SUPPORT MODE          Standard
+    # RELEASE                        17.05a.00.147
+
+    try:
+        return execute_sql(Query.VANTAGE_VERSION.value).fetchall()[0][0]
+    except:
+        return None
+
+
+def __set_vantage_version():
+    """
+    Function to set the configuration option vantage_version.
+
+    PARAMETERS:
+        None
+
+    RETURNS:
+        None
+
+    RAISES:
+        TeradataMLException
+
+    EXAMPLES:
+        __set_vantage_version()
+    """
+    vantage_version = _get_vantage_version()
+    if vantage_version is None:
+        # Raise warning here.
+        warnings.warn(Messages.get_message(
+            MessageCodes.UNABLE_TO_GET_VANTAGE_VERSION).format("vantage_version", configure.vantage_version))
+    elif "vantage1.1" in vantage_version.lower().replace(" ", ""):
+        configure.vantage_version = "vantage1.1"
+    elif "mlengine9.0" in vantage_version.lower().replace(" ", ""):
+        configure.vantage_version = "vantage1.3"
+    elif "mlengine08.10" in vantage_version.lower().replace(" ", ""):
+        configure.vantage_version = "vantage2.0"
+    else:
+        # If "pm.versionInfo" does not exist, then vantage version is 1.0
+        configure.vantage_version = "vantage1.0"
+
+
+def _get_function_mappings():
+    """
+    Function to return function aliases for analytical functions.
+
+    PARAMETERS:
+        None
+
+    RETURNS:
+        Dict of function aliases of the format
+        {'mle' : {'func_name': "alias_name", ...},
+        'sqle' : {'func_name': "alias_name", ...}
+        ......
+        }
+
+    RAISES:
+        None
+
+    EXAMPLES:
+        get_function_aliases()
+    """
+    global function_alias_mappings
+    return function_alias_mappings
+
+
+def _get_user():
+    """
+    DESCRIPTION:
+        An internal function to get the database username associated with the current context.
+
+    PARAMETERS:
+        None.
+
+    RETURNS:
+        Database username associated with the current context.
+
+    RAISES:
+        TeradataMlException
+
+    EXAMPLES:
+        _get_user()
+    """
+    global td_user
+    if not td_user:
+        td_user = _get_database_username()
+    return td_user
+
+
+class ContextUtilFuncs():
+    @staticmethod
+    def _check_alias_config_file_exists(vantage_version, alias_config_file):
+        """
+        Function to validate whether alias_config_file exists for the current vantage version.
+
+        PARAMETERS:
+            vantage_version:
+                Required Argument.
+                Specifies the current vantage version.
+
+            alias_config_file:
+                Required Argument.
+                Specifies the location of configuration file to be read.
+
+        RETURNS:
+            True, if the file 'alias_config_file' is present in the
+            teradataml/config directory for the current vantage version.
+
+        RAISES:
+            TeradataMLException
+
+        EXAMPLES:
+            ContextUtilFuncs._check_alias_config_file_exists("vantage1.0", "config_file_location")
+
+        """
+        # Raise exception if alias config file is not defined.
+        if not Path(alias_config_file).exists():
+            raise TeradataMlException(Messages.get_message(
+                MessageCodes.CONFIG_ALIAS_CONFIG_FILE_NOT_FOUND).format(alias_config_file,
+                                                                        vantage_version),
+                                      MessageCodes.CONFIG_ALIAS_CONFIG_FILE_NOT_FOUND)
+        return True
+
+    @staticmethod
+    def _get_function_mappings_from_config_file(alias_config_file):
+        """
+        Function to return the function mappings given the location of configuration file in
+        argument 'alias_config_file'.
+
+        PARAMETERS:
+            alias_config_file:
+                Required Argument.
+                Specifies the location of configuration file to be read.
+
+        RETURNS:
+            Function mappings as a dictionary of function_names to alias_names.
+
+        RAISES:
+            TeradataMLException
+
+        EXAMPLES:
+            ContextUtilFuncs._get_function_mappings_from_config_file("config_file_location")
+
+        """
+        repeated_function_names = []
+        function_mappings = {}
+        invalid_function_mappings = []
+        invalid_function_mappings_line_nos = []
+        # Reading configuration files
+        with open(alias_config_file, 'r') as fread:
+            for line_no, line in enumerate(fread.readlines()):
+                line = line.strip()
+
+                # Ignoring empty lines in the config files.
+                if line == "":
+                    continue
+
+                # If the separator ":" is not present.
+                if ':' not in line:
+                    invalid_function_mappings.append(line)
+                    invalid_function_mappings_line_nos.append(str(line_no + 1))
+                else:
+                    func_name, alias_name = line.split(":")
+                    func_name = func_name.strip()
+                    alias_name = alias_name.strip()
+
+                    # First line of 'alias_config_file' has header "functionName:aliasName".
+                    if line_no == 0 and func_name == "functionName" and alias_name == "aliasName":
+                        continue
+
+                    if func_name == "" or alias_name == "":
+                        invalid_function_mappings.append(line)
+                        invalid_function_mappings_line_nos.append(str(line_no + 1))
+                        continue
+
+                    if func_name.lower() in function_mappings:
+                        repeated_function_names.append(func_name.lower())
+
+                    # Loading function maps with lower values for key.
+                    function_mappings[func_name.lower()] = alias_name
+
+        # Presence of Invalid function mappings in the 'alias_config_file'.
+        if len(invalid_function_mappings) > 0:
+            err_ = Messages.get_message(MessageCodes.CONFIG_ALIAS_INVALID_FUNC_MAPPING)
+            err_ = err_.format("', '".join(invalid_function_mappings),
+                               ", ".join(invalid_function_mappings_line_nos),
+                               alias_config_file)
+            raise TeradataMlException(err_, MessageCodes.CONFIG_ALIAS_INVALID_FUNC_MAPPING)
+
+        # Raising teradataml exception if there are any duplicates in function names.
+        if len(repeated_function_names) > 0:
+            raise TeradataMlException(Messages.get_message(
+                MessageCodes.CONFIG_ALIAS_DUPLICATES).format(alias_config_file,
+                                                             ", ".join(repeated_function_names)),
+                                      MessageCodes.CONFIG_ALIAS_DUPLICATES)
+
+        return function_mappings