added attribute conn

app_utils/conf.py

@@ -2,20 +2,11 @@ import sqlite3
 from pathlib import Path
 from typing import Any, Dict
 
-from pydantic import BaseModel
-
-class PersistentSettings(BaseModel):
+class PersistentSettings:
     """
-    This pydantic model will try to initialize itself from
-    the database upon every instantiation
-
-    It further supplies an update function, that allows to write
-    back any changes into the database, under its key.
+    This class manages the persistent settings and the database connection.
     """
 
-    class Config:
-        arbitrary_types_allowed = True  # Exclude conn from Pydantic validation
-
     def __init__(self, **data: Any):
         # Connect to the SQLite database
         self.conn = sqlite3.connect("config.db")
@@ -29,7 +20,9 @@ class PersistentSettings(BaseModel):
         """)
 
         # Fetch settings from the database and initialize
-        super().__init__(**self.fetch_settings(), **data)
+        self.settings = self.fetch_settings()
+        for key, value in data.items():
+            self.settings[key] = value
 
     def fetch_settings(self) -> Dict[str, Any]:
         """
@@ -43,12 +36,12 @@ class PersistentSettings(BaseModel):
 
     def update(self, **data: Any) -> None:
         """
-        Persist the pydantic-dict that represents the model
+        Persist the dictionary that represents the settings into the database
         """
         cursor = self.conn.cursor()
 
         # Update or insert each key-value pair into the database
-        for key, value in {**self.dict(), **data}.items():
+        for key, value in data.items():
             cursor.execute(
                 "INSERT OR REPLACE INTO settings (key, value) VALUES (?, ?)",
                 (key, value)
@@ -65,15 +58,11 @@ class PersistentSettings(BaseModel):
         self.conn.close()
 
 class TortoiseConfig(PersistentSettings):
-    EXTRA_VOICES_DIR: str = ""
-    AR_CHECKPOINT: str = "."
-    DIFF_CHECKPOINT: str = "."
-    LOW_VRAM: bool = True
-    conn: sqlite3.Connection = None
-
     def __init__(self, **data: Any):
         super().__init__(**data)
-        if not Path(self.AR_CHECKPOINT).is_file():
-            self.AR_CHECKPOINT = "."
-        if not Path(self.DIFF_CHECKPOINT).is_file():
-            self.DIFF_CHECKPOINT = "."
+        if not Path(self.settings.get("AR_CHECKPOINT", "")).is_file():
+            self.settings["AR_CHECKPOINT"] = "."
+        if not Path(self.settings.get("DIFF_CHECKPOINT", "")).is_file():
+            self.settings["DIFF_CHECKPOINT"] = "."
+        self.EXTRA_VOICES_DIR = self.settings.get("EXTRA_VOICES_DIR", "")
+        self.LOW_VRAM = self.settings.get("LOW_VRAM", True)