RepoMirrors/setools
1
0
Fork 0
mirror of https://github.com/SELinuxProject/setools synced 2025-02-19 13:37:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

RuleListModel: restore parent class for rule list models

Browse Source
This commit is contained in:
Chris PeBenito 2015-12-15 10:10:28 -05:00
parent f89f875a62
commit fe970680af
1 changed files with 30 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

72
setoolsgui/apol/rulemodels.py
View File

@ -21,17 +21,34 @@ from PyQt5.QtCore import Qt, QAbstractTableModel, QModelIndex
from setools.policyrep.exception import RuleNotConditional, RuleUseError
class MLSRuleListModel(QAbstractTableModel):
class RuleListModel(QAbstractTableModel):
"""MLS rule model. Represents rules as a column."""
"""Base class for rule list models."""
def __init__(self, parent):
super(MLSRuleListModel, self).__init__(parent)
super(RuleListModel, self).__init__(parent)
self.resultlist = []
def headerData(self, section, orientation, role):
raise NotImplementedError
def columnCount(self, parent=QModelIndex()):
return 5
def rowCount(self, parent=QModelIndex()):
if self.resultlist:
return len(self.resultlist)
else:
return 0
def data(self, index, role):
raise NotImplementedError
class MLSRuleListModel(RuleListModel):
"""MLS rule model. Represents rules as a column."""
def headerData(self, section, orientation, role):
if role == Qt.DisplayRole and orientation == Qt.Horizontal:
if section == 0:
@ -45,13 +62,7 @@ class MLSRuleListModel(QAbstractTableModel):
elif section == 4:
return "Default Range"
else:
raise ValueError("Invalid column number")
def rowCount(self, parent=QModelIndex()):
if self.resultlist:
return len(self.resultlist)
else:
return 0
raise ValueError("Invalid column number: {0}".format(section))
def data(self, index, role):
if role == Qt.DisplayRole:
@ -72,23 +83,16 @@ class MLSRuleListModel(QAbstractTableModel):
elif col == 4:
return str(self.resultlist[row].default)
else:
raise ValueError("Invalid column number")
raise ValueError("Invalid column number: {0}".format(col))
elif role == Qt.UserRole:
# get the whole rule for user role
return self.resultlist[row].statement()
class RBACRuleListModel(QAbstractTableModel):
class RBACRuleListModel(RuleListModel):
"""RBAC rule model. Represents rules as a column."""
def __init__(self, parent):
super(RBACRuleListModel, self).__init__(parent)
self.resultlist = []
def columnCount(self, parent=QModelIndex()):
return 5
def headerData(self, section, orientation, role):
if role == Qt.DisplayRole and orientation == Qt.Horizontal:
if section == 0:
@ -102,13 +106,7 @@ class RBACRuleListModel(QAbstractTableModel):
elif section == 4:
return "Default Role"
else:
raise ValueError("Invalid column number")
def rowCount(self, parent=QModelIndex()):
if self.resultlist:
return len(self.resultlist)
else:
return 0
raise ValueError("Invalid column number: {0}".format(section))
def data(self, index, role):
if role == Qt.DisplayRole:
@ -137,23 +135,16 @@ class RBACRuleListModel(QAbstractTableModel):
except RuleUseError:
return None
else:
raise ValueError("Invalid column number")
raise ValueError("Invalid column number: {0}".format(col))
elif role == Qt.UserRole:
# get the whole rule for user role
return self.resultlist[row].statement()
class TERuleListModel(QAbstractTableModel):
class TERuleListModel(RuleListModel):
"""Type Enforcement rule model. Represents rules as a column."""
def __init__(self, parent):
super(TERuleListModel, self).__init__(parent)
self.resultlist = []
def columnCount(self, parent=QModelIndex()):
return 6
def headerData(self, section, orientation, role):
if role == Qt.DisplayRole and orientation == Qt.Horizontal:
if section == 0:
@ -169,13 +160,10 @@ class TERuleListModel(QAbstractTableModel):
elif section == 5:
return "Conditional Expression"
else:
raise ValueError("Invalid column number")
raise ValueError("Invalid column number: {0}".format(section))
def rowCount(self, parent=QModelIndex()):
if self.resultlist:
return len(self.resultlist)
else:
return 0
def columnCount(self, parent=QModelIndex()):
return 6
def data(self, index, role):
if role == Qt.DisplayRole:
@ -204,7 +192,7 @@ class TERuleListModel(QAbstractTableModel):
except RuleNotConditional:
return None
else:
raise ValueError("Invalid column number")
raise ValueError("Invalid column number: {0}".format(col))
elif role == Qt.UserRole:
# get the whole rule for user role
return self.resultlist[row].statement()