110 lines
4.3 KiB
Python
110 lines
4.3 KiB
Python
import json
|
|
|
|
from PySide6.QtCore import QAbstractTableModel, QModelIndex, Qt, Slot, Signal, QJsonDocument
|
|
from PySide6.QtQml import QJSValue
|
|
|
|
from .EmployeeDAO import EmployeeDAO
|
|
# from ..PyqcrmFlags import PyqcrmFlags, PyqcrmAppliEmpyFlags
|
|
from ..ConfigLoader import ConfigLoader
|
|
import re
|
|
|
|
|
|
class EmployeeModel(QAbstractTableModel):
|
|
addedNewEmployee = Signal(bool)
|
|
__data = None
|
|
__employee_dao = None
|
|
__visible_index = None
|
|
__visible_columns = None
|
|
__col_name = ""
|
|
__col_skip = 2
|
|
__everyone = True
|
|
|
|
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)
|
|
def addEmployee(self, new_employee):
|
|
if 'worklicense' in new_employee:
|
|
new_employee['worklicense'] = int(new_employee['worklicense'])
|
|
new_employee['residencetype'] = int(new_employee['residencetype'])
|
|
self.__employee_dao.addEmployee(new_employee, self.__key, False)
|
|
|
|
@Slot(QJSValue)
|
|
def addApplicant(self, applicant: QJSValue):
|
|
self.__employee_dao.addEmployee({
|
|
"city": applicant.property("city").toString(),
|
|
"email": applicant.property("email").toString(),
|
|
"firstname": applicant.property("firstname").toString(),
|
|
"formofaddress": applicant.property("formofaddress").toString(),
|
|
"houseno": applicant.property("houseno").toString(),
|
|
"lastname": applicant.property("lastname").toString(),
|
|
"mobile": applicant.property("mobile").toString(),
|
|
"phone": applicant.property("phone").toString(),
|
|
"postcode": applicant.property("postcode").toInt(),
|
|
"street": applicant.property("street").toString(),
|
|
"title": applicant.property("title").toString(),
|
|
}, self.__key, True)
|
|
|
|
@Slot(bool)
|
|
def __refreshView(self, added):
|
|
if added:
|
|
self.__getData()
|
|
self.addedNewEmployee.emit(added)
|
|
|
|
def __getData(self, criterion="Alle", processed=False, fired=False, every_state=True):
|
|
self.beginResetModel()
|
|
rows, self.__visible_columns = self.__employee_dao.getEmployees(self.__key, criterion, processed, fired,
|
|
every_state)
|
|
self.__data = rows
|
|
self.endResetModel()
|
|
|
|
def rowCount(self, parent=QModelIndex()):
|
|
return len(self.__data)
|
|
|
|
def columnCount(self, parent=QModelIndex()):
|
|
return len(self.__visible_columns) - self.__col_skip
|
|
|
|
@Slot(str)
|
|
def viewCriterion(self, criterion):
|
|
self.__everyone = criterion == 'Alle'
|
|
self.__col_skip = 2
|
|
self.__getData(criterion, criterion == 'Erledigt', False, criterion == 'Alle')
|
|
|
|
def data(self, index, role=Qt.DisplayRole):
|
|
if role == Qt.DisplayRole:
|
|
row = self.__data[index.row()]
|
|
applicant_col = index.column() + self.__col_skip
|
|
tr = row[
|
|
applicant_col] # if type(row[index.column() + 2]) is str else str(row[index.column() + 2], "utf-8")
|
|
if applicant_col == 2 and self.__everyone:
|
|
tr = 'Ja' if tr == 1 else 'Nein'
|
|
else:
|
|
if tr:
|
|
tr = re.sub("Keine Angabe ", "", tr)
|
|
# print(f"Data: {tr}")
|
|
# return row[index.column() + 2]
|
|
return tr
|
|
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 + self.__col_skip]
|
|
return self.__col_name
|
|
return super().headerData(section, orientation, role)
|
|
|
|
@Slot(int)
|
|
def onRowClicked(self, row):
|
|
# print(self.__data)
|
|
print(f"Selected table row: {row}, corresponding DB ID: {self.__data[row][0]}")
|
|
# if not self.__employee_dict['employee'] or self.__data[row][0] != self.__employee_dict['employee']['id']:
|
|
# self.__employee = self.__employee_dao.getEmployee(self.__data[row][0], self.__key)
|
|
# print(self.__business)
|
|
# self.__getEmployeeInfo()
|
|
# self.__getContactInfo()
|
|
# print(self.__business_dict)
|