Skip to content

Commit 5dec0cf

Browse files
committed
Message box
1 parent c7ccb5b commit 5dec0cf

File tree

2 files changed

+21
-10
lines changed

2 files changed

+21
-10
lines changed

oqtopus/core/package_prepare_task.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -90,15 +90,16 @@ def __download_module_version(self, module_version):
9090
logging.info(f"Downloading from '{url}' to '{self.zip_file}'")
9191
data_size = 0
9292
with open(self.zip_file, "wb") as file:
93+
next_emit_threshold = 5 * 1024 * 1024 # 5MB threshold
9394
for data in response.iter_content(chunk_size=None):
95+
file.write(data)
9496

9597
self.__checkForCanceled()
9698

9799
data_size += len(data)
98-
# print(f"Recived '{data_size}' bytes")
99-
# self.signalPackagingProgress.emit(data_size)
100-
101-
file.write(data)
100+
if data_size >= next_emit_threshold: # Emit signal when threshold is exceeded
101+
self.signalPackagingProgress.emit(data_size)
102+
next_emit_threshold += 5 * 1024 * 1024 # Update to the next threshold
102103

103104
def __extract_zip_file(self, zip_file):
104105
temp_dir = PluginUtils.plugin_temp_path()

oqtopus/gui/main_dialog.py

Lines changed: 16 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ class MainDialog(QDialog, DIALOG_UI):
5353
def __init__(self, modules_registry, parent=None):
5454
QDialog.__init__(self, parent)
5555
self.setupUi(self)
56-
self.buttonBox.rejected.connect(self.accept)
56+
self.buttonBox.rejected.connect(self.__closeDialog)
5757
self.buttonBox.helpRequested.connect(self.__helpRequested)
5858

5959
self.__modules_registry = modules_registry
@@ -79,9 +79,6 @@ def __init__(self, modules_registry, parent=None):
7979
self.__packagePrepareTaskProgress
8080
)
8181

82-
def __helpRequested(self):
83-
QDesktopServices.openUrl(QUrl("https://github.com/oqtopus/Oqtopus"))
84-
8582
def __initGuiModules(self):
8683
self.module_module_comboBox.clear()
8784
self.module_module_comboBox.addItem(self.tr("Please select a module"), None)
@@ -134,6 +131,16 @@ def __initGuiModuleInfo(self):
134131
self.moduleInfo_install_pushButton.clicked.connect(self.__installModuleClicked)
135132
self.moduleInfo_upgrade_pushButton.clicked.connect(self.__upgradeModuleClicked)
136133

134+
def __closeDialog(self):
135+
if self.__packagePrepareTask.isRunning():
136+
self.__packagePrepareTask.cancel()
137+
self.__packagePrepareTask.wait()
138+
139+
self.accept()
140+
141+
def __helpRequested(self):
142+
QDesktopServices.openUrl(QUrl("https://github.com/oqtopus/Oqtopus"))
143+
137144
def __loadDatabaseInformations(self):
138145
self.db_servicesConfigFilePath_label.setText(pgserviceparser.conf_path().as_posix())
139146

@@ -278,11 +285,14 @@ def __packagePrepareTaskFinished(self):
278285
self.__data_model_dir = os.path.join(self.__packagePrepareTask.package_dir, "datamodel")
279286
pumConfigFilename = os.path.join(self.__data_model_dir, ".pum.yaml")
280287
if not os.path.exists(pumConfigFilename):
281-
raise Exception(
288+
QMessageBox.critical(
289+
self,
290+
self.tr("Error"),
282291
self.tr(
283292
f"The selected file '{self.__packagePrepareTask.zip_file}' does not contain a valid .pum.yaml file."
284-
)
293+
),
285294
)
295+
return
286296

287297
self.__pum_config = PumConfig.from_yaml(pumConfigFilename)
288298

0 commit comments

Comments
 (0)