151 lines
5.4 KiB
Python
151 lines
5.4 KiB
Python
import os
|
|
|
|
from PyQt6 import QtCore, QtGui, QtWidgets
|
|
|
|
from ..models.MessagesModel import MessageItemDataRole, MessagesModelItem
|
|
from ..designs.widget_message import Ui_Form
|
|
from gotify_tray.database import Downloader
|
|
from gotify_tray.database import Settings
|
|
from gotify_tray.utils import convert_links, get_image
|
|
from gotify_tray.gui.themes import get_theme_file
|
|
from gotify_tray.gotify.models import GotifyMessageModel
|
|
|
|
|
|
settings = Settings("gotify-tray")
|
|
|
|
|
|
class MessageWidget(QtWidgets.QWidget, Ui_Form):
|
|
deletion_requested = QtCore.pyqtSignal(MessagesModelItem)
|
|
image_popup = QtCore.pyqtSignal(str, QtCore.QPoint)
|
|
|
|
def __init__(
|
|
self,
|
|
app: QtWidgets.QApplication,
|
|
parent: QtWidgets.QWidget,
|
|
message_item: MessagesModelItem,
|
|
icon: QtGui.QIcon | None = None,
|
|
):
|
|
super(MessageWidget, self).__init__(parent)
|
|
self.app = app
|
|
self.parent = parent
|
|
self.setupUi(self)
|
|
self.setAutoFillBackground(True)
|
|
self.message_item = message_item
|
|
message: GotifyMessageModel = message_item.data(MessageItemDataRole.MessageRole)
|
|
|
|
# Fonts
|
|
self.set_fonts()
|
|
|
|
# Display the message priority as a color
|
|
self.set_priority_color(message.priority)
|
|
|
|
# Display message contents
|
|
self.label_title.setText(message.title)
|
|
self.label_date.setText(message.date.strftime("%Y-%m-%d, %H:%M"))
|
|
|
|
if markdown := (
|
|
message.get("extras", {}).get("client::display", {}).get("contentType")
|
|
== "text/markdown"
|
|
):
|
|
self.label_message.setTextFormat(QtCore.Qt.TextFormat.MarkdownText)
|
|
|
|
# If the message is only an image URL, then instead of showing the message,
|
|
# download the image and show it in the message label
|
|
if image_url := get_image(message.message):
|
|
downloader = Downloader()
|
|
filename = downloader.get_filename(image_url)
|
|
self.set_message_image(filename)
|
|
else:
|
|
self.label_message.setText(convert_links(message.message))
|
|
|
|
# Show the application icon
|
|
if icon:
|
|
image_size = settings.value("MessageWidget/image/size", type=int)
|
|
pixmap = icon.pixmap(QtCore.QSize(image_size, image_size))
|
|
self.label_image.setPixmap(pixmap)
|
|
else:
|
|
self.label_image.hide()
|
|
|
|
# Set MessagesModelItem's size hint based on the size of this widget
|
|
self.gridLayout_frame.setContentsMargins(0, 0, 5, 0)
|
|
self.gridLayout.setContentsMargins(4, 5, 4, 0)
|
|
self.adjustSize()
|
|
size_hint = self.message_item.sizeHint()
|
|
self.message_item.setSizeHint(QtCore.QSize(size_hint.width(), self.height()))
|
|
|
|
self.set_icons()
|
|
|
|
self.link_callbacks()
|
|
|
|
def set_fonts(self):
|
|
font_title = QtGui.QFont()
|
|
font_date = QtGui.QFont()
|
|
font_content = QtGui.QFont()
|
|
|
|
if s := settings.value("MessageWidget/font/title", type=str):
|
|
font_title.fromString(s)
|
|
else:
|
|
font_title.setBold(True)
|
|
|
|
if s := settings.value("MessageWidget/font/date", type=str):
|
|
font_date.fromString(s)
|
|
else:
|
|
font_date.setItalic(True)
|
|
|
|
if s := settings.value("MessageWidget/font/message", type=str):
|
|
font_content.fromString(s)
|
|
|
|
self.label_title.setFont(font_title)
|
|
self.label_date.setFont(font_date)
|
|
self.label_message.setFont(font_content)
|
|
|
|
def set_icons(self):
|
|
self.pb_delete.setIcon(QtGui.QIcon(get_theme_file(self.app, "trashcan.svg")))
|
|
self.pb_delete.setIconSize(QtCore.QSize(24, 24))
|
|
|
|
def set_message_image(self, filename: str):
|
|
pixmap = QtGui.QPixmap(filename)
|
|
|
|
# Make sure the image fits within the listView
|
|
W = settings.value("MessageWidget/content_image/W_percentage", type=float) * (
|
|
self.parent.width() - self.label_image.width()
|
|
)
|
|
H = (
|
|
settings.value("MessageWidget/content_image/H_percentage", type=float)
|
|
* self.parent.height()
|
|
)
|
|
|
|
if pixmap.width() > W or pixmap.height() > H:
|
|
pixmap = pixmap.scaled(
|
|
QtCore.QSize(int(W), int(H)),
|
|
aspectRatioMode=QtCore.Qt.AspectRatioMode.KeepAspectRatio,
|
|
transformMode=QtCore.Qt.TransformationMode.SmoothTransformation,
|
|
)
|
|
|
|
self.label_message.setPixmap(pixmap)
|
|
|
|
def set_priority_color(self, priority: int):
|
|
if not settings.value("MessageWidget/priority_color", type=bool):
|
|
self.label_priority.setFixedWidth(0) # set width to 0 instead of hiding, so we still get the content margins
|
|
return
|
|
|
|
if priority >= 4 and priority <= 7:
|
|
self.label_priority.setStyleSheet("background-color: rgba(230, 126, 34, 0.7);")
|
|
elif priority > 7:
|
|
self.label_priority.setStyleSheet("background-color: #e74c3c;")
|
|
|
|
def link_hovered_callback(self, link: str):
|
|
if not settings.value("ImagePopup/enabled", type=bool):
|
|
return
|
|
|
|
qurl = QtCore.QUrl(link)
|
|
_, ext = os.path.splitext(qurl.fileName())
|
|
if ext in settings.value("ImagePopup/extensions", type=list):
|
|
self.image_popup.emit(link, QtGui.QCursor.pos())
|
|
|
|
def link_callbacks(self):
|
|
self.pb_delete.clicked.connect(
|
|
lambda: self.deletion_requested.emit(self.message_item)
|
|
)
|
|
self.label_message.linkHovered.connect(self.link_hovered_callback)
|