1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
# SPDX-License-Identifier: LGPL-2.1-only
from collections import OrderedDict
from contextlib import suppress
from PyQt6 import QtWidgets
import setools
from .criteria import OptionsPlacement
from .name import NameWidget
class RangedWidget(NameWidget):
"""
Base class for widgets providing a QLineEdit that saves the input to the attributes
of the specified query. This supports inputs of ranges, e.g. port ranges and MLS ranges.
"""
def __init__(self, title: str, query: setools.PolicyQuery, attrname: str, /, *,
required: bool = False,
validation: str = "",
enable_range_opts: bool = False,
options_placement: OptionsPlacement = OptionsPlacement.RIGHT,
parent: QtWidgets.QWidget | None = None):
# Not much we can do here. Leave all validation to the query.
super().__init__(title, query, attrname, [], validation, enable_regex=False,
options_placement=options_placement,
required=required, parent=parent)
# the rstrip("_") below is to avoid names like "range__overlap"
self.criteria_opts = OrderedDict[str, QtWidgets.QRadioButton]()
if enable_range_opts:
equ = QtWidgets.QRadioButton("Equal", parent=self)
equ.setChecked(True)
equ.toggled.connect(self._update_range_opts)
self.criteria_opts[""] = equ
ovl = QtWidgets.QRadioButton("Overlap", parent=self)
ovl.setChecked(False)
ovl.toggled.connect(self._update_range_opts)
self.criteria_opts[f"{attrname.rstrip('_')}_overlap"] = ovl
sub = QtWidgets.QRadioButton("Subset", parent=self)
sub.setChecked(False)
sub.toggled.connect(self._update_range_opts)
self.criteria_opts[f"{attrname.rstrip('_')}_subset"] = sub
sup = QtWidgets.QRadioButton("Superset", parent=self)
sup.setChecked(False)
sup.toggled.connect(self._update_range_opts)
self.criteria_opts[f"{attrname.rstrip('_')}_superset"] = sup
# place option radio buttons
match options_placement:
case OptionsPlacement.BELOW:
self.top_layout.addWidget(equ, 1, 0, 1, 1)
self.top_layout.addWidget(ovl, 1, 1, 1, 1)
self.top_layout.addWidget(sub, 2, 0, 1, 1)
self.top_layout.addWidget(sup, 2, 1, 1, 1)
case OptionsPlacement.RIGHT:
self.top_layout.addWidget(equ, 0, 1, 1, 1)
self.top_layout.addWidget(ovl, 0, 2, 1, 1)
self.top_layout.addWidget(sub, 1, 1, 1, 1)
self.top_layout.addWidget(sup, 1, 2, 1, 1)
case _:
raise AssertionError("Invalid options placement, this is an SETools bug.")
def _update_range_opts(self, value: bool = True) -> None:
"""Update the query based on the range opts radio button state."""
if not value:
return # only apply updates once per radio button switch
for name, w in self.criteria_opts.items():
if name: # empty means equal, which is the default
self.log.debug(f"Setting {name} to {w.isChecked()}")
setattr(self.query, name, w.isChecked())
def save(self, settings: dict) -> None:
super().save(settings)
for name, w in self.criteria_opts.items():
if name:
settings[name] = w.isChecked()
def load(self, settings: dict) -> None:
for name, w in self.criteria_opts.items():
if name:
with suppress(AttributeError, KeyError):
w.setChecked(settings[name])
super().load(settings)
|
