summer/summer/templates/RoleList.py

71 lines
2.3 KiB
Python

__author__ = 'tanshu'
from PyQt5 import uic, QtCore, QtWidgets
file = '/home/tanshu/Programming/summer/summer/QtDesignerFiles/ListForm.ui'
base, form = uic.loadUiType(file)
class RoleList(base, form):
def __init__(self, router, roles, parent=None):
super(RoleList, self).__init__()
self.setupUi(self)
self.translateUi()
self.router = router
self.roles = roles
self.model = RolesTableModel(roles)
self.tblItems.setModel(self.model)
self.tblItems.doubleClicked.connect(self.show_role)
self.btnAddNew.clicked.connect(self.add_new_role)
self.setAttribute(QtCore.Qt.WA_DeleteOnClose, True)
self.tblItems.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows)
def translateUi(self):
self.setObjectName("RoleList")
_translate = QtCore.QCoreApplication.translate
self.setWindowTitle(_translate("ListForm", "Roles"))
self.btnAddNew.setText(_translate("ListForm", "Add Role"))
def update_data(self, data):
self.roles = data
self.model = RolesTableModel(self.roles)
self.tblItems.setModel(self.model)
def show_role(self, index):
role = self.model.roles[index.row()]
self.router.show_role(role['RoleID'])
def add_new_role(self):
self.router.show_role(None)
class RolesTableModel(QtCore.QAbstractTableModel):
def __init__(self, roles=None, parent=None):
QtCore.QAbstractTableModel.__init__(self, parent)
self.roles = roles if roles is not None else []
def rowCount(self, parent):
return len(self.roles)
def columnCount(self, parent):
return 2
def data(self, index, role):
if role == QtCore.Qt.DisplayRole:
item = self.roles[index.row()]
column = index.column()
if column == 0:
return item['Name']
elif column == 1:
return ", ".join(item['Permissions'])
def headerData(self, section, orientation, role):
if role == QtCore.Qt.DisplayRole:
if orientation == QtCore.Qt.Horizontal:
if section == 0:
return 'Name'
elif section == 1:
return 'Permissions'
else:
return section + 1