AddCustomer, AddContact finished

lib/DB/BusinessModel.py

@@ -2,6 +2,7 @@
 from PySide6.QtCore import QAbstractTableModel, QModelIndex, Qt, Slot, Signal
 from .BusinessDAO import BusinessDAO
 from ..PyqcrmFlags import PyqcrmFlags
+from ..ConfigLoader import ConfigLoader
 
 # USERS TABLE
 # CUSTOMER_COLUMN_NAMES = \
@@ -66,11 +67,15 @@ class BusinessModel(QAbstractTableModel):
 
     def __init__(self):
         super().__init__()
+        self.__key = ConfigLoader().getKey()
         self.__getData()
 
+
     def __getData(self, criterion = "Alle"):
         self.beginResetModel()
-        rows, self.__visible_columns = BusinessDAO().getBusiness(criterion)
+        rows, self.__visible_columns = BusinessDAO().getBusiness(self.__key, criterion)
+
+        print(rows)
         self.__data = rows
         self.endResetModel()
 
@@ -78,17 +83,21 @@ class BusinessModel(QAbstractTableModel):
         return len (self.__data)
 
     def columnCount(self, parent= QModelIndex()):
-        return len(self.__visible_columns) - 1
+        #return len(self.__visible_columns) - 1
+        return len(self.__visible_columns)
+
 
     def data(self, index, role= Qt.DisplayRole):
         if role == Qt.DisplayRole:
             row = self.__data[index.row()]
-            return row[index.column() + 1]
+            #return row[index.column() + 1]
+            return row[index.column()]
         return None
 
     def headerData(self, section, orientation, role= Qt.DisplayRole):
         if orientation == Qt.Horizontal and role ==Qt.DisplayRole:
-            self.__col_name = self.__visible_columns[section + 1]
+            #self.__col_name = self.__visible_columns[section + 1]
+            self.__col_name = self.__visible_columns[section]
             return self.__col_name
         return super().headerData(section, orientation, role)
 
@@ -116,7 +125,7 @@ class BusinessModel(QAbstractTableModel):
 
     @Slot(dict, int)
     def addBusiness(self, business, contact_id):
-        BusinessDAO().addBusiness(business, contact_id)
+        BusinessDAO().addBusiness(self.__key, business, contact_id)
         self.__getData()
 
     @Slot(dict)