cancel addition properly

plotlyst · Sep 2, 2023 · c2952de · c2952de
1 parent 220a3c6
commit c2952de
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/main/python/plotlyst/view/widget/graphics.py b/src/main/python/plotlyst/view/widget/graphics.py
@@ -447,6 +447,7 @@ def keyPressEvent(self, event: QKeyEvent) -> None:
                 self.endLink()
             elif self.isAdditionMode():
                 self.cancelItemAddition.emit()
+                self.endAdditionMode()
             else:
                 self.clearSelection()
         elif event.key() == Qt.Key.Key_Delete or event.key() == Qt.Key.Key_Backspace:
@@ -484,6 +485,7 @@ def mouseReleaseEvent(self, event: 'QGraphicsSceneMouseEvent') -> None:
                 self.endLink()
         elif self.isAdditionMode() and event.button() & Qt.MouseButton.RightButton:
             self.cancelItemAddition.emit()
+            self.endAdditionMode()
         elif self._selectionMode and event.button() & Qt.MouseButton.LeftButton:
             self._selectionMode = False
             self._selectionRect.setVisible(False)