Insert a new value in the selected position

1 files changed, 12 insertions, 3 deletions

diff --git a/activity.py b/activity.py
index 4ba7af3..8a5a387 100644
--- a/activity.py
+++ b/activity.py
@@ -239,8 +239,8 @@ class SimpleGraph(activity.Activity):
         self.options.set_visible(False)
 
     def _add_value(self, widget, label="", value="0.0"):
-        self.labels_and_values.add_value(label, value)
-        self.chart_data.append((label, float(value)))
+        pos = self.labels_and_values.add_value(label, value)
+        self.chart_data.insert(pos, (label, float(value)))
         self._update_chart_data()
 
     def _remove_value(self, widget):
@@ -475,7 +475,14 @@ class ChartData(gtk.TreeView):
         self.show_all()
 
     def add_value(self, label, value):
-        iter = self.model.append([label, value])
+        selected = self.get_selection().get_selected()[1]
+        if not selected:
+            path = 0
+
+        elif selected:
+            path = self.model.get_path(selected)[0]
+
+        iter = self.model.insert(path, [label, value])
 
         self.set_cursor(self.model.get_path(iter),
                         self.get_column(1),
@@ -483,6 +490,8 @@ class ChartData(gtk.TreeView):
 
         logger.info("Added: %s, Value: %s" % (label, value))
 
+        return path
+
     def remove_selected_value(self):
         path, column = self.get_cursor()
         path = path[0]