From 9f020c0ee56316f75e98a113f5dfbfa451a2762e Mon Sep 17 00:00:00 2001
From: George Lacey <git@george.ooo>
Date: Tue, 4 May 2021 12:05:10 +0100
Subject: [PATCH] Add get all methods

---
 src/database/borgdatabase.py                  | 11 ++++++
 src/database/connection/databaseconnection.py | 37 ++++++++++++++++---
 2 files changed, 43 insertions(+), 5 deletions(-)

diff --git a/src/database/borgdatabase.py b/src/database/borgdatabase.py
index 7b004d6..54c385d 100644
--- a/src/database/borgdatabase.py
+++ b/src/database/borgdatabase.py
@@ -17,6 +17,8 @@ class BorgDatabase(object):
         self.error_conn = ErrorConn(db_path,
                                     table_name=self.error_name)
 
+    # region INSERT
+
     def insert_record(self, repo, archive, stats):
         repo_id = self.repo_conn.insert(repo)
         archive_id = self.archive_conn.insert(archive, repo_id)
@@ -24,3 +26,12 @@ class BorgDatabase(object):
 
     def insert_error(self, borg_error):
         self.error_conn.insert(borg_error)
+
+    # endregion
+
+    # region GET
+
+    def get_repos(self):
+        return self.repo_conn.get_all()
+
+    # endregion
diff --git a/src/database/connection/databaseconnection.py b/src/database/connection/databaseconnection.py
index 15e0ede..cd48d60 100644
--- a/src/database/connection/databaseconnection.py
+++ b/src/database/connection/databaseconnection.py
@@ -16,6 +16,16 @@ class DatabaseConnection(ABC):
         self._create_table()
         self.sql_commit()
 
+    # region INIT
+
+    @abstractmethod
+    def _create_table(self):
+        raise NotImplementedError
+
+    # endregion
+
+    # region PROPERTIES
+
     @property
     def sql_lock(self):
         return self.__sql_lock
@@ -24,6 +34,13 @@ class DatabaseConnection(ABC):
     def sql_cursor(self):
         return self.__sql_database.cursor()
 
+    # endregion
+
+    # region GENERIC SQL
+
+    def sql_commit(self):
+        self.__sql_database.commit()
+
     def sql_execute(self, statement: str, args: tuple = None):
         with self.__sql_lock:
             cursor = self.sql_cursor
@@ -63,8 +80,9 @@ class DatabaseConnection(ABC):
 
             return cursor.fetchone()
 
-    def sql_commit(self):
-        self.__sql_database.commit()
+    # endregion
+
+    # region MODIFICATION
 
     def insert(self, record, repo_id=None, archive_id=None):
         exists, primary_key = self.exists(record)
@@ -97,9 +115,18 @@ class DatabaseConnection(ABC):
     def _exists(self, record) -> (str, tuple):
         raise NotImplementedError
 
-    @abstractmethod
-    def _create_table(self):
-        raise NotImplementedError
+    # endregion
+
+    # region QUERIES
+
+    def get_all(self):
+        result = self.sql_execute_one(f"SELECT * FROM {self._sql_table};")
+        if result is None:
+            return None
+        else:
+            return result[0]
+
+    # endregion
 
     def stop(self):
         with self.__sql_lock: