fsfwgen/utility/sql_writer.py

import sqlite3

from fsfwgen.core import get_console_logger

LOGGER = get_console_logger()


class SqlWriter:
    def __init__(self, db_filename: str):
        self.filename = db_filename
        self.conn = sqlite3.connect(self.filename)

    def open(self, sql_creation_command: str):
        LOGGER.info(f'SQL Writer: Opening {self.filename}')
        self.conn.execute(sql_creation_command)

    def delete(self, sql_deletion_command):
        LOGGER.info("SQL Writer: Deleting SQL table")
        self.conn.execute(sql_deletion_command)

    def write_entries(self, sql_insertion_command, current_entry):
        cur = self.conn.cursor()
        cur.execute(sql_insertion_command, current_entry)
        return cur.lastrowid

    def commit(self):
        LOGGER.info("SQL Writer: Commiting SQL table")
        self.conn.commit()

    def close(self):
        self.conn.close()

    def sql_writing_helper(
            self, creation_cmd, insertion_cmd, mib_table: dict, deletion_cmd: str = ""
    ):
        if deletion_cmd != "":
            self.delete(deletion_cmd)
        self.open(creation_cmd)
        for i in mib_table:
            self.write_entries(insertion_cmd, mib_table[i])
        self.commit()
        self.close()