Make MindMapModel inherit from gtk.TreeStore

1 files changed, 8 insertions, 8 deletions

diff --git a/treeview.py b/treeview.py
index cec8562..e255fa8 100644
--- a/treeview.py
+++ b/treeview.py
@@ -69,18 +69,18 @@ class TreeView(gtk.ScrolledWindow):
         self.set_size_request(200, -1)
 
     def __name_edited_cb(self, cell_renderer, path, new_text):
-        thought = self._tree_view.props.model.get_thought(path[0])
-        thought.name = new_text
+        model = self._tree_view.props.model
+        model.set(model.get_iter(path), 1, new_text)
 
     def __x_edited_cb(self, cell_renderer, path, new_text):
-        thought = self._tree_view.props.model.get_thought(path[0])
-        thought.x = int(new_text)
+        model = self._tree_view.props.model
+        model.set(model.get_iter(path), 2, float(new_text))
 
     def __y_edited_cb(self, cell_renderer, path, new_text):
-        thought = self._tree_view.props.model.get_thought(path[0])
-        thought.y = int(new_text)
+        model = self._tree_view.props.model
+        model.set(model.get_iter(path), 3, float(new_text))
 
     def __color_edited_cb(self, cell_renderer, path, new_text):
-        thought = self._tree_view.props.model.get_thought(path[0])
-        thought.color = new_text
+        model = self._tree_view.props.model
+        model.set(model.get_iter(path), 4, new_text)