58 lines
1.8 KiB
Python
58 lines
1.8 KiB
Python
# This Python file uses the following encoding: utf-8
|
|
from PySide6.QtCore import QAbstractTableModel, QModelIndex, Qt, Slot
|
|
|
|
|
|
CUSTOMER_COLUMN_NAMES = \
|
|
{
|
|
"userid": "Identification",
|
|
"username": "Benutzername",
|
|
"password": "Passwort",
|
|
"enabled": "Aktiv",
|
|
"roleid": "Rolle",
|
|
"gecos": "Information"
|
|
}
|
|
|
|
class BusinessModel(QAbstractTableModel):
|
|
|
|
def __init__(self, con, cols):
|
|
super().__init__()
|
|
self.__con = con
|
|
self.__visible_columns = cols
|
|
self.__data = self.__getData()
|
|
|
|
def __getData(self):
|
|
cursor = self.__con.cursor()
|
|
cursor.execute("SELECT * FROM users")
|
|
self.__all_cols = [desc[0] for desc in cursor.description]
|
|
rows = cursor.fetchall()
|
|
return rows
|
|
|
|
def rowCount(self, parent= QModelIndex()):
|
|
return len (self.__data)
|
|
|
|
def columnCount(self, parent= QModelIndex()):
|
|
return len(self.__visible_columns)
|
|
|
|
def data(self, index, role= Qt.DisplayRole):
|
|
if role == Qt.DisplayRole:
|
|
# row = index.row()
|
|
row = self.__data[index.row()]
|
|
col_name = self.__visible_columns[index.column()]
|
|
col_index = self.__all_cols.index(col_name)
|
|
return row[col_index]
|
|
return None
|
|
|
|
def headerData(self, section, orientation, role= Qt.DisplayRole):
|
|
#header= ["ID", "Benutzername", "Passwort", "Information", "Rolle", "Aktiv"]
|
|
if orientation == Qt.Horizontal and role ==Qt.DisplayRole:
|
|
#return header[section]
|
|
col_name = self.__visible_columns[section]
|
|
return CUSTOMER_COLUMN_NAMES.get(col_name, col_name)
|
|
return super().headerData(section, orientation, role)
|
|
#return None
|
|
|
|
@Slot(int)
|
|
def onRowClicked(self, row):
|
|
print(row)
|
|
|