implemented interface to save data to a hdf5 file

philipstarkey · philipstarkey · commit b6cc9d3ff869 · 2012-06-08T05:37:19.000Z
implemented interface to call callback functions when settings are changed/saved.
diff --git a/__init__.py b/__init__.py
@@ -1,35 +1,62 @@
 import os
 import gtk
-
+import h5py
 
 # Create a generic interface for displaying pages of settings
 
 class Settings(object):
     
-    def __init__(self,parent = None,page_classes = []):
+    def __init__(self,storage='hdf5',file=None,parent = None,page_classes = []):
         self.pages = {}
+        self.instantiated_pages = {}
         self.dialog_open = False
         self.parent = parent
+        self.storage = storage
+        self.file = file
+        self.callback_list = []
+        
+        if not self.file:
+            raise Exception('You must specify a file to load/save preferences from')
+        
         for c in page_classes:
             self.add_settings_interface(c)
-        
+            
     # This function can be called to add a interface
     # Each one of these will display as a seperate page in the settings window
     # You can not add a class more than once!
     # Classes must have unique Class.name attributes! (This might change later...)
     def add_settings_interface(self,setting_class):
         if setting_class.name in self.pages:
             return False
-        self.pages[setting_class.name] = setting_class()   
+                   
+        self.pages[setting_class.name] = setting_class(self.load(setting_class.__name__))   
         return True
         
+    def load(self,name):
+        if self.storage == 'hdf5':
+            with h5py.File(self.file,'r+') as h5file: 
+                # does the settings group exist?
+                if 'preferences' not in h5file:
+                    h5file['/'].create_group('preferences')
+                    
+                # is there an entry for this preference type?
+                group = h5file['/preferences']
+                if name not in group.attrs:
+                    group.attrs[name] = repr({})
+                data = eval(group.attrs[name])                    
+            return data    
+        else:
+            raise Exception("the Settings module cannot handle the storage type: %s"%str(self.storage))
+        
     # A simple interface for accessing values in the settings interface
     def get_value(self,settings_class,value_name):
         return self.pages[settings_class.name].get_value(value_name)
         
     # goto_page should be the CLASS which you wish to go to!
     def create_dialog(self,goto_page=None):
         if not self.dialog_open:
+            self.instantiated_pages = {}
+            
             builder = gtk.Builder()
             builder.add_from_file(os.path.join(os.path.dirname(os.path.realpath(__file__)),'settings_interface.glade'))
             builder.connect_signals(self)
@@ -40,7 +67,21 @@ def create_dialog(self,goto_page=None):
             #sorted(a.items(),key=lambda x: x[1])
             set_page = None
             for name, c in sorted(self.pages.items()):
-                page = c.create_dialog(self.notebook)
+                page,icon = c.create_dialog(self.notebook)
+                
+                # save page
+                self.instantiated_pages[c.__class__] = page
+                
+                # Create label
+                if isinstance(icon,gtk.Image):
+                    # use their icon
+                    pass
+                else:
+                    # use default icon
+                    pass
+                    
+                tab_label = gtk.Label(c.name)
+                self.notebook.append_page(page,tab_label)
                 
                 if goto_page and isinstance(c,goto_page):
                     # this is the page we want to go to!
@@ -57,12 +98,34 @@ def create_dialog(self,goto_page=None):
             self.window.show()
             self.dialog_open = True
         else:
+            if goto_page and goto_page in self.instantiated_pages:
+                
+                self.notebook.set_current_page(self.notebook.page_num(self.instantiated_pages[goto_page]))
+                
             self.window.present()
+    
+    def register_callback(self,callback):
+        self.callback_list.append(callback)
         
+    def remove_callback(self,callback):
+        self.callback_list.remove(callback)
+    
     def on_save(self,widget):
         # Save the settings
-        for page in self.pages.values():
-            page.save()    
+        if self.storage == 'hdf5':
+            with h5py.File(self.file,'r+') as h5file:
+                group = h5file['/preferences']
+                for page in self.pages.values():
+                    group.attrs[page.__class__.__name__] = repr(page.save()) 
+        else:
+            # this should never happen as the exception will have been raised on load!
+            pass
+            
+        # run callback functions!
+        # Notifies other areas of the program that settings have changed
+        for callback in self.callback_list:
+            callback()
+        
         self.close()
             
     def on_cancel(self,widget):
