Add custom widgets preference option.

Author: alejandroautalan

pygubu/__init__.py

@@ -1,8 +1,9 @@
 from __future__ import unicode_literals
 
-__all__ = ['Builder', 'TkApplication']
+__all__ = ['Builder', 'TkApplication', 'BuilderObject', 'register_widget']
 
 from pygubu.builder import Builder
+from pygubu.builder.builderobject import BuilderObject, register_widget
 
 
 __version__ = '0.9.7.2'

pygubu/builder/__init__.py

@@ -326,7 +326,20 @@ def _import_class(self, modulename):
         if modulename.startswith('ttk.'):
             importlib.import_module('pygubu.builder.ttkstdwidgets')
         else:
-            importlib.import_module(modulename)
+            # Import module as full path
+            try:
+                importlib.import_module(modulename)
+            except ImportError as e:
+                # A single module can contain various widgets
+                # try to import the first part of the path
+                if '.' in modulename:
+                    first, last = modulename.rsplit('.', 1)
+                    try:
+                        importlib.import_module(first)
+                    except ImportError as e:
+                        importlib.import_module(last)
+                else:
+                    raise e
 
     def _realize(self, master, element):
         """Builds a widget from xml element using master as parent."""

pygubudesigner/main.py

@@ -47,7 +47,7 @@
 from pygubu.widgets.accordionframe import AccordionFrame
 from pygubu.widgets.autoarrangeframe import AutoArrangeFrame
 import pygubu.widgets.simpletooltip as tooltip
-from pygubudesigner.preferences import PreferencesUI
+from pygubudesigner.preferences import PreferencesUI, get_custom_widgets
 
 #translator function
 _ = translator
@@ -64,6 +64,15 @@
         modulename = "{0}.{1}".format(widgets_pkg, mfile[:-3])
         importlib.import_module(modulename)
 
+#initialize custom widgets
+for path in get_custom_widgets():
+    dirname, fname = os.path.split(path)
+    if fname.endswith('.py'):
+        if dirname not in sys.path:
+            sys.path.append(dirname)
+        modulename = fname[:-3]
+        importlib.import_module(modulename)
+
 #Initialize images
 DESIGNER_DIR = os.path.dirname(os.path.abspath(__file__))
 IMAGES_DIR = os.path.join(DESIGNER_DIR, "images")

pygubudesigner/preferences.py

@@ -9,10 +9,49 @@
     import ttk
     import tkMessageBox as messagebox
     import tkFileDialog as filedialog
+try:
+    import ConfigParser as configparser
+except:
+    import configparser
+
+from appdirs import AppDirs
 import pygubu
 
 
 FILE_PATH = os.path.dirname(os.path.abspath(__file__))
+dirs = AppDirs('pygubu-designer')
+CONFIG_FILE = os.path.join(dirs.user_data_dir, 'config')
+
+CUSTOM_WIDGETS = 'CUSTOM_WIDGETS'
+config = configparser.SafeConfigParser()
+config.add_section(CUSTOM_WIDGETS)
+
+
+def initialize_configfile():
+    if not os.path.exists(dirs.user_data_dir):
+        os.makedirs(dirs.user_data_dir)
+    if not os.path.exists(CONFIG_FILE):
+        with open(CONFIG_FILE, 'w') as configfile:
+            config.write(configfile)
+
+def save_configfile():
+    with open(CONFIG_FILE, 'w') as configfile:
+        config.write(configfile)
+
+def load_configfile():
+    if not os.path.exists(CONFIG_FILE):
+        initialize_configfile()
+    else:
+        config.read(CONFIG_FILE)
+
+def get_custom_widgets():
+    paths = []
+    for k, p in config.items(CUSTOM_WIDGETS):
+        paths.append(p)
+    return paths
+
+# Get user configuration
+load_configfile()
 
 
 class PreferencesUI(object):
@@ -21,8 +60,8 @@ def __init__(self, master, translator=None):
         self.master = master
         self.translator = translator
         self.dialog = None
-        self.modulestv = None
         self._create_preferences_dialog()
+        self._load_options()
 
     def _create_preferences_dialog(self):
         builder = pygubu.Builder(self.translator)
@@ -31,21 +70,47 @@ def _create_preferences_dialog(self):
 
         top = self.master.winfo_toplevel()
         self.dialog = dialog = builder.get_object('preferences', top)
-        self.modulestv = builder.get_object('modules')
-
+        self.cwtv = builder.get_object('cwtv')
+        self.path_remove = builder.get_object('path_remove')
         builder.connect_callbacks(self)
+        
+    def _load_options(self):
+        for k, p in config.items(CUSTOM_WIDGETS):
+            self.cwtv.insert('', 'end', text=p)
+        self._configure_path_remove()
+
+    def _save_options(self):
+        config.remove_section(CUSTOM_WIDGETS)
+        config.add_section(CUSTOM_WIDGETS)
+        paths = []
+        for iid in self.cwtv.get_children():
+            txt = self.cwtv.item(iid, 'text')
+            paths.append(txt)
+        for j, p in enumerate(paths):
+            config.set(CUSTOM_WIDGETS, 'w{0}'.format(j), p)
+        save_configfile()
 
-    def on_btnclose_clicked(self):
+    def _configure_path_remove(self):
+        if len(self.cwtv.get_children()):
+            self.path_remove.configure(state='normal')
+        else:
+            self.path_remove.configure(state='disabled')
+        
+    def on_dialog_close(self, event=None):
+        self._save_options()
         self.dialog.close()
 
     def on_pathremove_clicked(self):
-        pass
+        items = self.cwtv.selection()
+        self.cwtv.delete(*items)
+        self._configure_path_remove()
 
     def on_pathadd_clicked(self):
         options = {
             'defaultextension': '.py',
             'filetypes': ((_('Python module'), '*.py'), (_('All'), '*.*'))}
         fname = filedialog.askopenfilename(**options)
         if fname:
-            self.modulestv.insert('', tk.END, text=fname)
+            self.cwtv.insert('', tk.END, text=fname)
+            self._configure_path_remove()