agent_2203

from mysql.connector import MySQLConnection, Error
from sys import exit

conn = {}

def connect(host, database, user, password):
    global conn
    try:
        conn = MySQLConnection(
            host=host,
            database = database,
            user = user,
            password = password
        )
        if conn.is_connected(): print('Connected to DB.')
    except Error as e:
        print('Failed to connect to DB. Exception caught:')
        print(e)
        exit()

def request(req, data=None):
    global conn
    cursor = conn.cursor(buffered=True)
    try:
        cursor.execute(req, data)
        conn.commit()
        data = cursor.fetchall()
        cursor.close()
        return data
    except Error as err:
        print(err)
        return []