diff --git a/src/database_handler.py b/src/database_handler.py index 977bb8a..63c413c 100644 --- a/src/database_handler.py +++ b/src/database_handler.py @@ -1,10 +1,13 @@ -import logging as log import os import sqlite3 from enum import Enum +from logger import LoggerWrapper + DATABASE_PATH = os.path.join(os.path.dirname(os.path.abspath(__file__)), '..', 'data', 'spotify_scraped.db') +log = LoggerWrapper() + class Table(Enum): TRACK_INFORMATION = "track_information" @@ -82,6 +85,7 @@ class Database: # Commit the changes self.conn.commit() + log.debug("Initialised tables") def add_row(self, table: Table, values): """Add a new row into the specified table""" @@ -91,13 +95,17 @@ class Database: self.cursor.execute(query, values) self.conn.commit() except Exception as e: - log.debug(f"Error: {e}") + log.error(f"Error while inserting row into table {table.value}: {e}") def read_all_rows(self, table: Table, column: str = "*"): """Read all rows from the specified table""" - self.cursor.execute(f"SELECT {column} FROM {table.value}") - rows = self.cursor.fetchall() - return rows + try: + self.cursor.execute(f"SELECT {column} FROM {table.value}") + rows = self.cursor.fetchall() + return rows + except Exception as e: + log.error(f"Error while reading all rows from table {table.value}: {e}") + return [] def close(self): """Close the database connection""" @@ -125,5 +133,6 @@ class Database: rows = self.cursor.fetchall() return rows except Exception as e: - log.error(f"Error retrieving total overview: {e}") + log.error(f"Error retrieving total overview: {e}" + f"\nQuery Executed: {query}") return []