initial commit

gotify_tray/tasks.py

@@ -0,0 +1,148 @@
+import abc
+import logging
+
+from PyQt6 import QtCore
+from PyQt6.QtCore import pyqtSignal
+
+from . import gotify
+
+
+logger = logging.getLogger("logger")
+
+
+class BaseTask(QtCore.QThread):
+    failed = pyqtSignal()
+
+    def __init__(self):
+        super(BaseTask, self).__init__()
+        self.running = False
+
+    @abc.abstractmethod
+    def task(self):
+        ...
+
+    def run(self):
+        self.running = True
+        try:
+            self.task()
+        except Exception as e:
+            logger.error(f"{self.__class__.__name__} failed: {e}")
+            self.failed.emit()
+        finally:
+            self.running = False
+
+
+class DeleteMessageTask(BaseTask):
+    deleted = pyqtSignal(bool)
+
+    def __init__(self, message_id: int, gotify_client: gotify.GotifyClient):
+        super(DeleteMessageTask, self).__init__()
+        self.message_id = message_id
+        self.gotify_client = gotify_client
+
+    def task(self):
+        success = self.gotify_client.delete_message(self.message_id)
+        self.deleted.emit(success)
+
+
+class DeleteApplicationMessagesTask(BaseTask):
+    deleted = pyqtSignal(bool)
+
+    def __init__(self, appid: int, gotify_client: gotify.GotifyClient):
+        super(DeleteApplicationMessagesTask, self).__init__()
+        self.appid = appid
+        self.gotify_client = gotify_client
+
+    def task(self):
+        success = self.gotify_client.delete_application_messages(self.appid)
+        self.deleted.emit(success)
+
+
+class DeleteAllMessagesTask(BaseTask):
+    deleted = pyqtSignal(bool)
+
+    def __init__(self, gotify_client: gotify.GotifyClient):
+        super(DeleteAllMessagesTask, self).__init__()
+        self.gotify_client = gotify_client
+
+    def task(self):
+        success = self.gotify_client.delete_messages()
+        self.deleted.emit(success)
+
+
+class GetApplicationsTask(BaseTask):
+    success = pyqtSignal(list)
+    error = pyqtSignal(gotify.GotifyErrorModel)
+
+    def __init__(self, gotify_client: gotify.GotifyClient):
+        super(GetApplicationsTask, self).__init__()
+        self.gotify_client = gotify_client
+
+    def task(self):
+        result = self.gotify_client.get_applications()
+        if isinstance(result, gotify.GotifyErrorModel):
+            self.error.emit(result)
+        else:
+            self.success.emit(result)
+
+
+class GetApplicationMessagesTask(BaseTask):
+    success = pyqtSignal(gotify.GotifyPagedMessagesModel)
+    error = pyqtSignal(gotify.GotifyErrorModel)
+
+    def __init__(self, appid: int, gotify_client: gotify.GotifyClient):
+        super(GetApplicationMessagesTask, self).__init__()
+        self.appid = appid
+        self.gotify_client = gotify_client
+
+    def task(self):
+        result = self.gotify_client.get_application_messages(self.appid)
+        if isinstance(result, gotify.GotifyErrorModel):
+            self.error.emit(result)
+        else:
+            self.success.emit(result)
+
+
+class GetMessagesTask(BaseTask):
+    success = pyqtSignal(gotify.GotifyPagedMessagesModel)
+    error = pyqtSignal(gotify.GotifyErrorModel)
+
+    def __init__(self, gotify_client: gotify.GotifyClient):
+        super(GetMessagesTask, self).__init__()
+        self.gotify_client = gotify_client
+
+    def task(self):
+        result = self.gotify_client.get_messages()
+        if isinstance(result, gotify.GotifyErrorModel):
+            self.error.emit(result)
+        else:
+            self.success.emit(result)
+
+
+class VerifyServerInfoTask(BaseTask):
+    success = pyqtSignal()
+    incorrect_token = pyqtSignal()
+    incorrect_url = pyqtSignal()
+
+    def __init__(self, url: str, client_token: str):
+        super(VerifyServerInfoTask, self).__init__()
+        self.url = url
+        self.client_token = client_token
+
+    def task(self):
+        try:
+            gotify_client = gotify.GotifyClient(self.url, self.client_token)
+            result = gotify_client.get_messages(limit=1)
+
+            if isinstance(result, gotify.GotifyPagedMessagesModel):
+                self.success.emit()
+                return
+            elif (
+                isinstance(result, gotify.GotifyErrorModel)
+                and result["error"] == "Unauthorized"
+            ):
+                self.incorrect_token.emit()
+                return
+            self.incorrect_url.emit()
+        except Exception as e:
+            self.incorrect_url.emit()