a better main window

gotify_tray/gui/models/MessagesModel.py

@@ -0,0 +1,23 @@
+import enum
+
+from typing import cast
+from PyQt6 import QtCore, QtGui
+from gotify_tray import gotify
+
+
+class MessageItemDataRole(enum.IntEnum):
+    MessageRole = QtCore.Qt.ItemDataRole.UserRole + 1
+
+
+class MessagesModelItem(QtGui.QStandardItem):
+    def __init__(self, message: gotify.GotifyMessageModel, *args, **kwargs):
+        super(MessagesModelItem, self).__init__()
+        self.setData(message, MessageItemDataRole.MessageRole)
+
+
+class MessagesModel(QtGui.QStandardItemModel):
+    def setItem(self, row: int, column: int, item: MessagesModelItem) -> None:
+        super(MessagesModel, self).setItem(row, column, item)
+
+    def itemFromIndex(self, index: QtCore.QModelIndex) -> MessagesModelItem:
+        return cast(MessagesModelItem, super(MessagesModel, self).itemFromIndex(index))