Implement undo/redo

1 files changed, 10 insertions, 0 deletions

diff --git a/labyrinthactivity.py b/labyrinthactivity.py
index 27a10f1..d802893 100644
--- a/labyrinthactivity.py
+++ b/labyrinthactivity.py
@@ -29,6 +29,8 @@ class LabyrinthActivity(activity.Activity):
 
         self._edit_toolbar = activity.EditToolbar()
         toolbox.add_toolbar(_('Edit'), self._edit_toolbar)
+        self._edit_toolbar.undo.child.connect('clicked', self.__undo_cb)
+        self._edit_toolbar.redo.child.connect('clicked', self.__redo_cb)
         self._edit_toolbar.show()
 
         self.undo = UndoManager.UndoManager (self,
@@ -51,6 +53,14 @@ class LabyrinthActivity(activity.Activity):
 
         self.set_focus_child (self.MainArea)
 
+        self.undo.unblock()
+
+    def __undo_cb(self, button):
+        self.undo.undo_action(None)
+
+    def __redo_cb(self, button):
+        self.undo.redo_action(None)
+
     def main_area_focus_cb (self, arg, event, extended = False):
         self.MainArea.grab_focus ()