mirror of
https://github.com/SELinuxProject/setools
synced 2025-04-24 12:24:27 +00:00
* Remove "Model" from names of model classes. * Remove remaining *_detail() functions. * Add typing. * Make models dir a package. * Update for superclass data() method as fallback. * Switch to match/case statements. Signed-off-by: Chris PeBenito <pebenito@ieee.org>
40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
# Copyright 2016, Tresys Technology, LLC
|
|
#
|
|
# SPDX-License-Identifier: LGPL-2.1-only
|
|
#
|
|
#
|
|
from PyQt5 import QtCore
|
|
import setools
|
|
|
|
from .table import SEToolsTableModel
|
|
|
|
|
|
class PortconTable(SEToolsTableModel[setools.Portcon]):
|
|
|
|
"""Table-based model for portcons."""
|
|
|
|
headers = ["Port/Port Range", "Protocol", "Context"]
|
|
|
|
def data(self, index: QtCore.QModelIndex, role: int = QtCore.Qt.ItemDataRole.DisplayRole):
|
|
if not self.item_list or not index.isValid():
|
|
return None
|
|
|
|
row = index.row()
|
|
col = index.column()
|
|
rule = self.item_list[row]
|
|
|
|
match role:
|
|
case QtCore.Qt.ItemDataRole.DisplayRole:
|
|
match col:
|
|
case 0:
|
|
low, high = rule.ports
|
|
if low == high:
|
|
return str(low)
|
|
return f"{low}-{high}"
|
|
case 1:
|
|
return rule.protocol.name
|
|
case 2:
|
|
return str(rule.context)
|
|
|
|
return super().data(index, role)
|