diff --git a/src/View/Scenarios/GraphWidget.py b/src/View/Scenarios/GraphWidget.py index 8b8ea3fa..bba78edd 100644 --- a/src/View/Scenarios/GraphWidget.py +++ b/src/View/Scenarios/GraphWidget.py @@ -470,9 +470,16 @@ class GraphWidget(QGraphicsView): self.changeScenario.emit(self.sender()) def new_scenario(self, pos): + must_save = self.dialog_save() + if must_save == "Cancel": + return + def fn(): self._close_other_window() - self._study.save() + + if must_save == "Save": + self._study.save() + self._undo.push( AddScenariosCommand( self._study, @@ -500,9 +507,16 @@ class GraphWidget(QGraphicsView): self.changeScenario.emit(self.sender()) def duplicate_scenario(self, item): + must_save = self.dialog_save() + if must_save == "Cancel": + return + def fn(): self._close_other_window() - # self._study.save() + + if must_save == "Save": + self._study.save() + self._undo.push( DuplicateScenariosCommand( self._study,