WMGZON/controllers/database/database.py

from abc import ABC, abstractmethod
from typing import Mapping, Any
import sqlite3
import os


class DatabaseController(ABC):
    """ Abstract Base Class to handle database access for each component
        in the web app
    """

    __data_dir = "./data/"
    __db_name = "wmgzon.db"

    # Use test file if necessary
    if os.environ.get("ENVIRON") == "test":
        __db_name = "test_" + __db_name

    __sqlitefile = __data_dir + __db_name

    def __init__(self):
        """ Initialises the object and creates a connection to the local
            DB on the server
        """
        self._conn = None
        try:
            # Creates a connection and specifies a flag to parse all types
            # back down into Python declared types e.g. date & time
            self._conn = sqlite3.connect(
                self.__sqlitefile, detect_types=sqlite3.PARSE_DECLTYPES)
        except sqlite3.Error as e:
            # Close the connection if still open
            if self._conn:
                self._conn.close()
            print(e)

    def __del__(self):
        """ Object Destructor which kills the connection to the database """
        if self._conn is not None:
            self._conn.close()

    def new_instance(self, of: type, with_fields: Mapping[str, Any]):
        """ Takes a dictionary of fields and returns the object
            with those fields populated
        """
        obj = of.__new__(of)
        for attr, value in with_fields.items():
            setattr(obj, attr, value)
        return obj

    """
        Set of CRUD methods to allow for Data manipulation on the backend
    """

    @abstractmethod
    def create(self):
        """ Abstract method used to create a new record of a given
            type within the database
        """
        pass

    @abstractmethod
    def read(self):
        """ Abstract method used to read a record of a given
            type from the database
        """
        pass

    @abstractmethod
    def update(self):
        """ Abstract method used to update a record of a given
            type within the database
        """
        pass

    @abstractmethod
    def delete(self):
        """ Abstract method used to delete record of a given
            type from the database
        """
        pass