query_handler.py

import logging
import json
import os
import sys
root_folder = os.path.abspath(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
sys.path.append(root_folder)
import db_connection

def exec_query_with_message(sql_query):
    message = None
    with db_connection.Db_Connection() as conn:
        logging.info("Connection established")
        with conn.cursor() as cursor:
            cursor.execute(sql_query)
            message = cursor.fetchone()[0]
            if message == None:
                return message

    return message

def exec_query_with_records(sql_query):
    records = []
    with db_connection.Db_Connection() as conn:
        logging.info("Connection established")
        with conn.cursor() as cursor:
            cursor.execute(sql_query)
            row = cursor.fetchone()
            while row:
                records.append(row)
                row = cursor.fetchone()

    return records



def exec_query_with_no_response(sql_query):
    message = None
    with db_connection.Db_Connection() as conn:
        logging.info("Connection established")
        with conn.cursor() as cursor:
            cursor.execute(sql_query)