"""Optimizer for DB"""

class OptimizerDB:
    def __init__(self, db_config):
        self.config = db_config
        self.connection = None

    def connect(self):
        import sqlite3
        self.connection = sqlite3.connect(self.config['database'])
        print("Connected to the database")

    def disconnect(self):
        if self.connection:
            self.connection.close()
            print("Disconnected from the database")

    def execute_query(self, query, params=None):
        cursor = self.connection.cursor()
        if params:
            cursor.execute(query, params)
        else:
            cursor.execute(query)
        self.connection.commit()
        return cursor.fetchall()

    def get_optimizer_data(self, optimizer_id):
        query = "SELECT * FROM optimizers