122 lines
4.2 KiB
Python
122 lines
4.2 KiB
Python
from PyQt5.QtWidgets import (
|
|
QAbstractScrollArea,
|
|
QDialog,
|
|
QHBoxLayout,
|
|
QHeaderView,
|
|
QPlainTextEdit,
|
|
QTableWidget,
|
|
QTableWidgetItem,
|
|
QVBoxLayout,
|
|
QLabel,
|
|
QPushButton,
|
|
)
|
|
from logging import error
|
|
from pprint import pformat
|
|
|
|
|
|
class QuestionBoxDetails(QDialog):
|
|
"""
|
|
Dialog box primitive that displays arbitrary `data`, and prompts user for a `reply` as a QPushButton
|
|
|
|
Args:
|
|
- `reply`: An integer representing a user choice based on key of `choices`
|
|
- `choices`: A list of tuples, where each tuple represents a valid reply
|
|
e.g. [(reply: int, name: str),]
|
|
|
|
Notes:
|
|
You may supply an arbitrary number of choices, but here are my suggestions:
|
|
2 = The alternative
|
|
1 = The default
|
|
0 = Cancel
|
|
-1 = Cancel action (non-response)
|
|
"""
|
|
def __init__(self, title: str, description: str, data: str | list | dict, choices: list[tuple[int, str]]):
|
|
super(QuestionBoxDetails, self).__init__()
|
|
self.title: str = title
|
|
self.description: str = description
|
|
self.data: str | list | dict = data
|
|
self.reply: int = -1
|
|
self.setWindowTitle(title)
|
|
self.setMinimumSize(600, 400)
|
|
self.setMaximumSize(600, 400)
|
|
layout = QVBoxLayout()
|
|
h_layout = QHBoxLayout()
|
|
|
|
# Labels & input fields
|
|
label = QLabel(description)
|
|
layout.addWidget(label)
|
|
|
|
if isinstance(self.data, str):
|
|
self.input_field = QPlainTextEdit(pformat(self.data))
|
|
layout.addWidget(self.input_field)
|
|
else:
|
|
table: QTableWidget = QTableWidget()
|
|
table.setSizeAdjustPolicy(QAbstractScrollArea.SizeAdjustPolicy.AdjustToContents)
|
|
table.horizontalHeader().setStretchLastSection(True) # type: ignore
|
|
table.horizontalHeader().setSectionResizeMode(QHeaderView.Interactive) # type: ignore
|
|
if isinstance(self.data, list):
|
|
table.setRowCount(len(data))
|
|
table.setColumnCount(1)
|
|
for i, item in enumerate(self.data):
|
|
table.setItem(i, 0, QTableWidgetItem(str(item)))
|
|
layout.addWidget(table)
|
|
elif isinstance(self.data, dict):
|
|
try:
|
|
# | TIT2 | title goes here |
|
|
# | TDRC | 2025-05-05 |
|
|
table.setRowCount(len(data.keys())) # type: ignore
|
|
table.setColumnCount(2)
|
|
table.setHorizontalHeaderLabels(['Key', 'Value'])
|
|
for i, (k, v) in enumerate(data.items()): # type: ignore
|
|
table.setItem(i, 0, QTableWidgetItem(str(k)))
|
|
table.setItem(i, 1, QTableWidgetItem(str(v)))
|
|
layout.addWidget(table)
|
|
except Exception as e:
|
|
data = str(self.data)
|
|
self.input_field = QPlainTextEdit(pformat(data + "\n\n" + str(e)))
|
|
layout.addWidget(self.input_field)
|
|
error(f'Tried to load self.data as dict but could not. {e}')
|
|
|
|
# dynamically load the button choices
|
|
def create_button(choice):
|
|
def func():
|
|
self.reply = choice[0]
|
|
self.close()
|
|
button = QPushButton(choice[1])
|
|
button.clicked.connect(func)
|
|
return button
|
|
|
|
for idx, choice in enumerate(choices):
|
|
button = create_button(choice)
|
|
h_layout.addWidget(button)
|
|
if idx == 0:
|
|
button.setFocus()
|
|
|
|
|
|
# # ok
|
|
# ok_button = QPushButton("Confirm")
|
|
# ok_button.clicked.connect(self.ok)
|
|
# h_layout.addWidget(ok_button)
|
|
# # cancel
|
|
# cancel_button = QPushButton("no")
|
|
# cancel_button.clicked.connect(self.cancel)
|
|
# h_layout.addWidget(cancel_button)
|
|
|
|
layout.addLayout(h_layout)
|
|
self.setLayout(layout)
|
|
|
|
def execute(self):
|
|
"""
|
|
Returns when `self` is closed
|
|
"""
|
|
self.exec_()
|
|
return self.reply
|
|
|
|
def cancel(self):
|
|
self.reply = False
|
|
self.close()
|
|
|
|
def ok(self):
|
|
self.reply = True
|
|
self.close()
|