#1816 - Added the final pieces of the puzzle to get data up from NIC → session manager → software manager → service.

- Implemented a basic sim DB that matches UC2 data manipulation DB in IY.
- Added a test that confirms DB queries can be sent over the network.

src/primaite/simulator/system/services/database.py

@@ -1,9 +1,9 @@
 import sqlite3
 from ipaddress import IPv4Address
 from sqlite3 import OperationalError
-from typing import Dict, Optional, Any, List, Union
+from typing import Any, Dict, List, Optional, Union
 
-from prettytable import PrettyTable, MARKDOWN
+from prettytable import MARKDOWN, PrettyTable
 
 from primaite.simulator.file_system.file_system import File
 from primaite.simulator.network.transmission.network_layer import IPProtocol
@@ -13,7 +13,12 @@ from primaite.simulator.system.services.service import Service
 
 
 class DatabaseService(Service):
-    """A generic SQL Server Service."""
+    """
+    A class for simulating a generic SQL Server service.
+
+    This class inherits from the `Service` class and provides methods to manage and query a SQLite database.
+    """
+
     backup_server: Optional[IPv4Address] = None
     "The IP Address of the server the "
 
@@ -28,12 +33,21 @@ class DatabaseService(Service):
         self._cursor = self._conn.cursor()
 
     def tables(self) -> List[str]:
+        """
+        Get a list of table names present in the database.
+
+        :return: List of table names.
+        """
         sql = "SELECT name FROM sqlite_master WHERE type='table' AND name != 'sqlite_sequence';"
         results = self._process_sql(sql)
         return [row[0] for row in results["data"]]
 
     def show(self, markdown: bool = False):
-        """Prints a Table names in the Database."""
+        """
+        Prints a list of table names in the database using PrettyTable.
+
+        :param markdown: Whether to output the table in Markdown format.
+        """
         table = PrettyTable(["Table"])
         if markdown:
             table.set_style(MARKDOWN)
@@ -44,10 +58,17 @@ class DatabaseService(Service):
         print(table)
 
     def _create_db_file(self):
+        """Creates the Simulation File and sqlite file in the file system."""
         self._db_file: File = self.file_system.create_file(folder_name="database", file_name="database.db", real=True)
         self.folder = self._db_file.folder
 
     def _process_sql(self, query: str) -> Dict[str, Union[int, List[Any]]]:
+        """
+        Executes the given SQL query and returns the result.
+
+        :param query: The SQL query to be executed.
+        :return: Dictionary containing status code and data fetched.
+        """
         try:
             self._cursor.execute(query)
             self._conn.commit()
@@ -69,11 +90,15 @@ class DatabaseService(Service):
         return super().describe_state()
 
     def receive(self, payload: Any, session_id: str, **kwargs) -> bool:
+        """
+        Processes the incoming SQL payload and sends the result back.
+
+        :param payload: The SQL query to be executed.
+        :param session_id: The session identifier.
+        :return: The status code of the SQL execution.
+        """
         result = self._process_sql(payload)
         software_manager: SoftwareManager = self.software_manager
         software_manager.send_payload_to_session_manager(payload=result, session_id=session_id)
 
-        return result["status_code"]
-
-    def send(self, payload: Any, session_id: str, **kwargs) -> bool:
-        pass
+        return result["status_code"] == 200