59 lines
1.9 KiB
Python
59 lines
1.9 KiB
Python
from PySide6.QtCore import QAbstractTableModel, QModelIndex, Qt, Slot, Signal
|
|
from .EmployeeDAO import EmployeeDAO
|
|
from ..PyqcrmFlags import PyqcrmFlags, PyqcrmAppliEmpyFlags
|
|
from ..ConfigLoader import ConfigLoader
|
|
|
|
|
|
class EmployeeModel(QAbstractTableModel):
|
|
__data = None
|
|
__employee_dao = None
|
|
__visible_index = None
|
|
__visible_columns = None
|
|
__col_name = ""
|
|
__employee_dao = None
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.__employee_dao = EmployeeDAO()
|
|
self.__employee_dao.newEmployeeAdded.connect(self.__refreshView)
|
|
self.__conf = ConfigLoader().getConfig()
|
|
self.__key = self.__conf['pyqcrm']['ENCRYPTION_KEY']
|
|
self.__getData()
|
|
|
|
@Slot(dict, bool)
|
|
def addEmployee(self, new_employee, employee = False):
|
|
print(new_employee)
|
|
self.__employee_dao.addEmployee(new_employee, self.__key, employee)
|
|
|
|
@Slot(str)
|
|
def viewCriterion(self, criterion):
|
|
self.__getData(criterion)
|
|
|
|
@Slot()
|
|
def __refreshView(self):
|
|
self.__getData()
|
|
|
|
def __getData(self, criterion = "Alle", processed = False):
|
|
self.beginResetModel()
|
|
rows, self.__visible_columns = self.__employee_dao.getEmployees(self.__key, criterion, 0, processed)
|
|
self.__data = rows
|
|
self.endResetModel()
|
|
|
|
def rowCount(self, parent= QModelIndex()):
|
|
return len (self.__data)
|
|
|
|
def columnCount(self, parent= QModelIndex()):
|
|
return len(self.__visible_columns) - 2
|
|
|
|
def data(self, index, role= Qt.DisplayRole):
|
|
if role == Qt.DisplayRole:
|
|
row = self.__data[index.row()]
|
|
return row[index.column() + 2]
|
|
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 + 2]
|
|
return self.__col_name
|
|
return super().headerData(section, orientation, role)
|