fix run button after failed schema import

opengisch · Dec 23, 2022 · 6562631 · 6562631
1 parent 0a01b57
commit 6562631
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/QgisModelBaker/gui/panel/session_panel.py b/QgisModelBaker/gui/panel/session_panel.py
@@ -288,12 +288,17 @@ def run(self, edited_command=None):
                     self.progress_bar.setValue(0)
                     if not self.db_action_type == DbActionType.GENERATE:
                         self.set_button_to_create_without_constraints()
+                    else:
+                        self.set_button_to_create()
                     self.is_running = False
                     return False
             except JavaNotFoundError as e:
                 self.print_info.emit(e.error_string, LogColor.COLOR_FAIL)
                 self.progress_bar.setValue(0)
-                self.set_button_to_create_without_constraints()
+                if not self.db_action_type == DbActionType.GENERATE:
+                    self.set_button_to_create_without_constraints()
+                else:
+                    self.set_button_to_create()
                 self.is_running = False
                 return False