From 89dc1d3010220ce70487a77dc89692d681d60ac89c47c7aa4ea212482a75b890 Mon Sep 17 00:00:00 2001
From: linuxero <mageia@mtailounie.net>
Date: Thu, 12 Dec 2024 00:13:58 +0100
Subject: [PATCH] Now I can go to bed

---
 lib/DB/BusinessModel.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/lib/DB/BusinessModel.py b/lib/DB/BusinessModel.py
index e76b68a..701b176 100644
--- a/lib/DB/BusinessModel.py
+++ b/lib/DB/BusinessModel.py
@@ -66,11 +66,13 @@ class BusinessModel(QAbstractTableModel):
 
     def __init__(self):
         super().__init__()
-        self.__data = self.__getData()
+        self.__getData()
 
     def __getData(self, criterion = "Alle"):
+        self.beginResetModel()
         rows, self.__visible_columns = BusinessDAO().getBusiness(criterion)
-        return rows
+        self.__data = rows
+        self.endResetModel()
 
     def rowCount(self, parent= QModelIndex()):
         return len (self.__data)
@@ -109,14 +111,13 @@ class BusinessModel(QAbstractTableModel):
 
     @Slot(str)
     def viewCriterion(self, criterion):
-        self.beginResetModel()
-        self.__data = self.__getData(criterion)
-        self.endResetModel()
+        self.__getData(criterion)
+
 
     @Slot(dict, int)
     def addBusiness(self, business, contact_id):
         BusinessDAO().addBusiness(business, contact_id)
-        self.__data = self.__getData()
+        self.__getData()
 
     @Slot(dict)
     def setContact(self, contact):