Switch to MariaDB connector and Create UserManager Class

lib/ConfigLoader.py

@@ -2,15 +2,26 @@
 import toml
 from platformdirs import user_config_dir
 from pathlib import Path
-from PySide6.QtCore import QObject, Slot
+from PySide6.QtCore import QObject, Slot, Signal
 from .Vermasseln import Vermasseln
 import shutil
 from urllib.parse import urlparse
 from .DB.DbManager import DbManager
 import os
+from Crypto.Random import get_random_bytes
+from base64 import b64encode, b64decode
+from .DB.UserManager import UserManager
+from .PyqcrmFlags import PyqcrmFlags
+
+
 
 class ConfigLoader(QObject):
     __config = None
+    __version = "0.1-alpha"
+
+
+    dbConnectionError = Signal(str, bool)
+    adminUserError = Signal(str, bool)
 
     def __init__(self):
         super().__init__()
@@ -33,56 +44,69 @@ class ConfigLoader(QObject):
         shutil.copyfile(confile, self.config_dir+ '/pyqcrm.toml')
 
     @Slot(dict, result= bool)
-    def addAdminUser(self, admin_config):
-        print(admin_config)
+    def addAdminUser(self, user_config):
+        admin = UserManager(user_config["user"], PyqcrmFlags.ADMIN).createUser()
         return True
 
     @Slot(dict, result= bool)
     def setConfig(self, app_config):
-
+        base_conf = self.__initializeConfig()
         conf = self.__checkDbConnection(app_config)
-        if  conf:
-            try:
-                with open (self.config_dir + '/pyqcrm.toml', 'w') as f:
-                    config = Vermasseln().oscarVermasseln(toml.dumps(app_config))
-                    f.write(config)
-            except FileNotFoundError:
-                conf = False
-                print("Konnte die Konfiguration nicht speichern.")
-            conf = self.__checkAdminUser()
+        app_config = toml.dumps(app_config)
 
-        return conf
+        if  conf:
+            app_config = base_conf + app_config
+            self.__config = toml.loads(app_config)
+            self.__saveConfig()
+            conf = self.__checkAdminUser()
 
 
     def __configLoad(self):
         try:
             with open (self.config_dir + '/pyqcrm.toml', 'r') as f:
                 config = f.read()
-
                 self.__config = toml.loads(Vermasseln().entschluesseln(config))
         except FileNotFoundError:
             print("Konnte die Konfiguration nicht laden.")
         except TypeError:
-            print("Invalid Configuration")
-        # except Exception as e:
-        #     print(str(e))
+            print(f"Invalid Configuration: {__file__}")
+        except Exception as e:
+            print(str(e))
 
 
     def getConfig(self):
-
         return self.__config
 
-    def __createConfig(self):
+    def __initializeConfig(self):
+        encrypt_key = b64encode(get_random_bytes(32)).decode("utf-8")
+        conf = f"[pyqcrm]\nVERSION = \"{self.__version}\"\nENCRYPTION_KEY = \"{encrypt_key}\"\n\n"
+        return conf
 
-        with open(self.config_dir + '/pyqcrm.toml', "w") as datei:
-            datei.write("[pyqcrm]")
 
     def __checkDbConnection(self, db_config):
         con = DbManager(db_config['database']).getConnection()
         if con:
+            self.dbConnectionError.emit("Connection OK", True)
             return True
         else:
+            self.dbConnectionError.emit("Connection fehlgeschlagen", False)
             return False
 
     def __checkAdminUser(self):
-        pass
+
+        self.adminUserError.emit("Kein Admin vorhanden", False)
+
+    @Slot(str)
+    def setEncyrptKey(self, key):
+        self.__config['pyqcrm']['ENCRYPTION_KEY'] = key
+        self.__saveConfig()
+
+    def __saveConfig(self):
+        try:
+            with open (self.config_dir + '/pyqcrm.toml', 'w') as f:
+                config = Vermasseln().oscarVermasseln(toml.dumps(self.__config))
+                f.write(config)
+        except FileNotFoundError:
+            print("Konnte die Konfiguration nicht speichern.")
+
+