psycopg2 postgres python3 connect to database and execute query

db_helpers

import psycopg2.extras
import os

def connect_db(autocommit=False):
    """ Connects to a data base and returns a connection and a cursor. """
    conn = psycopg2.connect(
        database=os.environ.get('DATABASE_DBNAME'),
        user=os.environ.get('DATABASE_USERNAME'),
        host=os.environ.get('DATABASE_HOST'),
        password=os.environ.get('DATABASE_PASSWORD')
    )
    conn.autocommit = autocommit
    cursor = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
    return {'conn': conn, 'cursor': cursor}
    
    
    
def execute_query(a_query, autocommit=True):
    """ Executes a query, printing out rows returned if any, returning the results, the connection and the cursor. """
    db = connect_db(autocommit=autocommit)
    conn, cursor = db.get('conn'), db.get('cursor')
    cursor.execute(a_query)
    result = cursor.fetchall()
    if result:
        print(len(result), "records fetched")
    return {'result':result, 'conn':conn, 'cursor':cursor}
    
    
if if __name__ == '__main__':
    q = """ Select * from table limit 1 """
    res = execute_query(q)
    print(res)