Function get_db_connection

web/regression/python_test_utils/test_utils.py:53–78

This function returns the connection object of psycopg

(db, username, password, host, port, sslmode="prefer",
                      max_connections=None)

def get_db_connection(db, username, password, host, port, sslmode="prefer",
                      max_connections=None):
    """This function returns the connection object of psycopg"""
    if max_connections:
        with psycopg.connect(
            dbname=db,
            user=username,
            password=password,
            host=host,
            port=port,
            sslmode=sslmode,
            autocommit=True,
        ) as conn:
            cur = conn.cursor()
            cur.execute('ALTER SYSTEM SET max_connections TO 100;')
            cur.execute('SELECT pg_reload_conf();')

    connection = psycopg.connect(
        dbname=db,
        user=username,
        password=password,
        host=host,
        port=port,
        sslmode=sslmode
    )
    return connection


def get_server_version(connection):