1241 lines
		
	
	
		
			50 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			1241 lines
		
	
	
		
			50 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| #
 | |
| # Gramps - a GTK+/GNOME based genealogy program
 | |
| #
 | |
| # Copyright (C) 2000-2006  Donald N. Allingham
 | |
| # Copyright (C) 2008       Brian G. Matherly
 | |
| # Copyright (C) 2010       Jakim Friant
 | |
| # Copyright (C) 2011       Paul Franklin
 | |
| #
 | |
| # This program is free software; you can redistribute it and/or modify
 | |
| # it under the terms of the GNU General Public License as published by
 | |
| # the Free Software Foundation; either version 2 of the License, or
 | |
| # (at your option) any later version.
 | |
| #
 | |
| # This program is distributed in the hope that it will be useful,
 | |
| # but WITHOUT ANY WARRANTY; without even the implied warranty of
 | |
| # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | |
| # GNU General Public License for more details.
 | |
| #
 | |
| # You should have received a copy of the GNU General Public License
 | |
| # along with this program; if not, write to the Free Software
 | |
| # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 | |
| #
 | |
| 
 | |
| #-------------------------------------------------------------------------
 | |
| #
 | |
| # Python modules
 | |
| #
 | |
| #-------------------------------------------------------------------------
 | |
| from __future__ import print_function
 | |
| 
 | |
| import traceback
 | |
| import os
 | |
| import sys
 | |
| 
 | |
| #-------------------------------------------------------------------------
 | |
| #
 | |
| # set up logging
 | |
| #
 | |
| #-------------------------------------------------------------------------
 | |
| import logging
 | |
| LOG = logging.getLogger(".gui.plug")
 | |
| 
 | |
| #-------------------------------------------------------------------------
 | |
| #
 | |
| # GTK modules
 | |
| #
 | |
| #-------------------------------------------------------------------------
 | |
| from gi.repository import Gtk
 | |
| from gi.repository import Gdk
 | |
| from gi.repository import Pango
 | |
| from gi.repository import GObject
 | |
| 
 | |
| #-------------------------------------------------------------------------
 | |
| #
 | |
| # GRAMPS modules
 | |
| #
 | |
| #-------------------------------------------------------------------------
 | |
| from gramps.gen.const import GRAMPS_LOCALE as glocale
 | |
| _ = glocale.translation.gettext
 | |
| ngettext = glocale.translation.ngettext # else "nearby" comments are ignored
 | |
| from ..managedwindow import ManagedWindow
 | |
| from gramps.gen.errors import UnavailableError, WindowActiveError
 | |
| from gramps.gen.plug import PluginRegister, PTYPE_STR, load_addon_file
 | |
| from ..utils import open_file_with_default_application
 | |
| from ..pluginmanager import GuiPluginManager
 | |
| from . import tool
 | |
| from ._guioptions import add_gui_options
 | |
| from ..dialog import InfoDialog, OkDialog
 | |
| from ..editors import EditPerson
 | |
| from ..glade import Glade
 | |
| from ..listmodel import ListModel, NOSORT, TOGGLE
 | |
| from gramps.gen.const import URL_WIKISTRING, USER_HOME, WIKI_EXTRAPLUGINS_RAWDATA
 | |
| from gramps.gen.constfunc import win, conv_to_unicode
 | |
| from gramps.gen.config import config
 | |
| from ..widgets.progressdialog import (LongOpStatus, ProgressMonitor,
 | |
|                                       GtkProgressDialog)
 | |
| 
 | |
| def display_message(message):
 | |
|     """
 | |
|     A default callback for displaying messages.
 | |
|     """
 | |
|     print(message)
 | |
| 
 | |
| #-------------------------------------------------------------------------
 | |
| #
 | |
| # PluginStatus: overview of all plugins
 | |
| #
 | |
| #-------------------------------------------------------------------------
 | |
| class PluginStatus(ManagedWindow):
 | |
|     """Displays a dialog showing the status of loaded plugins"""
 | |
|     HIDDEN = '<span color="red">%s</span>' % _('Hidden')
 | |
|     AVAILABLE = '<span weight="bold" color="blue">%s</span>'\
 | |
|                                 % _('Visible')
 | |
|     
 | |
|     def __init__(self, dbstate, uistate, track=[]):
 | |
|         self.dbstate = dbstate
 | |
|         self.__uistate = uistate
 | |
|         self.title = _("Plugin Manager")
 | |
|         ManagedWindow.__init__(self, uistate, track,
 | |
|                                              self.__class__)
 | |
| 
 | |
|         self.__pmgr = GuiPluginManager.get_instance()
 | |
|         self.__preg = PluginRegister.get_instance()
 | |
|         self.set_window(Gtk.Dialog("", uistate.window,
 | |
|                                    Gtk.DialogFlags.DESTROY_WITH_PARENT,
 | |
|                                    (Gtk.STOCK_CLOSE, Gtk.ResponseType.CLOSE)),
 | |
|                         None, self.title)
 | |
|         self.window.set_size_request(750, 400)
 | |
|         self.window.connect('response', self.close)
 | |
|         
 | |
|         notebook = Gtk.Notebook()
 | |
|         
 | |
|         #first page with all registered plugins
 | |
|         vbox_reg = Gtk.VBox()
 | |
|         scrolled_window_reg =  Gtk.ScrolledWindow()
 | |
|         self.list_reg =  Gtk.TreeView()
 | |
|         #  model: plugintype, hidden, pluginname, plugindescr, pluginid
 | |
|         self.model_reg = Gtk.ListStore(GObject.TYPE_STRING, GObject.TYPE_STRING, 
 | |
|                 GObject.TYPE_STRING, GObject.TYPE_STRING, GObject.TYPE_STRING)
 | |
|         self.selection_reg = self.list_reg.get_selection()
 | |
|         self.list_reg.set_model(self.model_reg)
 | |
|         self.list_reg.set_rules_hint(True)
 | |
|         self.list_reg.connect('button-press-event', self.button_press_reg)
 | |
|         col0_reg = Gtk.TreeViewColumn(_('Type'), Gtk.CellRendererText(), text=0)
 | |
|         col0_reg.set_sort_column_id(0)
 | |
|         col0_reg.set_resizable(True)
 | |
|         self.list_reg.append_column(col0_reg)
 | |
|         col = Gtk.TreeViewColumn(_('Status'), Gtk.CellRendererText(), markup=1)
 | |
|         col.set_sort_column_id(1)
 | |
|         self.list_reg.append_column(col)
 | |
|         col2_reg = Gtk.TreeViewColumn(_('Name'), Gtk.CellRendererText(), text=2)
 | |
|         col2_reg.set_sort_column_id(2)
 | |
|         col2_reg.set_resizable(True)
 | |
|         self.list_reg.append_column(col2_reg)
 | |
|         col = Gtk.TreeViewColumn(_('Description'), Gtk.CellRendererText(), text=3)
 | |
|         col.set_sort_column_id(3)
 | |
|         col.set_resizable(True)
 | |
|         self.list_reg.append_column(col)
 | |
|         self.list_reg.set_search_column(2)
 | |
| 
 | |
|         scrolled_window_reg.add(self.list_reg)
 | |
|         vbox_reg.pack_start(scrolled_window_reg, True, True, 0)
 | |
|         hbutbox = Gtk.HButtonBox()
 | |
|         hbutbox.set_layout(Gtk.ButtonBoxStyle.SPREAD)
 | |
|         self.__info_btn = Gtk.Button(_("Info"))
 | |
|         hbutbox.add(self.__info_btn)
 | |
|         self.__info_btn.connect('clicked', self.__info, self.list_reg, 4) # id_col
 | |
|         self.__hide_btn = Gtk.Button(_("Hide/Unhide"))
 | |
|         hbutbox.add(self.__hide_btn)
 | |
|         self.__hide_btn.connect('clicked', self.__hide, 
 | |
|                                 self.list_reg, 4, 1) # list, id_col, hide_col
 | |
|         if __debug__:
 | |
|             self.__edit_btn = Gtk.Button(_("Edit"))
 | |
|             hbutbox.add(self.__edit_btn)
 | |
|             self.__edit_btn.connect('clicked', self.__edit, self.list_reg, 4) # id_col
 | |
|             self.__load_btn = Gtk.Button(_("Load"))
 | |
|             hbutbox.add(self.__load_btn)
 | |
|             self.__load_btn.connect('clicked', self.__load, self.list_reg, 4) # id_col
 | |
|         vbox_reg.pack_start(hbutbox, False, False, 0)
 | |
|         
 | |
|         notebook.append_page(vbox_reg, 
 | |
|                              tab_label=Gtk.Label(label=_('Registered Plugins')))
 | |
|         
 | |
|         #second page with loaded plugins
 | |
|         vbox_loaded = Gtk.VBox()
 | |
|         scrolled_window = Gtk.ScrolledWindow()
 | |
|         self.list = Gtk.TreeView()
 | |
|         self.model = Gtk.ListStore(GObject.TYPE_STRING, GObject.TYPE_STRING, 
 | |
|                                    GObject.TYPE_STRING, object, 
 | |
|                                    GObject.TYPE_STRING, GObject.TYPE_STRING)
 | |
|         self.selection = self.list.get_selection()
 | |
|         self.list.set_model(self.model)
 | |
|         self.list.set_rules_hint(True)
 | |
|         self.list.connect('button-press-event', self.button_press)
 | |
|         self.list.connect('cursor-changed', self.cursor_changed)
 | |
|         col = Gtk.TreeViewColumn(_('Loaded'), Gtk.CellRendererText(),
 | |
|                                  markup=0)
 | |
|         col.set_sort_column_id(0)
 | |
|         col.set_resizable(True)
 | |
|         self.list.append_column(col)
 | |
|         col1 = Gtk.TreeViewColumn(_('File'), Gtk.CellRendererText(), 
 | |
|                                   text=1)
 | |
|         col1.set_sort_column_id(1)
 | |
|         col1.set_resizable(True)
 | |
|         self.list.append_column(col1)
 | |
|         col = Gtk.TreeViewColumn(_('Status'), Gtk.CellRendererText(), 
 | |
|                                  markup=5)
 | |
|         col.set_sort_column_id(5)
 | |
|         self.list.append_column(col)
 | |
|         col2 = Gtk.TreeViewColumn(_('Message'), Gtk.CellRendererText(), text=2)
 | |
|         col2.set_sort_column_id(2)
 | |
|         col2.set_resizable(True)
 | |
|         self.list.append_column(col2)
 | |
|         self.list.set_search_column(1)
 | |
| 
 | |
|         scrolled_window.add(self.list)
 | |
|         vbox_loaded.pack_start(scrolled_window, True, True, 0)
 | |
|         hbutbox = Gtk.HButtonBox()
 | |
|         hbutbox.set_layout(Gtk.ButtonBoxStyle.SPREAD)
 | |
|         self.__info_btn = Gtk.Button(_("Info"))
 | |
|         hbutbox.add(self.__info_btn)
 | |
|         self.__info_btn.connect('clicked', self.__info, self.list, 4) # id_col
 | |
|         self.__hide_btn = Gtk.Button(_("Hide/Unhide"))
 | |
|         hbutbox.add(self.__hide_btn)
 | |
|         self.__hide_btn.connect('clicked', self.__hide,
 | |
|                                 self.list, 4, 5) # list, id_col, hide_col
 | |
| 
 | |
|         if __debug__:
 | |
|             self.__edit_btn = Gtk.Button(_("Edit"))
 | |
|             hbutbox.add(self.__edit_btn)
 | |
|             self.__edit_btn.connect('clicked', self.__edit, self.list, 4) # id_col
 | |
|             self.__load_btn = Gtk.Button(_("Load"))
 | |
|             self.__load_btn.set_sensitive(False)
 | |
|             hbutbox.add(self.__load_btn)
 | |
|             self.__load_btn.connect('clicked', self.__load, self.list, 4) # id_col
 | |
|         vbox_loaded.pack_start(hbutbox, False, False, 5)
 | |
|         notebook.append_page(vbox_loaded, 
 | |
|                              tab_label=Gtk.Label(label=_('Loaded Plugins')))
 | |
| 
 | |
|         #third page with method to install plugin
 | |
|         install_page = Gtk.VBox()
 | |
|         scrolled_window = Gtk.ScrolledWindow()
 | |
|         self.addon_list = Gtk.TreeView()
 | |
|         # model: help_name, name, ptype, image, desc, use, rating, contact, download, url
 | |
|         self.addon_model = Gtk.ListStore(GObject.TYPE_STRING, 
 | |
|                                          GObject.TYPE_STRING, 
 | |
|                                          GObject.TYPE_STRING, 
 | |
|                                          GObject.TYPE_STRING, 
 | |
|                                          GObject.TYPE_STRING, 
 | |
|                                          GObject.TYPE_STRING, 
 | |
|                                          GObject.TYPE_STRING, 
 | |
|                                          GObject.TYPE_STRING, 
 | |
|                                          GObject.TYPE_STRING, 
 | |
|                                          GObject.TYPE_STRING)
 | |
|         self.addon_list.set_model(self.addon_model)
 | |
|         self.addon_list.set_rules_hint(True)
 | |
|         #self.addon_list.connect('button-press-event', self.button_press)
 | |
|         col = Gtk.TreeViewColumn(_('Addon Name'), Gtk.CellRendererText(),
 | |
|                                  text=1)
 | |
|         col.set_sort_column_id(1)
 | |
|         self.addon_list.append_column(col)
 | |
|         col = Gtk.TreeViewColumn(_('Type'), Gtk.CellRendererText(),
 | |
|                                  text=2)
 | |
|         col.set_sort_column_id(2)
 | |
|         self.addon_list.append_column(col)
 | |
|         col = Gtk.TreeViewColumn(_('Description'), Gtk.CellRendererText(),
 | |
|                                  text=4)
 | |
|         col.set_sort_column_id(4)
 | |
|         self.addon_list.append_column(col)
 | |
|         self.addon_list.connect('cursor-changed', self.button_press_addon)
 | |
| 
 | |
|         install_row = Gtk.HBox()
 | |
|         install_row.pack_start(Gtk.Label(label=_("Path to Addon:")), False, True, 0)
 | |
|         self.install_addon_path = Gtk.Entry()
 | |
| 
 | |
|         button = Gtk.Button()
 | |
|         img = Gtk.Image()
 | |
|         img.set_from_stock(Gtk.STOCK_OPEN, Gtk.IconSize.BUTTON)
 | |
|         button.add(img)
 | |
|         button.connect('clicked', self.__select_file)
 | |
|         install_row.pack_start(self.install_addon_path, True, True, 0)
 | |
|         install_row.pack_start(button, False, False, 0)
 | |
| 
 | |
|         scrolled_window.add(self.addon_list)
 | |
|         install_page.pack_start(scrolled_window, True, True, 0)
 | |
|         #add some spce under the scrollbar
 | |
|         install_page.pack_start(Gtk.Label(label=''), False, False, 0)
 | |
|         #path to addon path line
 | |
|         install_page.pack_start(install_row, False, False, 0)
 | |
| 
 | |
|         hbutbox = Gtk.HButtonBox()
 | |
|         hbutbox.set_layout(Gtk.ButtonBoxStyle.SPREAD)
 | |
|         self.__add_btn = Gtk.Button(_("Install Addon"))
 | |
|         hbutbox.add(self.__add_btn)
 | |
|         self.__add_btn.connect('clicked', self.__get_addon_top)
 | |
|         self.__add_all_btn = Gtk.Button(_("Install All Addons"))
 | |
|         hbutbox.add(self.__add_all_btn)
 | |
|         self.__add_all_btn.connect('clicked', self.__get_all_addons) 
 | |
|         self.__refresh_btn = Gtk.Button(_("Refresh Addon List"))
 | |
|         hbutbox.add(self.__refresh_btn)
 | |
|         self.__refresh_btn.connect('clicked', self.__refresh_addon_list) 
 | |
|         install_page.pack_start(hbutbox, False, True, 5)
 | |
|         # notebook.append_page(install_page, 
 | |
|         #                      tab_label=Gtk.Label(label=_('Install Addons')))
 | |
| 
 | |
|         #add the notebook to the window
 | |
|         self.window.get_content_area().pack_start(notebook, True, True, 0)
 | |
|         
 | |
|         if __debug__:
 | |
|             # Only show the "Reload" button when in debug mode 
 | |
|             # (without -O on the command line)
 | |
|             self.__reload_btn = Gtk.Button(_("Reload"))
 | |
|             self.window.action_area.add(self.__reload_btn)
 | |
|             self.__reload_btn.connect('clicked', self.__reload)
 | |
|         
 | |
|         #obtain hidden plugins from the pluginmanager
 | |
|         self.hidden = self.__pmgr.get_hidden_plugin_ids()
 | |
|         
 | |
|         self.window.show_all()
 | |
|         self.__populate_lists()
 | |
|         self.list_reg.columns_autosize()
 | |
| 
 | |
|     def __refresh_addon_list(self, obj):
 | |
|         """
 | |
|         Reloads the addons from the wiki into the list.
 | |
|         """
 | |
|         if sys.version_info[0] < 3:
 | |
|             from urllib2 import urlopen
 | |
|         else:
 | |
|             from urllib.request import urlopen
 | |
|         from ..utils import ProgressMeter
 | |
|         URL = "%s%s" % (URL_WIKISTRING, WIKI_EXTRAPLUGINS_RAWDATA)
 | |
|         try:
 | |
|             fp = urlopen(URL)
 | |
|         except:
 | |
|             print("Error: cannot open %s" % URL)
 | |
|             return
 | |
|         pm = ProgressMeter(_("Refreshing Addon List"))
 | |
|         pm.set_pass(header=_("Reading gramps-project.org..."))
 | |
|         state = "read"
 | |
|         rows = []
 | |
|         row = []
 | |
|         lines = fp.readlines()
 | |
|         pm.set_pass(total=len(lines), header=_("Reading gramps-project.org..."))
 | |
|         for line in lines:
 | |
|             pm.step()
 | |
|             if line.startswith("|-") or line.startswith("|}"):
 | |
|                 if row != []:
 | |
|                     rows.append(row)
 | |
|                 state = "row"
 | |
|                 row = []
 | |
|             elif state == "row":
 | |
|                 if line.startswith("|"):
 | |
|                     row.append(line[1:].strip())
 | |
|             else:
 | |
|                 state = "read"
 | |
|         fp.close()
 | |
|         rows.sort(key=lambda row: (row[1], row[0]))
 | |
|         self.addon_model.clear()
 | |
|         # clear the config list:
 | |
|         config.get('plugin.addonplugins')[:] = []
 | |
|         pm.set_pass(total=len(rows), header=_("Checking addon..."))
 | |
|         for row in rows:
 | |
|             pm.step()
 | |
|             try:
 | |
|                 # from wiki:
 | |
|                 help_name, ptype, image, desc, use, rating, contact, download = row
 | |
|             except:
 | |
|                 continue
 | |
|             help_url = _("Unknown Help URL")
 | |
|             if help_name.startswith("[[") and help_name.endswith("]]"):
 | |
|                 name = help_name[2:-2]
 | |
|                 if "|" in name:
 | |
|                     help_url, name = name.split("|", 1)
 | |
|             elif help_name.startswith("[") and help_name.endswith("]"):
 | |
|                 name = help_name[1:-1]
 | |
|                 if " " in name:
 | |
|                     help_url, name = name.split(" ", 1)
 | |
|             else:
 | |
|                 name = help_name
 | |
|             url = _("Unknown URL")
 | |
|             if download.startswith("[[") and download.endswith("]]"):
 | |
|                 # Not directly possible to get the URL:
 | |
|                 url = download[2:-2]
 | |
|                 if "|" in url:
 | |
|                     url, text = url.split("|", 1)
 | |
|                 # need to get a page that says where it is:
 | |
|                 fp = urlopen("%s%s%s" % (URL_WIKISTRING, url, 
 | |
|                                 "&action=edit&externaledit=true&mode=file"))
 | |
|                 for line in fp:
 | |
|                     if line.startswith("URL="):
 | |
|                         junk, url = line.split("=", 1)
 | |
|                         break
 | |
|                 fp.close()
 | |
|             elif download.startswith("[") and download.endswith("]"):
 | |
|                 url = download[1:-1]
 | |
|                 if " " in url:
 | |
|                     url, text = url.split(" ", 1)
 | |
|             if (url.endswith(".zip") or 
 | |
|                 url.endswith(".ZIP") or 
 | |
|                 url.endswith(".tar.gz") or 
 | |
|                 url.endswith(".tgz")):
 | |
|                 # Then this is ok:
 | |
|                 self.addon_model.append(row=[help_name, name, ptype, image, desc, use, 
 | |
|                                              rating, contact, download, url])
 | |
|                 config.get('plugin.addonplugins').append([help_name, name, ptype, image, desc, use, 
 | |
|                                                           rating, contact, download, url])
 | |
|         pm.close()
 | |
|         config.save()
 | |
| 
 | |
|     def __get_all_addons(self, obj):
 | |
|         """
 | |
|         Get all addons from the wiki and install them.
 | |
|         """
 | |
|         from ..utils import ProgressMeter
 | |
|         pm = ProgressMeter(_("Install all Addons"), _("Installing..."), message_area=True)
 | |
|         pm.set_pass(total=len(self.addon_model))
 | |
|         for row in self.addon_model:
 | |
|             pm.step()
 | |
|             (help_name, name, ptype, image, desc, use, rating, contact, 
 | |
|              download, url) = row
 | |
|             load_addon_file(url, callback=pm.append_message)
 | |
|         self.uistate.viewmanager.do_reg_plugins(self.dbstate, self.uistate)
 | |
|         pm.message_area_ok.set_sensitive(True)
 | |
|         self.__rebuild_load_list()
 | |
|         self.__rebuild_reg_list()
 | |
| 
 | |
|     def __get_addon_top(self, obj):
 | |
|         """
 | |
|         Toplevel method to get an addon.
 | |
|         """
 | |
|         from ..utils import ProgressMeter
 | |
|         pm = ProgressMeter(_("Installing Addon"), message_area=True)
 | |
|         pm.set_pass(total=2, header=_("Reading gramps-project.org..."))
 | |
|         pm.step()
 | |
|         self.__get_addon(obj, callback=pm.append_message)
 | |
|         pm.step()
 | |
|         pm.message_area_ok.set_sensitive(True)
 | |
| 
 | |
|     def __get_addon(self, obj, callback=display_message):
 | |
|         """
 | |
|         Get an addon from the wiki or file system and install it.
 | |
|         """
 | |
|         path = self.install_addon_path.get_text()
 | |
|         load_addon_file(path, callback)
 | |
|         self.uistate.viewmanager.do_reg_plugins(self.dbstate, self.uistate)
 | |
|         self.__rebuild_load_list()
 | |
|         self.__rebuild_reg_list()
 | |
| 
 | |
|     def __select_file(self, obj):
 | |
|         """
 | |
|         Select a file from the file system.
 | |
|         """
 | |
|         fcd = Gtk.FileChooserDialog(_("Load Addon"), 
 | |
|                                     buttons=(Gtk.STOCK_CANCEL,
 | |
|                                              Gtk.ResponseType.CANCEL,
 | |
|                                              Gtk.STOCK_OPEN,
 | |
|                                              Gtk.ResponseType.OK))
 | |
|         name = self.install_addon_path.get_text()
 | |
|         dir = os.path.dirname(name)
 | |
|         if not os.path.isdir(dir):
 | |
|             dir = USER_HOME
 | |
|             name = ''
 | |
|         elif not os.path.isfile(name):
 | |
|             name = ''
 | |
|         fcd.set_current_folder(dir)
 | |
|         if name:
 | |
|             fcd.set_filename(name)
 | |
| 
 | |
|         status = fcd.run()
 | |
|         if status == Gtk.ResponseType.OK:
 | |
|             path = conv_to_unicode(fcd.get_filename())
 | |
|             if path:
 | |
|                 self.install_addon_path.set_text(path)
 | |
|         fcd.destroy()
 | |
| 
 | |
|     def __populate_lists(self):
 | |
|         """ Build the lists of plugins """
 | |
|         self.__populate_load_list()
 | |
|         self.__populate_reg_list()
 | |
|         self.__populate_addon_list()
 | |
| 
 | |
|     def __populate_addon_list(self):
 | |
|         """
 | |
|         Build the list of addons from the config setting.
 | |
|         """
 | |
|         self.addon_model.clear()
 | |
|         for row in config.get('plugin.addonplugins'):
 | |
|             try:
 | |
|                 help_name, name, ptype, image, desc, use, rating, contact, download, url = row
 | |
|             except:
 | |
|                 continue
 | |
|             self.addon_model.append(row=[help_name, name, ptype, image, desc, use, 
 | |
|                                          rating, contact, download, url])
 | |
| 
 | |
|     def __populate_load_list(self):
 | |
|         """ Build list of loaded plugins"""
 | |
|         fail_list = self.__pmgr.get_fail_list()
 | |
|         
 | |
|         for i in fail_list:
 | |
|             # i = (filename, (exception-type, exception, traceback), pdata)
 | |
|             err = i[1][0]
 | |
|             pdata = i[2]
 | |
|             hidden = pdata.id in self.hidden
 | |
|             if hidden:
 | |
|                 hiddenstr = self.HIDDEN
 | |
|             else:
 | |
|                 hiddenstr = self.AVAILABLE
 | |
|             if err == UnavailableError:
 | |
|                 self.model.append(row=[
 | |
|                     '<span color="blue">%s</span>' % _('Unavailable'),
 | |
|                     i[0], str(i[1][1]), None, pdata.id, hiddenstr])
 | |
|             else:
 | |
|                 self.model.append(row=[
 | |
|                     '<span weight="bold" color="red">%s</span>' % _('Fail'),
 | |
|                     i[0], str(i[1][1]), i[1], pdata.id, hiddenstr])
 | |
| 
 | |
|         success_list = sorted(self.__pmgr.get_success_list(),
 | |
|                               key=lambda x: (x[0], x[2]._get_name()))
 | |
|         for i in success_list:
 | |
|             # i = (filename, module, pdata)
 | |
|             pdata = i[2]
 | |
|             modname = i[1].__name__
 | |
|             hidden = pdata.id in self.hidden
 | |
|             if hidden:
 | |
|                 hiddenstr = self.HIDDEN
 | |
|             else:
 | |
|                 hiddenstr = self.AVAILABLE
 | |
|             self.model.append(row=[
 | |
|                 '<span weight="bold" color="#267726">%s</span>' % _("OK"),
 | |
|                 i[0], pdata.description, None, pdata.id, hiddenstr])
 | |
|         
 | |
|     def __populate_reg_list(self):
 | |
|         """ Build list of registered plugins"""
 | |
|         for (type, typestr) in PTYPE_STR.items():
 | |
|             registered_plugins = []
 | |
|             for pdata in self.__preg.type_plugins(type):
 | |
|                 #  model: plugintype, hidden, pluginname, plugindescr, pluginid
 | |
|                 hidden = pdata.id in self.hidden
 | |
|                 if hidden:
 | |
|                     hiddenstr = self.HIDDEN
 | |
|                 else:
 | |
|                     hiddenstr = self.AVAILABLE
 | |
|                 registered_plugins.append([typestr, hiddenstr, pdata.name,
 | |
|                                            pdata.description, pdata.id])
 | |
|             for row in sorted(registered_plugins):
 | |
|                 self.model_reg.append(row)
 | |
| 
 | |
|     def __rebuild_load_list(self):
 | |
|         self.model.clear()
 | |
|         self.__populate_load_list()
 | |
|     
 | |
|     def __rebuild_reg_list(self):
 | |
|         self.model_reg.clear()
 | |
|         self.__populate_reg_list()
 | |
| 
 | |
|     def cursor_changed(self, obj):
 | |
|         if __debug__:
 | |
|             selection = obj.get_selection()
 | |
|             if selection:
 | |
|                 model, node = selection.get_selected()
 | |
|                 if node:
 | |
|                     data = model.get_value(node, 3)
 | |
|                     self.__load_btn.set_sensitive(data is not None)
 | |
| 
 | |
|     def button_press(self, obj, event):
 | |
|         """ Callback function from the user clicking on a line """
 | |
|         if event.type == Gdk.EventType._2BUTTON_PRESS and event.button == 1:
 | |
|             model, node = self.selection.get_selected()
 | |
|             data = model.get_value(node, 3)
 | |
|             name = model.get_value(node, 1)
 | |
|             if data:
 | |
|                 PluginTrace(self.uistate, [], data, name)
 | |
|                 
 | |
|     def button_press_reg(self, obj, event):
 | |
|         """ Callback function from the user clicking on a line in reg plugin
 | |
|         """
 | |
|         if event.type == Gdk.EventType._2BUTTON_PRESS and event.button == 1:
 | |
|             self.__info(obj, self.list_reg, 4)
 | |
|     
 | |
|     def button_press_addon(self, obj):
 | |
|         """ Callback function from the user clicking on a line in reg plugin
 | |
|         """
 | |
|         selection = self.addon_list.get_selection()
 | |
|         if selection:
 | |
|             model, node = selection.get_selected()
 | |
|             if node:
 | |
|                 url = model.get_value(node, 9)
 | |
|                 self.install_addon_path.set_text(url)
 | |
|     
 | |
|     def build_menu_names(self, obj):
 | |
|         return (self.title, "")
 | |
|     
 | |
|     def __reload(self, obj):
 | |
|         """ Callback function from the "Reload" button """
 | |
|         self.__pmgr.reload_plugins()
 | |
|         self.__rebuild_load_list()
 | |
|         self.__rebuild_reg_list()
 | |
|     
 | |
|     def __info(self, obj, list_obj, id_col):
 | |
|         """ Callback function from the "Info" button
 | |
|         """
 | |
|         selection = list_obj.get_selection()
 | |
|         model, node = selection.get_selected()
 | |
|         if not node:
 | |
|             return
 | |
|         id = model.get_value(node, id_col)
 | |
|         pdata = self.__preg.get_plugin(id)
 | |
|         typestr = pdata.ptype
 | |
|         auth = ' - '.join(pdata.authors)
 | |
|         email = ' - '.join(pdata.authors_email)
 | |
|         if len(auth) > 60: 
 | |
|             auth = auth[:60] + '...'
 | |
|         if len(email) > 60: 
 | |
|             email = email[:60] + '...'
 | |
|         if pdata:
 | |
|             infotxt = """%(plugnam)s: %(name)s [%(typestr)s]
 | |
| 
 | |
| %(plugdes)s: %(descr)s
 | |
| %(plugver)s: %(version)s
 | |
| %(plugaut)s: %(authors)s
 | |
| %(plugmel)s: %(email)s
 | |
| %(plugfil)s: %(fname)s
 | |
| %(plugpat)s: %(fpath)s
 | |
| """ % {
 | |
|             'name': pdata.name,
 | |
|             'typestr': typestr,
 | |
|             'descr': pdata.description,
 | |
|             'version': pdata.version,
 | |
|             'authors': auth,
 | |
|             'email': email,
 | |
|             'fname': pdata.fname,
 | |
|             'fpath': pdata.fpath,
 | |
|             'plugnam': _("Plugin name"),
 | |
|             'plugdes': _("Description"),
 | |
|             'plugver': _("Version"),
 | |
|             'plugaut': _("Authors"),
 | |
|             'plugmel': _("Email"),
 | |
|             'plugfil': _("Filename"),
 | |
|             'plugpat': _("Location"),
 | |
|             }
 | |
|             InfoDialog(_('Detailed Info'), infotxt, parent=self.window)
 | |
|     
 | |
|     def __hide(self, obj, list_obj, id_col, hide_col):
 | |
|         """ Callback function from the "Hide" button
 | |
|         """
 | |
|         selection = list_obj.get_selection()
 | |
|         model, node = selection.get_selected()
 | |
|         if not node:
 | |
|             return
 | |
|         id = model.get_value(node, id_col)
 | |
|         if id in self.hidden:
 | |
|             #unhide
 | |
|             self.hidden.remove(id)
 | |
|             model.set_value(node, hide_col, self.AVAILABLE)
 | |
|             self.__pmgr.unhide_plugin(id)
 | |
|         else:
 | |
|             #hide
 | |
|             self.hidden.add(id)
 | |
|             model.set_value(node, hide_col, self.HIDDEN)
 | |
|             self.__pmgr.hide_plugin(id)
 | |
|     
 | |
|     def __load(self, obj, list_obj, id_col):
 | |
|         """ Callback function from the "Load" button
 | |
|         """
 | |
|         selection = list_obj.get_selection()
 | |
|         model, node = selection.get_selected()
 | |
|         if not node:
 | |
|             return
 | |
|         idv = model.get_value(node, id_col)
 | |
|         pdata = self.__preg.get_plugin(idv)
 | |
|         self.__pmgr.load_plugin(pdata)
 | |
|         self.__rebuild_load_list()
 | |
| 
 | |
|     def __edit(self, obj, list_obj, id_col):
 | |
|         """ Callback function from the "Load" button
 | |
|         """
 | |
|         selection = list_obj.get_selection()
 | |
|         model, node = selection.get_selected()
 | |
|         if not node:
 | |
|             return
 | |
|         id = model.get_value(node, id_col)
 | |
|         pdata = self.__preg.get_plugin(id)
 | |
|         if pdata.fpath and pdata.fname:
 | |
|             open_file_with_default_application(
 | |
|                 os.path.join(pdata.fpath, pdata.fname)
 | |
|                 )
 | |
| 
 | |
| #-------------------------------------------------------------------------
 | |
| #
 | |
| # Details for an individual plugin that failed
 | |
| #
 | |
| #-------------------------------------------------------------------------
 | |
| class PluginTrace(ManagedWindow):
 | |
|     """Displays a dialog showing the status of loaded plugins"""
 | |
|     
 | |
|     def __init__(self, uistate, track, data, name):
 | |
|         self.name = name
 | |
|         title = "%s: %s" % (_("Plugin Error"), name)
 | |
|         ManagedWindow.__init__(self, uistate, track, self)
 | |
| 
 | |
|         self.set_window(Gtk.Dialog("", uistate.window,
 | |
|                                    Gtk.DialogFlags.DESTROY_WITH_PARENT,
 | |
|                                    (Gtk.STOCK_CLOSE, Gtk.ResponseType.CLOSE)),
 | |
|                         None, title)
 | |
|         self.window.set_size_request(600, 400)
 | |
|         self.window.connect('response', self.close)
 | |
|         
 | |
|         scrolled_window = Gtk.ScrolledWindow()
 | |
|         scrolled_window.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC)
 | |
|         self.text = Gtk.TextView()
 | |
|         scrolled_window.add(self.text)
 | |
|         self.text.get_buffer().set_text(
 | |
|             "".join(traceback.format_exception(data[0],data[1],data[2])))
 | |
| 
 | |
|         self.window.get_content_area().add(scrolled_window)
 | |
|         self.window.show_all()
 | |
| 
 | |
|     def build_menu_names(self, obj):
 | |
|         return (self.name, None)
 | |
| 
 | |
| 
 | |
| #-------------------------------------------------------------------------
 | |
| #
 | |
| # Classes for tools
 | |
| #
 | |
| #-------------------------------------------------------------------------
 | |
| class LinkTag(Gtk.TextTag):
 | |
|     def __init__(self, link, buffer):
 | |
|         GObject.GObject.__init__(self, name=link)
 | |
|         tag_table = buffer.get_tag_table()
 | |
|         self.set_property('foreground', "#0000ff")
 | |
|         self.set_property('underline', Pango.Underline.SINGLE)
 | |
|         try:
 | |
|             tag_table.add(self)
 | |
|         except ValueError:
 | |
|             pass # already in table
 | |
| 
 | |
| class ToolManagedWindowBase(ManagedWindow):
 | |
|     """
 | |
|     Copied from src/ReportBase/_BareReportDialog.py BareReportDialog
 | |
|     """
 | |
|     border_pad = 6
 | |
|     HELP_TOPIC = None
 | |
|     def __init__(self, dbstate, uistate, option_class, name, callback=None):
 | |
|         self.name = name
 | |
|         ManagedWindow.__init__(self, uistate, [], self)
 | |
| 
 | |
|         self.extra_menu = None
 | |
|         self.widgets = []
 | |
|         self.frame_names = []
 | |
|         self.frames = {}
 | |
|         self.format_menu = None
 | |
|         self.style_button = None
 | |
| 
 | |
|         window = Gtk.Dialog('Tool')
 | |
|         self.set_window(window, None, self.get_title())
 | |
| 
 | |
|         #self.window.connect('response', self.close)
 | |
|         self.cancel = self.window.add_button(Gtk.STOCK_CLOSE,
 | |
|                                              Gtk.ResponseType.CANCEL)
 | |
|         self.cancel.connect('clicked', self.close)
 | |
| 
 | |
|         self.ok = self.window.add_button(Gtk.STOCK_EXECUTE, Gtk.ResponseType.OK)
 | |
|         self.ok.connect('clicked', self.on_ok_clicked)
 | |
| 
 | |
|         self.window.set_default_size(600, -1)
 | |
| 
 | |
|         # Set up and run the dialog.  These calls are not in top down
 | |
|         # order when looking at the dialog box as there is some
 | |
|         # interaction between the various frames.
 | |
| 
 | |
|         self.setup_title()
 | |
|         self.setup_header()
 | |
|         #self.tbl = Gtk.Table(4, 4, False)
 | |
|         #self.tbl.set_col_spacings(12)
 | |
|         #self.tbl.set_row_spacings(6)
 | |
|         #self.tbl.set_border_width(6)
 | |
|         #self.col = 0
 | |
|         #self.window.vbox.add(self.tbl)
 | |
| 
 | |
|         # Build the list of widgets that are used to extend the Options
 | |
|         # frame and to create other frames
 | |
|         self.add_user_options()
 | |
|         
 | |
|         self.notebook = Gtk.Notebook()
 | |
|         self.notebook.set_border_width(6)
 | |
|         self.window.get_content_area().add(self.notebook)
 | |
| 
 | |
|         self.results_text = Gtk.TextView() 
 | |
|         self.results_text.connect('button-press-event', 
 | |
|                                   self.on_button_press) 
 | |
|         self.results_text.connect('motion-notify-event', 
 | |
|                                   self.on_motion)
 | |
|         self.tags = []
 | |
|         self.link_cursor = Gdk.Cursor.new(Gdk.CursorType.LEFT_PTR)
 | |
|         self.standard_cursor = Gdk.Cursor.new(Gdk.CursorType.XTERM)
 | |
| 
 | |
|         self.setup_other_frames()
 | |
|         self.set_current_frame(self.initial_frame())
 | |
|         self.show()
 | |
| 
 | |
|     #------------------------------------------------------------------------
 | |
|     #
 | |
|     # Callback functions from the dialog
 | |
|     #
 | |
|     #------------------------------------------------------------------------
 | |
|     def on_cancel(self, *obj):
 | |
|         pass # cancel just closes
 | |
| 
 | |
|     def on_ok_clicked(self, obj):
 | |
|         """
 | |
|         The user is satisfied with the dialog choices. Parse all options
 | |
|         and run the tool.
 | |
|         """
 | |
|         # Save options
 | |
|         self.options.parse_user_options()
 | |
|         self.options.handler.save_options()
 | |
|         self.pre_run()
 | |
|         self.run() # activate results tab
 | |
|         self.post_run()
 | |
| 
 | |
|     def initial_frame(self):
 | |
|         return None
 | |
| 
 | |
|     def on_motion(self, view, event):
 | |
|         buffer_location = view.window_to_buffer_coords(Gtk.TextWindowType.TEXT, 
 | |
|                                                        int(event.x), 
 | |
|                                                        int(event.y))
 | |
|         iter = view.get_iter_at_location(*buffer_location)
 | |
|         for (tag, person_handle) in self.tags:
 | |
|             if iter.has_tag(tag):
 | |
|                 _window = view.get_window(Gtk.TextWindowType.TEXT)
 | |
|                 _window.set_cursor(self.link_cursor)
 | |
|                 return False # handle event further, if necessary
 | |
|         view.get_window(Gtk.TextWindowType.TEXT).set_cursor(self.standard_cursor)
 | |
|         return False # handle event further, if necessary
 | |
| 
 | |
|     def on_button_press(self, view, event):
 | |
|         buffer_location = view.window_to_buffer_coords(Gtk.TextWindowType.TEXT, 
 | |
|                                                        int(event.x), 
 | |
|                                                        int(event.y))
 | |
|         iter = view.get_iter_at_location(*buffer_location)
 | |
|         for (tag, person_handle) in self.tags:
 | |
|             if iter.has_tag(tag):
 | |
|                 person = self.db.get_person_from_handle(person_handle)
 | |
|                 if event.button == 1:
 | |
|                     if event.type == Gdk.EventType._2BUTTON_PRESS:
 | |
|                         try:
 | |
|                             EditPerson(self.dbstate, self.uistate, [], person)
 | |
|                         except WindowActiveError:
 | |
|                             pass
 | |
|                     else:
 | |
|                         self.uistate.set_active(person_handle, 'Person')
 | |
|                     return True # handled event
 | |
|         return False # did not handle event
 | |
| 
 | |
|     def results_write_link(self, text, person, person_handle):
 | |
|         self.results_write("   ")
 | |
|         buffer = self.results_text.get_buffer()
 | |
|         iter = buffer.get_end_iter()
 | |
|         offset = buffer.get_char_count()
 | |
|         self.results_write(text)
 | |
|         start = buffer.get_iter_at_offset(offset)
 | |
|         end = buffer.get_end_iter()
 | |
|         self.tags.append((LinkTag(person_handle, buffer), person_handle))
 | |
|         buffer.apply_tag(self.tags[-1][0], start, end)
 | |
|         
 | |
|     def results_write(self, text):
 | |
|         buffer = self.results_text.get_buffer()
 | |
|         mark = buffer.create_mark("end", buffer.get_end_iter())
 | |
|         self.results_text.scroll_to_mark(mark, 0.0, True, 0, 0)
 | |
|         buffer.insert_at_cursor(text)
 | |
|         buffer.delete_mark_by_name("end")        
 | |
| 
 | |
|     def write_to_page(self, page, text):
 | |
|         buffer = page.get_buffer()
 | |
|         mark = buffer.create_mark("end", buffer.get_end_iter())
 | |
|         page.scroll_to_mark(mark, 0.0, True, 0, 0)
 | |
|         buffer.insert_at_cursor(text)
 | |
|         buffer.delete_mark_by_name("end")        
 | |
| 
 | |
|     def clear(self, text):
 | |
|         # Remove all tags and clear text
 | |
|         buffer = text.get_buffer()
 | |
|         tag_table = buffer.get_tag_table()
 | |
|         start = buffer.get_start_iter()
 | |
|         end = buffer.get_end_iter()
 | |
|         for (tag, handle) in self.tags:
 | |
|             buffer.remove_tag(tag, start, end)
 | |
|             tag_table.remove(tag)
 | |
|         self.tags = []
 | |
|         buffer.set_text("")
 | |
| 
 | |
|     def results_clear(self):
 | |
|         # Remove all tags and clear text
 | |
|         buffer = self.results_text.get_buffer()
 | |
|         tag_table = buffer.get_tag_table()
 | |
|         start = buffer.get_start_iter()
 | |
|         end = buffer.get_end_iter()
 | |
|         for (tag, handle) in self.tags:
 | |
|             buffer.remove_tag(tag, start, end)
 | |
|             tag_table.remove(tag)
 | |
|         self.tags = []
 | |
|         buffer.set_text("")
 | |
|         
 | |
|     def pre_run(self):
 | |
|         from ..utils import ProgressMeter
 | |
|         self.progress = ProgressMeter(self.get_title())
 | |
|         
 | |
|     def run(self):
 | |
|         raise NotImplementedError("tool needs to define a run() method")
 | |
| 
 | |
|     def post_run(self):
 | |
|         self.progress.close()
 | |
| 
 | |
|     #------------------------------------------------------------------------
 | |
|     #
 | |
|     # Functions related to setting up the dialog window.
 | |
|     #
 | |
|     #------------------------------------------------------------------------
 | |
|     def get_title(self):
 | |
|         """The window title for this dialog"""
 | |
|         return "Tool" # self.title
 | |
| 
 | |
|     def get_header(self, name):
 | |
|         """The header line to put at the top of the contents of the
 | |
|         dialog box.  By default this will just be the name of the
 | |
|         selected person.  Most subclasses will customize this to give
 | |
|         some indication of what the report will be, i.e. 'Descendant
 | |
|         Report for %s'."""
 | |
|         return self.get_title()
 | |
|         
 | |
|     def setup_title(self):
 | |
|         """Set up the title bar of the dialog.  This function relies
 | |
|         on the get_title() customization function for what the title
 | |
|         should be."""
 | |
|         self.window.set_title(self.get_title())
 | |
| 
 | |
|     def setup_header(self):
 | |
|         """Set up the header line bar of the dialog.  This function
 | |
|         relies on the get_header() customization function for what the
 | |
|         header line should read.  If no customization function is
 | |
|         supplied by the subclass, the default is to use the full name
 | |
|         of the currently selected person."""
 | |
| 
 | |
|         title = self.get_header(self.get_title())
 | |
|         label = Gtk.Label(label='<span size="larger" weight="bold">%s</span>' % title)
 | |
|         label.set_use_markup(True)
 | |
|         self.window.get_content_area().pack_start(label, False, False, 
 | |
|                                                   self.border_pad)
 | |
| 
 | |
|     def add_frame_option(self, frame_name, label_text, widget):
 | |
|         """Similar to add_option this method takes a frame_name, a
 | |
|         text string and a Gtk Widget. When the interface is built,
 | |
|         all widgets with the same frame_name are grouped into a
 | |
|         GtkFrame. This allows the subclass to create its own sections,
 | |
|         filling them with its own widgets. The subclass is reponsible for
 | |
|         all managing of the widgets, including extracting the final value
 | |
|         before the report executes. This task should only be called in
 | |
|         the add_user_options task."""
 | |
|         
 | |
|         if frame_name in self.frames:
 | |
|             self.frames[frame_name].append((label_text, widget))
 | |
|         else:
 | |
|             self.frames[frame_name] = [(label_text, widget)]
 | |
|             self.frame_names.append(frame_name)
 | |
| 
 | |
|     def set_current_frame(self, name):
 | |
|         if name is None:
 | |
|             self.notebook.set_current_page(0)
 | |
|         else:
 | |
|             for frame_name in self.frame_names:
 | |
|                 if name == frame_name:
 | |
|                     if len(self.frames[frame_name]) > 0:
 | |
|                         fname, child = self.frames[frame_name][0]
 | |
|                         page = self.notebook.page_num(child)
 | |
|                         self.notebook.set_current_page(page)
 | |
|                         return
 | |
| 
 | |
|     def add_results_frame(self, frame_name="Results"):
 | |
|         if frame_name not in self.frames:
 | |
|             window = Gtk.ScrolledWindow()
 | |
|             window.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC)
 | |
|             window.add(self.results_text)
 | |
|             window.set_shadow_type(Gtk.ShadowType.IN)
 | |
|             self.frames[frame_name] = [[frame_name, window]] 
 | |
|             self.frame_names.append(frame_name)
 | |
|             l = Gtk.Label(label="<b>%s</b>" % _(frame_name))
 | |
|             l.set_use_markup(True)
 | |
|             self.notebook.append_page(window, l)
 | |
|             self.notebook.show_all()
 | |
|         else:
 | |
|             self.results_clear()
 | |
|         return self.results_text
 | |
| 
 | |
|     def add_page(self, frame_name="Help"):
 | |
|         if frame_name not in self.frames:
 | |
|             text = Gtk.TextView() 
 | |
|             text.set_wrap_mode(Gtk.WrapMode.WORD)
 | |
|             window = Gtk.ScrolledWindow()
 | |
|             window.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC)
 | |
|             window.add(text)
 | |
|             window.set_shadow_type(Gtk.ShadowType.IN)
 | |
|             self.frames[frame_name] = [[frame_name, window]] 
 | |
|             self.frame_names.append(frame_name)
 | |
|             l = Gtk.Label(label="<b>%s</b>" % _(frame_name))
 | |
|             l.set_use_markup(True)
 | |
|             self.notebook.append_page(window, l)
 | |
|             self.notebook.show_all()
 | |
|         else:
 | |
|             # FIXME: get text
 | |
|             #
 | |
|             text = self.frames[frame_name][0][1].something
 | |
|         return text
 | |
| 
 | |
|     def setup_other_frames(self):
 | |
|         """Similar to add_option this method takes a frame_name, a
 | |
|         text string and a Gtk Widget. When the interface is built,
 | |
|         all widgets with the same frame_name are grouped into a
 | |
|         GtkFrame. This allows the subclass to create its own sections,
 | |
|         filling them with its own widgets. The subclass is reponsible for
 | |
|         all managing of the widgets, including extracting the final value
 | |
|         before the report executes. This task should only be called in
 | |
|         the add_user_options task."""
 | |
|         for key in self.frame_names:
 | |
|             flist = self.frames[key]
 | |
|             table = Gtk.Table(n_rows=3, n_columns=len(flist))
 | |
|             table.set_col_spacings(12)
 | |
|             table.set_row_spacings(6)
 | |
|             table.set_border_width(6)
 | |
|             l = Gtk.Label(label="<b>%s</b>" % key)
 | |
|             l.set_use_markup(True)
 | |
|             self.notebook.append_page(table, l)
 | |
|             row = 0
 | |
|             for (text, widget) in flist:
 | |
|                 if text:
 | |
|                     text_widget = Gtk.Label(label='%s:' % text)
 | |
|                     text_widget.set_alignment(0.0, 0.5)
 | |
|                     table.attach(text_widget, 1, 2, row, row+1,
 | |
|                                  Gtk.AttachOptions.SHRINK|Gtk.AttachOptions.FILL, Gtk.AttachOptions.SHRINK)
 | |
|                     table.attach(widget, 2, 3, row, row+1,
 | |
|                                  yoptions=Gtk.AttachOptions.SHRINK)
 | |
|                 else:
 | |
|                     table.attach(widget, 2, 3, row, row+1,
 | |
|                                  yoptions=Gtk.AttachOptions.SHRINK)
 | |
|                 row += 1
 | |
|         self.notebook.show_all()
 | |
| 
 | |
|     #------------------------------------------------------------------------
 | |
|     #
 | |
|     # Functions related to extending the options
 | |
|     #
 | |
|     #------------------------------------------------------------------------
 | |
|     def add_user_options(self):
 | |
|         """Called to allow subclasses add widgets to the dialog form.
 | |
|         It is called immediately before the window is displayed. All
 | |
|         calls to add_option or add_frame_option should be called in
 | |
|         this task."""
 | |
|         add_gui_options(self)
 | |
| 
 | |
|     def build_menu_names(self, obj):
 | |
|         return (_('Main window'), self.get_title())
 | |
| 
 | |
| 
 | |
| 
 | |
| class ToolManagedWindowBatch(tool.BatchTool, ToolManagedWindowBase):
 | |
|     def __init__(self, dbstate, user, options_class, name, callback=None):
 | |
|         uistate = user.uistate
 | |
|         # This constructor will ask a question, set self.fail:
 | |
|         self.dbstate = dbstate
 | |
|         self.uistate = uistate
 | |
|         tool.BatchTool.__init__(self, dbstate, user, options_class, name)
 | |
|         if not self.fail:
 | |
|             ToolManagedWindowBase.__init__(self, dbstate, uistate, 
 | |
|                                            options_class, name, callback)
 | |
| 
 | |
| class ToolManagedWindow(tool.Tool, ToolManagedWindowBase):
 | |
|     def __init__(self, dbstate, uistate, options_class, name, callback=None):
 | |
|         self.dbstate = dbstate
 | |
|         self.uistate = uistate
 | |
|         tool.Tool.__init__(self, dbstate, options_class, name)
 | |
|         ToolManagedWindowBase.__init__(self, dbstate, uistate, options_class, 
 | |
|                                        name, callback)
 | |
| 
 | |
| #-------------------------------------------------------------------------
 | |
| #
 | |
| # UpdateAddons
 | |
| #
 | |
| #-------------------------------------------------------------------------
 | |
| class UpdateAddons(ManagedWindow):
 | |
| 
 | |
|     def __init__(self, uistate, track, addon_update_list):
 | |
|         self.title = _('Available Gramps Updates for Addons')
 | |
|         ManagedWindow.__init__(self, uistate, track, self.__class__)
 | |
| 
 | |
|         glade = Glade("updateaddons.glade")
 | |
|         self.update_dialog = glade.toplevel
 | |
|         self.set_window(self.update_dialog, glade.get_object('title'), 
 | |
|                         self.title)
 | |
|         self.window.set_size_request(750, 400)
 | |
| 
 | |
|         if win() and Gtk.get_minor_version() < 11:
 | |
|             self.window.set_transient_for(self.window.get_toplevel())
 | |
| 
 | |
|         apply_button = glade.get_object('apply')
 | |
|         cancel_button = glade.get_object('cancel')
 | |
|         select_all = glade.get_object('select_all')
 | |
|         select_all.connect("clicked", self.select_all_clicked)
 | |
|         select_none = glade.get_object('select_none')
 | |
|         select_none.connect("clicked", self.select_none_clicked)
 | |
|         apply_button.connect("clicked", self.install_addons)
 | |
|         cancel_button.connect("clicked", self.close)
 | |
| 
 | |
|         self.list = ListModel(glade.get_object("list"), [
 | |
|                 # name, click?, width, toggle
 | |
|                 {"name": _('Select'),
 | |
|                  "width": 60,
 | |
|                  "type": TOGGLE,
 | |
|                  "visible_col": 6,
 | |
|                  "editable": True},                         # 0 selected?
 | |
|                 (_('Type'), 1, 180),                        # 1 new gramplet
 | |
|                 (_('Name'), 2, 200),                        # 2 name (version)
 | |
|                 (_('Description'), 3, 200),                 # 3 description
 | |
|                 ('', NOSORT, 0),                            # 4 url
 | |
|                 ('', NOSORT, 0),                            # 5 id
 | |
|                 {"name": '', "type": TOGGLE},               # 6 visible? bool
 | |
|                 ], list_mode="tree")
 | |
|         pos = None
 | |
|         addon_update_list.sort(key=lambda x: "%s %s" % (x[0], x[2]["t"]))
 | |
|         last_category = None
 | |
|         for (status,plugin_url,plugin_dict) in addon_update_list:
 | |
|             count = get_count(addon_update_list, plugin_dict["t"])
 | |
|             category = _("%(adjective)s: %(addon)s") % {
 | |
|                 "adjective": status, 
 | |
|                 "addon": _(plugin_dict["t"])}
 | |
|             if last_category != category:
 | |
|                 last_category = category
 | |
|                 node = self.list.add([False, # initially selected?
 | |
|                                       category,
 | |
|                                       "",
 | |
|                                       "",
 | |
|                                       "",
 | |
|                                       "",
 | |
|                                       False]) # checkbox visible?
 | |
|             iter = self.list.add([False, # initially selected?
 | |
|                                   "%s %s" % (status, _(plugin_dict["t"])),
 | |
|                                   "%s (%s)" % (plugin_dict["n"],
 | |
|                                                plugin_dict["v"]),
 | |
|                                   plugin_dict["d"],
 | |
|                                   plugin_url,
 | |
|                                   plugin_dict["i"],
 | |
|                                   True], node=node)
 | |
|             if pos is None:
 | |
|                 pos = iter
 | |
|         if pos:
 | |
|             self.list.selection.select_iter(pos)
 | |
|         self.update_dialog.run()
 | |
| 
 | |
|     def build_menu_names(self, obj):
 | |
|         return (self.title, "")
 | |
|     
 | |
|     def select_all_clicked(self, widget):
 | |
|         """
 | |
|         Select all of the addons for download.
 | |
|         """
 | |
|         self.list.model.foreach(update_rows, True)
 | |
|         self.list.tree.expand_all()
 | |
| 
 | |
|     def select_none_clicked(self, widget):
 | |
|         """
 | |
|         Select none of the addons for download.
 | |
|         """
 | |
|         self.list.model.foreach(update_rows, False)
 | |
|         self.list.tree.expand_all()
 | |
| 
 | |
|     def install_addons(self, obj):
 | |
|         """
 | |
|         Process all of the selected addons.
 | |
|         """
 | |
|         self.update_dialog.hide()
 | |
|         model = self.list.model
 | |
| 
 | |
|         iter = model.get_iter_first()
 | |
|         length = 0
 | |
|         while iter:
 | |
|             iter = model.iter_next(iter)
 | |
|             if iter:
 | |
|                 length += model.iter_n_children(iter)
 | |
| 
 | |
|         longop = LongOpStatus(
 | |
|             _("Downloading and installing selected addons..."),
 | |
|             length, 1, # total, increment-by
 | |
|             can_cancel=True)
 | |
|         pm = ProgressMonitor(GtkProgressDialog,
 | |
|                              ("Title", self.window, Gtk.DialogFlags.MODAL))
 | |
|         pm.add_op(longop)
 | |
|         count = 0
 | |
|         if not config.get('behavior.do-not-show-previously-seen-updates'):
 | |
|             # reset list
 | |
|             config.get('behavior.previously-seen-updates')[:] = []
 | |
| 
 | |
|         iter = model.get_iter_first()
 | |
|         while iter:
 | |
|             for rowcnt in range(model.iter_n_children(iter)):
 | |
|                 child = model.iter_nth_child(iter, rowcnt)
 | |
|                 row = [model.get_value(child, n) for n in range(6)]
 | |
|                 if longop.should_cancel():
 | |
|                     break
 | |
|                 elif row[0]: # toggle on
 | |
|                     load_addon_file(row[4], callback=LOG.debug)
 | |
|                     count += 1
 | |
|                 else: # add to list of previously seen, but not installed
 | |
|                     if row[5] not in config.get('behavior.previously-seen-updates'):
 | |
|                         config.get('behavior.previously-seen-updates').append(row[5])
 | |
|                 longop.heartbeat()
 | |
|                 pm._get_dlg()._process_events()
 | |
|             iter = model.iter_next(iter)
 | |
| 
 | |
|         if not longop.was_cancelled():
 | |
|             longop.end()
 | |
|         if count:
 | |
|             OkDialog(_("Done downloading and installing addons"),
 | |
|                      # translators: leave all/any {...} untranslated
 | |
|                      "%s %s" % (ngettext("{number_of} addon was installed.",
 | |
|                                          "{number_of} addons were installed.",
 | |
|                                          count).format(number_of=count),
 | |
|                                 _("You need to restart Gramps to see new views.")),
 | |
|                      self.window)
 | |
|         else:
 | |
|             OkDialog(_("Done downloading and installing addons"),
 | |
|                      _("No addons were installed."),
 | |
|                      self.window)
 | |
|         self.close()
 | |
| 
 | |
| #-------------------------------------------------------------------------
 | |
| #
 | |
| # Local Functions
 | |
| #
 | |
| #-------------------------------------------------------------------------
 | |
| def update_rows(model, path, iter, user_data):
 | |
|     """
 | |
|     Update the rows of a model.
 | |
|     """
 | |
|     #path: (8,)   iter: <GtkTreeIter at 0xbfa89fa0>
 | |
|     #path: (8, 0) iter: <GtkTreeIter at 0xbfa89f60>
 | |
|     if len(path.get_indices()) == 2:
 | |
|         row = model[path]
 | |
|         row[0] = user_data
 | |
|         model.row_changed(path, iter)
 | |
| 
 | |
| def get_count(addon_update_list, category):
 | |
|     """
 | |
|     Get the count of matching category items.
 | |
|     """
 | |
|     count = 0
 | |
|     for (status,plugin_url,plugin_dict) in addon_update_list:
 | |
|         if plugin_dict["t"] == category and plugin_url:
 | |
|             count += 1
 | |
|     return count
 |