Added "Save dataset" option to navigator context menu.

fluents/navigator.py

@@ -35,7 +35,7 @@ class NavigatorView (gtk.TreeView):
         # Activate context menu
         self.menu = NavigatorMenu(self)
         self.connect('popup_menu', self.popup_menu)
-        self.connect('button_press_event', self.mouse_popup_menu)
+        self.connect('button_press_event', self.on_mouse_event)
 
         self.textrenderer = textrenderer = gtk.CellRendererText()
         pixbufrenderer = gtk.CellRendererPixbuf()
@@ -160,7 +160,20 @@ class NavigatorView (gtk.TreeView):
     def popup_menu(self, *rest):
         self.menu.popup(None, None, None, 0, 0)
 
-    def mouse_popup_menu(self, widget, event):
+    def on_mouse_event(self, widget, event):
+        path = widget.get_path_at_pos(int(event.x), int(event.y))
+
+        if path:
+            iter = self.data_tree.get_iter(path[0])
+            obj = self.data_tree.get_value(iter, 2)
+        else:
+            obj = None
+
+        if isinstance(obj, dataset.Dataset):
+            self.menu.set_dataset(obj)
+        else:
+            self.menu.set_dataset(None)
+
         if event.button == 3:
             self.menu.popup(None, None, None, event.button, event.time)
 
@@ -169,27 +182,34 @@ class NavigatorMenu(gtk.Menu):
     def __init__(self, navigator):
         gtk.Menu.__init__(self)
         self.navigaotor = navigator
+        self.dataset = None
         
         # Populate menu
         self.load_item = gtk.MenuItem('Load dataset')
-        self.load_item.connect('activate', on_load_dataset, navigator)
+        self.load_item.connect('activate', self.on_load_dataset, navigator)
         self.append(self.load_item)
         self.load_item.show()
 
         self.save_item = gtk.MenuItem('Save dataset')
-        self.save_item.connect('activate', on_save_dataset, navigator)
+        self.save_item.connect('activate', self.on_save_dataset, navigator)
         self.append(self.save_item)
+        self.save_item.show()
+
+    def set_dataset(self, ds):
+        self.dataset = ds
+        if ds == None:
+            self.save_item.set_property('sensitive', False)
+        else:
+            self.save_item.set_property('sensitive', True)
         
 
-def on_load_dataset(item, navigator):
+    def on_load_dataset(self, item, navigator):
         dialog = gtk.FileChooserDialog('Load dataset')
         dialog.add_buttons(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL, gtk.STOCK_OPEN, gtk.RESPONSE_OK)
         retval = dialog.run()
         if retval in [gtk.RESPONSE_CANCEL, gtk.RESPONSE_DELETE_EVENT]:
-        print "Cancelled"
+            pass
         elif retval == gtk.RESPONSE_OK:
-        print "OK"
-        print dialog.get_filename()
             fd = open(dialog.get_filename())
             ds = dataset.read_ftsv(fd)
             fd.close()
@@ -208,6 +228,19 @@ def on_load_dataset(item, navigator):
             print "unknown; ", retval 
         dialog.destroy()
 
-def on_save_dataset(item):
+    def on_save_dataset(self, item, navigator):
-    pass  
+        dialog = gtk.FileChooserDialog('Save dataset')
+        dialog.add_buttons(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL, gtk.STOCK_SAVE, gtk.RESPONSE_OK)
+        retval = dialog.run()
+        if retval in [gtk.RESPONSE_CANCEL, gtk.RESPONSE_DELETE_EVENT]:
+            print "Cancelled"
+        elif retval == gtk.RESPONSE_OK:
+            print "OK"
+            print dialog.get_filename()
+            fd = open(dialog.get_filename(), 'w')
+            dataset.write_ftsv(fd, self.dataset)
+            fd.close()
+        else:
+            print "unknown; ", retval 
+        dialog.destroy()