92 lines
3.4 KiB
Python
92 lines
3.4 KiB
Python
import os.path
|
|
from PyQt5 import uic
|
|
from PyQt5.QtCore import Qt, pyqtSlot, pyqtSignal
|
|
from PyQt5.QtWidgets import QWidget
|
|
from threads import DownloadThread, ThreadStatus
|
|
from utilities import pop_up, resource_path
|
|
from constants import Constants, Messages
|
|
|
|
Ui_Download_window, _ = uic.loadUiType(
|
|
resource_path(os.path.join("ui", "download_db_window.ui"))
|
|
)
|
|
|
|
|
|
class DownloadWindow(QWidget, Ui_Download_window):
|
|
"""Subclass QWidget and Ui_Download_window. It is the window displayed during the database download."""
|
|
|
|
complete = pyqtSignal()
|
|
|
|
def __init__(self):
|
|
"""Initialize the window."""
|
|
super().__init__()
|
|
self.setupUi(self)
|
|
self.setWindowFlags(
|
|
#Qt.Window |
|
|
Qt.CustomizeWindowHint |
|
|
Qt.WindowTitleHint |
|
|
Qt.WindowCloseButtonHint #|
|
|
# Qt.WindowStaysOnTopHint
|
|
)
|
|
|
|
self.__no_internet_msg = pop_up(self, title=Messages.NO_CONNECTION,
|
|
text=Messages.NO_CONNECTION_MSG,
|
|
connection=self.close)
|
|
|
|
self.__bad_db_download_msg = pop_up(self, title=Messages.BAD_DOWNLOAD,
|
|
text=Messages.BAD_DOWNLOAD_MSG,
|
|
connection=self.close)
|
|
|
|
self.__download_thread = DownloadThread()
|
|
self.__download_thread.finished.connect(self.__wait_close)
|
|
self.__download_thread.progress.connect(self.__display_progress)
|
|
self.cancel_btn.clicked.connect(self.__terminate_process)
|
|
|
|
def start_download(self):
|
|
"""Start the download thread."""
|
|
self.__download_thread.start()
|
|
|
|
def __downlaod_format_str(self, n, speed):
|
|
"""Return a well-formatted string with downloaded MB and speed."""
|
|
return f"Downloaded MB: {n}\nSpeed: {speed} MB/s"
|
|
|
|
def show(self):
|
|
"""Extends QWidget.show. Set downloaded MB and speed to zero."""
|
|
self.status_lbl.setText(self.__downlaod_format_str(0, 0))
|
|
super().show()
|
|
|
|
@pyqtSlot(int, float)
|
|
def __display_progress(self, progress, speed):
|
|
"""Display the downloaded MB and speed."""
|
|
if progress != Constants.EXTRACTING_CODE:
|
|
self.status_lbl.setText(self.__downlaod_format_str(progress, speed))
|
|
elif progress == Constants.EXTRACTING_CODE:
|
|
self.status_lbl.setText(Constants.EXTRACTING_MSG + '\n')
|
|
|
|
@pyqtSlot()
|
|
def __terminate_process(self):
|
|
"""Terminate the download thread and close."""
|
|
if self.__download_thread.isRunning():
|
|
self.__download_thread.terminate()
|
|
self.__download_thread.wait()
|
|
self.close()
|
|
|
|
@pyqtSlot()
|
|
def __wait_close(self):
|
|
"""Decide the action based on the download thread status and close."""
|
|
if self.__download_thread.status is ThreadStatus.OK:
|
|
self.complete.emit()
|
|
self.close()
|
|
elif self.__download_thread.status is ThreadStatus.NO_CONNECTION_ERR:
|
|
self.__no_internet_msg.show()
|
|
elif self.__download_thread.status is ThreadStatus.BAD_DOWNLOAD_ERR:
|
|
self.__bad_db_download_msg.show()
|
|
else:
|
|
self.close()
|
|
|
|
def reject(self):
|
|
"""Extends QWidget.reject. Terminate the download thread."""
|
|
if self.__download_thread.isRunning():
|
|
self.__download_thread.terminate()
|
|
self.__download_thread.wait()
|
|
super().reject()
|