76 lines
2.9 KiB
Python
76 lines
2.9 KiB
Python
from PyQt5 import uic
|
|
from PyQt5.QtCore import Qt, pyqtSlot
|
|
from PyQt5.QtWidgets import QWidget, QMessageBox
|
|
from threads import DownloadThread, ThreadStatus
|
|
Ui_Download_window, _ = uic.loadUiType("download_db_window.ui")
|
|
|
|
class DownloadWindow(QWidget, Ui_Download_window):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.setupUi(self)
|
|
self.setWindowFlags(
|
|
#Qt.Window |
|
|
Qt.CustomizeWindowHint |
|
|
Qt.WindowTitleHint |
|
|
Qt.WindowCloseButtonHint #|
|
|
# Qt.WindowStaysOnTopHint
|
|
)
|
|
self.everything_ok = True
|
|
self.no_internet_msg = QMessageBox(self)
|
|
self.no_internet_msg.setWindowTitle("No internet connection")
|
|
self.no_internet_msg.setText("Unable to establish an internet connection.")
|
|
# self.no_internet_msg.buttonClicked.connect(self.close)
|
|
self.no_internet_msg.finished.connect(self.close)
|
|
|
|
self.bad_db_download_msg = QMessageBox(self)
|
|
self.bad_db_download_msg.setWindowTitle("Something wrong")
|
|
self.bad_db_download_msg.setText("""Something went wrong with the downaload.
|
|
Check your internet connection and try again.""")
|
|
self.bad_db_download_msg.finished.connect(self.close)
|
|
|
|
self.bad_file_msg = QMessageBox(self)
|
|
self.bad_file_msg.setWindowTitle("Bad file detected")
|
|
self.bad_file_msg.setText("""The downloaded file seems to be corrupted.
|
|
The old database has not been deleted and
|
|
the downloaded file has been discarded.""")
|
|
self.bad_file_msg.finished.connect(self.close)
|
|
|
|
self.download_thread = DownloadThread()
|
|
self.download_thread.finished.connect(self.wait_close)
|
|
|
|
self.cancel_btn.clicked.connect(self.terminate_process)
|
|
|
|
def show_no_connection_warning(self):
|
|
self.bad_db_download_msg.setText(f"""Unable to correctly download the database.
|
|
Reason: {self.download_thread.reason}""")
|
|
self.no_internet_msg.show()
|
|
self.everything_ok = False
|
|
|
|
def show_bad_download_warning(self):
|
|
self.bad_db_download_msg.show()
|
|
self.everything_ok = False
|
|
|
|
@pyqtSlot()
|
|
def terminate_process(self):
|
|
if self.download_thread.isRunning():
|
|
self.download_thread.terminate()
|
|
self.download_thread.wait()
|
|
self.close()
|
|
|
|
@pyqtSlot()
|
|
def wait_close(self):
|
|
if self.download_thread.status == ThreadStatus.ok:
|
|
self.close()
|
|
elif self.download_thread.status == ThreadStatus.no_connection_err:
|
|
self.show_no_connection_warning()
|
|
elif self.download_thread.status == ThreadStatus.bad_download_err:
|
|
self.show_bad_download_warning
|
|
else:
|
|
self.close()
|
|
|
|
def reject(self):
|
|
if self.download_thread.isRunning():
|
|
self.download_thread.terminate()
|
|
self.download_thread.wait()
|
|
super().reject()
|