inventory-thing/utils/sql_utils.py

38 lines
958 B
Python

import mysql.connector
from mysql.connector import Error
def connect_database(host, username, password, database):
connection = None
try:
connection = mysql.connector.connect(
host=host,
user=username,
passwd=password,
database=database
)
print("SQL connection established")
except Error as err:
print(f'Error: {err}')
return connection
def send_query(connection, query, parameters):
cursor = connection.cursor()
try:
cursor.execute(query, parameters)
connection.commit()
return True
except Error as err:
print(f'Error: {err}')
return False
def read_query(connection, query):
cursor = connection.cursor()
result = None
try:
cursor.execute(query)
result = cursor.fetchall()
connection.commit()
return result
except Error as err:
print(f'Error: {err}')