aswi2021jmsd-gitlab4/src/impl/db_manager.py @ 993ee5a1

import sqlite3
from sqlite3 import Error

from ..constants import DATABASE_FILE
from ..dao.repository import IRepository


class DBManager(IRepository):

    @staticmethod
    def create(sql: str, *args) -> bool:
        conn = None
        try:
            sql: str = args[0]
            values = args[1:]

            conn = sqlite3.connect(DATABASE_FILE)
            c = conn.cursor()
            c.execute(sql, values)
            conn.commit()
            c.close()
        except Error as e:
            print(e)
            return False
        finally:
            if conn:
                conn.close()
        return True

    @staticmethod
    def read(sql: str, item_id: int = None):
        conn = None
        try:
            conn = sqlite3.connect(DATABASE_FILE)
            c = conn.cursor()

            if item_id is not None:
                list_item_id = [item_id]
                c.execute(sql, list_item_id)
            else:
                c.execute(sql)

            records = c.fetchall()
            c.close()
        except Error as e:
            print(e)
            return []
        finally:
            if conn:
                conn.close()
        return records

    @staticmethod
    def update(sql: str, *args) -> bool:
        conn = None
        try:
            conn = sqlite3.connect(DATABASE_FILE)
            c = conn.cursor()
            c.execute(sql, args)
            conn.commit()
            c.close()
        except Error as e:
            print(e)
            return False
        finally:
            if conn:
                conn.close()
        return True

    @staticmethod
    def delete(sql: str, item_id: int) -> bool:
        conn = None
        try:
            conn = sqlite3.connect(DATABASE_FILE)
            c = conn.cursor()
            c.execute(sql, (item_id,))
            conn.commit()
            c.close()
        except Error as e:
            print(e)
            return False
        finally:
            if conn:
                conn.close()
        return True