import os
import pymssql


class DatabaseConnection:
    def __init__(self, server, database, username, password):
        self.connection = pymssql.connect(server, username, password, database)

    def insert_data(self, table_name, data_values):
        columnas = ('img_size', 'intersect_over_union', 'probabilidad', 'user_text', 'feedback_text', 'check_status')
        insert_query = f"INSERT INTO {table_name} ({','.join([columna for columna in columnas])}) VALUES ({','.join(['%s' for _ in data_values])})"
        print(insert_query)
        cursor = self.connection.cursor()
        cursor.execute(insert_query, data_values)
        self.connection.commit()
        cursor.close()

server = os.environ.get('SERVER') # Reemplaza con el nombre de tu servidor
database = os.environ.get('DATABASE') # Reemplaza con el nombre de tu base de datos
username = os.environ.get('USERNAME') # Reemplaza con tu nombre de usuario
password = os.environ.get('PASSWORD') # Reemplaza con tu contraseña

db_connection = DatabaseConnection(server, database, username, password)