100 lines
3.2 KiB
Python
100 lines
3.2 KiB
Python
'''
|
|
Contains dialogs for notification and prompting.
|
|
'''
|
|
from PyQt6.QtWidgets import (
|
|
QLabel, QVBoxLayout, QDialog,
|
|
QDialogButtonBox, QMessageBox, QComboBox
|
|
)
|
|
from PyQt6.QtWebEngineWidgets import QWebEngineView
|
|
from PyQt6.QtCore import Qt
|
|
from tools import jinja_template_loading
|
|
import logging
|
|
from backend.db import models
|
|
from typing import Literal
|
|
|
|
logger = logging.getLogger(f"submissions.{__name__}")
|
|
|
|
env = jinja_template_loading()
|
|
|
|
|
|
class QuestionAsker(QDialog):
|
|
"""
|
|
dialog to ask yes/no questions
|
|
"""
|
|
def __init__(self, title:str, message:str):
|
|
super().__init__()
|
|
self.setWindowTitle(title)
|
|
# NOTE: set yes/no buttons
|
|
QBtn = QDialogButtonBox.StandardButton.Yes | QDialogButtonBox.StandardButton.No
|
|
self.buttonBox = QDialogButtonBox(QBtn)
|
|
self.buttonBox.accepted.connect(self.accept)
|
|
self.buttonBox.rejected.connect(self.reject)
|
|
self.layout = QVBoxLayout()
|
|
# NOTE: Text for the yes/no question
|
|
self.message = QLabel(message)
|
|
self.layout.addWidget(self.message)
|
|
self.layout.addWidget(self.buttonBox)
|
|
self.setLayout(self.layout)
|
|
|
|
|
|
class AlertPop(QMessageBox):
|
|
"""
|
|
Dialog to show an alert.
|
|
"""
|
|
def __init__(self, message:str, status:Literal['Information', 'Question', 'Warning', 'Critical'], owner:str|None=None):
|
|
super().__init__()
|
|
# NOTE: select icon by string
|
|
icon = getattr(QMessageBox.Icon, status)
|
|
self.setIcon(icon)
|
|
self.setInformativeText(message)
|
|
self.setWindowTitle(f"{owner} - {status.title()}")
|
|
|
|
class HTMLPop(QDialog):
|
|
|
|
def __init__(self, html:str, owner:str|None=None, title:str="python"):
|
|
super().__init__()
|
|
|
|
self.webview = QWebEngineView(parent=self)
|
|
self.layout = QVBoxLayout()
|
|
self.setWindowTitle(title)
|
|
self.webview.setHtml(html)
|
|
self.webview.setMinimumSize(600, 500)
|
|
self.webview.setMaximumSize(600, 500)
|
|
self.layout.addWidget(self.webview)
|
|
|
|
|
|
class ObjectSelector(QDialog):
|
|
"""
|
|
dialog to input BaseClass type manually
|
|
"""
|
|
def __init__(self, title:str, message:str, obj_type:str|models.BaseClass):
|
|
super().__init__()
|
|
self.setWindowTitle(title)
|
|
self.widget = QComboBox()
|
|
if isinstance(obj_type, str):
|
|
obj_type: models.BaseClass = getattr(models, obj_type)
|
|
items = [item.name for item in obj_type.query()]
|
|
self.widget.addItems(items)
|
|
self.widget.setEditable(False)
|
|
# NOTE: set yes/no buttons
|
|
QBtn = QDialogButtonBox.StandardButton.Ok | QDialogButtonBox.StandardButton.Cancel
|
|
self.buttonBox = QDialogButtonBox(QBtn)
|
|
self.buttonBox.accepted.connect(self.accept)
|
|
self.buttonBox.rejected.connect(self.reject)
|
|
self.layout = QVBoxLayout()
|
|
# NOTE: Text for the yes/no question
|
|
message = QLabel(message)
|
|
self.layout.addWidget(message)
|
|
self.layout.addWidget(self.widget)
|
|
self.layout.addWidget(self.buttonBox)
|
|
self.setLayout(self.layout)
|
|
|
|
def parse_form(self) -> str:
|
|
"""
|
|
Get KitType(str) from widget
|
|
|
|
Returns:
|
|
str: KitType as str
|
|
"""
|
|
return self.widget.currentText()
|