149 lines
5.5 KiB
Python
149 lines
5.5 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, update_widget_property
|
|
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,
|
|
parent: QtWidgets.QWidget,
|
|
message_item: MessagesModelItem,
|
|
icon: QtGui.QIcon | None = None,
|
|
):
|
|
super(MessageWidget, self).__init__(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)
|
|
|
|
if settings.value("locale", type=bool):
|
|
date_str = QtCore.QLocale.system().toString(message.date, QtCore.QLocale.FormatType.ShortFormat)
|
|
else:
|
|
date_str = message.date.toString("yyyy-MM-dd, hh:mm")
|
|
self.label_date.setText(date_str)
|
|
|
|
if 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(),
|
|
max(settings.value("MessageWidget/height/min", type=int), 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("trashcan.svg")))
|
|
|
|
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)
|
|
H = settings.value("MessageWidget/content_image/H_percentage", type=float)
|
|
W *= self.parent().width() - self.label_image.width()
|
|
H *= 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:
|
|
update_widget_property(self.label_priority, "priority", "medium")
|
|
elif priority > 7:
|
|
update_widget_property(self.label_priority, "priority", "high")
|
|
|
|
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)
|