Artemis 4 RC1

artemis/ui/downloader.py

@@ -0,0 +1,133 @@
+from PySide6.QtQml import QQmlApplicationEngine
+from PySide6.QtCore import QObject, Slot, Signal, QUrl, QSaveFile, QDir, QIODevice
+from PySide6.QtNetwork import QNetworkReply, QNetworkRequest, QNetworkAccessManager
+
+from artemis.utils.config_utils import *
+from artemis.utils.sys_utils import delete_file, match_hash, unpack_db
+from artemis.utils.constants import Messages
+from artemis.utils.sys_utils import delete_db_dir
+
+
+class UIDownloader(QObject):
+    # Python > QML Signals
+    show_ui = Signal()
+    close_ui = Signal()
+
+    def __init__(self, parent):
+        super().__init__()
+
+        self._parent = parent
+
+        self._engine = QQmlApplicationEngine()
+        self._engine.load('qrc:/ui/Downloader.qml')
+        self._window = self._engine.rootObjects()[0]
+        self._progress_bar = self._window.findChild(QObject, "progressBar")
+        self._label_progress = self._window.findChild(QObject, "labelProgress")
+
+        self._connect()
+
+
+    def _connect(self):
+        # QML > Python connections
+        self._window.onAbort.connect(self.on_abort)
+
+        # Python > QML connections
+        self.show_ui.connect(self._window.show)
+        self.close_ui.connect(self._window.close)
+
+
+    @Slot()
+    def on_start(self):
+        """ Start the download of the DB taking the needed url and size from
+            the attributes of the UpdatesController class
+        """
+        url_file = QUrl(self._parent.network_manager.remote_db_url)
+        dest_path = QDir(Constants.DB_DIR)
+        self.dest_file = dest_path.filePath(url_file.fileName())
+        self.file = QSaveFile(self.dest_file)
+
+        if self.file.open(QIODevice.WriteOnly):
+            # Start a GET HTTP request
+            self.manager = QNetworkAccessManager(self)
+            self.reply = self.manager.get(QNetworkRequest(url_file))
+            self.reply.downloadProgress.connect(self.on_progress)
+            self.reply.finished.connect(self.on_finished)
+            self.reply.readyRead.connect(self.on_ready_read)
+            self.reply.errorOccurred.connect(self.on_error)
+        else:
+            self.close_ui.emit()
+            self.show_popup_error(
+                self.file.errorString()
+            )
+
+
+    @Slot()
+    def on_abort(self):
+        """ Stop the download when user press abort button """
+        if self.reply:
+            self.reply.abort()
+            self._progress_bar.setProperty("value", 0)
+
+        if self.file:
+            self.file.cancelWriting()
+
+        self.close_ui.emit()
+
+
+    @Slot()
+    def on_ready_read(self):
+        """ Get available bytes and store them into the file """
+        if self.reply:
+            if self.reply.error() == QNetworkReply.NoError:
+                self.file.write(self.reply.readAll())
+
+
+    @Slot()
+    def on_finished(self):
+        """ Delete reply, close the file, check the hash for integrity,
+            extract the database and delete the downloaded zip
+        """
+        if self.reply:
+            self.reply.deleteLater()
+
+        if self.file:
+            self.file.commit()
+
+        self._label_progress.setProperty("text", "Checking DB integrity (SHA-256)")
+
+        if match_hash(self.dest_file, self._parent.network_manager.remote_db_hash):
+            self._label_progress.setProperty("text", "Unpacking archive...")
+            delete_db_dir('SigID')
+            unpack_db(self.dest_file, 'SigID')
+            delete_file(self.dest_file)
+            self._parent.load_db('SigID')
+            self.close_ui.emit()
+
+
+    @Slot(int, int)
+    def on_progress(self, bytesReceived: int):
+        """ Update progress bar and label
+        """
+        total_bytes = self._parent.network_manager.remote_db_size
+        self._label_progress.setProperty("text", "{:.1f} Mb / {:.1f} Mb".format(bytesReceived/10**6, total_bytes/10**6))
+        self._progress_bar.setProperty("to", total_bytes)
+        self._progress_bar.setProperty("value", bytesReceived)
+
+
+    @Slot(QNetworkReply.NetworkError)
+    def on_error(self, code: QNetworkReply.NetworkError):
+        """ Show a message if an error happen during download
+        """
+        if self.reply:
+            self.close_ui.emit()
+            self.show_popup_error(
+                self.reply.errorString()
+            )
+
+
+    def show_popup_error(self, error_msg):
+        self._parent.dialog_popup(
+            Messages.DIALOG_TYPE_ERROR,
+            Messages.GENERIC_ERROR,
+            Messages.GENERIC_ERROR_MSG.format(error_msg)
+        )