514 lines
17 KiB
Python
514 lines
17 KiB
Python
|
#
|
||
|
|
# Gramps - a GTK+/GNOME based genealogy program
|
||
|
|
#
|
||
|
|
# Copyright (C) 2004-2005 Donald N. Allingham
|
||
|
|
#
|
||
|
|
# 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
|
||
|
|
#
|
||
|
|
# $Id$
|
||
|
|
|
||
|
|
# Written by Alex Roitman
|
||
|
|
|
||
|
|
"""
|
||
|
|
General option handling, including saving and parsing.
|
||
|
|
"""
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------
|
||
|
|
#
|
||
|
|
# Standard Python modules
|
||
|
|
#
|
||
|
|
#-------------------------------------------------------------------------
|
||
|
|
import os
|
||
|
|
from gettext import gettext as _
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------
|
||
|
|
#
|
||
|
|
# SAX interface
|
||
|
|
#
|
||
|
|
#-------------------------------------------------------------------------
|
||
|
|
try:
|
||
|
|
from xml.sax import make_parser,handler,SAXParseException
|
||
|
|
except:
|
||
|
|
from _xmlplus.sax import make_parser,handler,SAXParseException
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------
|
||
|
|
#
|
||
|
|
# gramps modules
|
||
|
|
#
|
||
|
|
#-------------------------------------------------------------------------
|
||
|
|
import Utils
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------
|
||
|
|
#
|
||
|
|
# List of options for a single module
|
||
|
|
#
|
||
|
|
#-------------------------------------------------------------------------
|
||
|
|
class OptionList:
|
||
|
|
"""
|
||
|
|
Implements a set of options to parse and store for a given module.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
self.options = {}
|
||
|
|
|
||
|
|
def set_options(self,options):
|
||
|
|
"""
|
||
|
|
Sets the whole bunch of options for the OptionList.
|
||
|
|
@param options: list of options to set.
|
||
|
|
@type options: list
|
||
|
|
"""
|
||
|
|
self.options = options
|
||
|
|
|
||
|
|
def get_options(self):
|
||
|
|
"""
|
||
|
|
Returns the whole bunch of options for the OptionList.
|
||
|
|
@returns: list of options
|
||
|
|
@rtype: list
|
||
|
|
"""
|
||
|
|
return self.options
|
||
|
|
|
||
|
|
def set_option(self,name,value):
|
||
|
|
"""
|
||
|
|
Sets a particular option in the OptionList.
|
||
|
|
@param name: name of the option to set.
|
||
|
|
@type name: str
|
||
|
|
@param value: value of the option to set.
|
||
|
|
@type str
|
||
|
|
"""
|
||
|
|
self.options[name] = value
|
||
|
|
|
||
|
|
def remove_option(self,name):
|
||
|
|
"""
|
||
|
|
Removes a particular option from the OptionList.
|
||
|
|
@param name: name of the option to remove.
|
||
|
|
@type name: str
|
||
|
|
"""
|
||
|
|
if self.options.has_key(name):
|
||
|
|
del self.options[name]
|
||
|
|
|
||
|
|
def get_option(self,name):
|
||
|
|
"""
|
||
|
|
Returns the value of a particular option in the OptionList.
|
||
|
|
@param name: name of the option to retrieve
|
||
|
|
@type name: str
|
||
|
|
@returns: value associated with the passed option
|
||
|
|
@rtype: str
|
||
|
|
"""
|
||
|
|
return self.options.get(name,None)
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------
|
||
|
|
#
|
||
|
|
# Collection of option lists
|
||
|
|
#
|
||
|
|
#-------------------------------------------------------------------------
|
||
|
|
class OptionListCollection:
|
||
|
|
"""
|
||
|
|
Implements a collection of option lists.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(self,filename):
|
||
|
|
"""
|
||
|
|
Creates an OptionListCollection instance from the list defined
|
||
|
|
in the specified file.
|
||
|
|
@param filename: XML file that contains option definitions
|
||
|
|
@type filename: str
|
||
|
|
"""
|
||
|
|
|
||
|
|
self.filename = os.path.expanduser(filename)
|
||
|
|
self.option_list_map = {}
|
||
|
|
self.init_common()
|
||
|
|
self.parse()
|
||
|
|
|
||
|
|
def init_common(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def get_option_list_map(self):
|
||
|
|
"""
|
||
|
|
Returns the map of module names to option lists.
|
||
|
|
@returns: Returns the map of module names to option lists.
|
||
|
|
@rtype: dictionary
|
||
|
|
"""
|
||
|
|
return self.option_list_map
|
||
|
|
|
||
|
|
def get_option_list(self,name):
|
||
|
|
"""
|
||
|
|
Returns the option_list associated with the module name
|
||
|
|
@param name: name associated with the desired module.
|
||
|
|
@type name: str
|
||
|
|
@returns: returns the option list associated with the name,
|
||
|
|
or None of no such option exists
|
||
|
|
@rtype: str
|
||
|
|
"""
|
||
|
|
return self.option_list_map.get(name,None)
|
||
|
|
|
||
|
|
def get_module_names(self):
|
||
|
|
"""
|
||
|
|
Returns a list of all the module names in the OptionListCollection
|
||
|
|
@returns: returns the list of module names
|
||
|
|
@rtype: list
|
||
|
|
"""
|
||
|
|
return self.option_list_map.keys()
|
||
|
|
|
||
|
|
def set_option_list(self,name,option_list):
|
||
|
|
"""
|
||
|
|
Adds or replaces an option_list in the OptionListCollection.
|
||
|
|
@param name: name assocated with the module to add or replace.
|
||
|
|
@type name: str
|
||
|
|
@param option_list: list of options
|
||
|
|
@type option_list: str
|
||
|
|
"""
|
||
|
|
self.option_list_map[name] = option_list
|
||
|
|
|
||
|
|
def write_common(self,f):
|
||
|
|
"""
|
||
|
|
Stub function for common options. Overridden by reports.
|
||
|
|
"""
|
||
|
|
pass
|
||
|
|
|
||
|
|
def write_module_common(self,f,option_list):
|
||
|
|
"""
|
||
|
|
Stub function for common options. Overridden by reports.
|
||
|
|
"""
|
||
|
|
pass
|
||
|
|
|
||
|
|
def save(self):
|
||
|
|
"""
|
||
|
|
Saves the current OptionListCollection to the associated file.
|
||
|
|
"""
|
||
|
|
f = open(self.filename,"w")
|
||
|
|
f.write("<?xml version=\"1.0\" encoding=\"utf-8\"?>\n")
|
||
|
|
f.write('<options>\n')
|
||
|
|
|
||
|
|
self.write_common(f)
|
||
|
|
|
||
|
|
for module_name in self.get_module_names():
|
||
|
|
option_list = self.get_option_list(module_name)
|
||
|
|
f.write('<module name="%s">\n' % module_name)
|
||
|
|
options = option_list.get_options()
|
||
|
|
for option_name in options.keys():
|
||
|
|
if type(options[option_name]) in (type(list()),type(tuple())):
|
||
|
|
f.write(' <option name="%s" value="" length="%d">\n' % (
|
||
|
|
option_name, len(options[option_name]) ) )
|
||
|
|
for list_index in range(len(options[option_name])):
|
||
|
|
f.write(' <listitem number="%d" value="%s"/>\n' % (
|
||
|
|
list_index, options[option_name][list_index]) )
|
||
|
|
f.write(' </option>\n')
|
||
|
|
else:
|
||
|
|
f.write(' <option name="%s" value="%s"/>\n' % (
|
||
|
|
option_name,options[option_name]) )
|
||
|
|
|
||
|
|
self.write_module_common(f,option_list)
|
||
|
|
|
||
|
|
f.write('</module>\n')
|
||
|
|
|
||
|
|
f.write('</options>\n')
|
||
|
|
f.close()
|
||
|
|
|
||
|
|
def parse(self):
|
||
|
|
"""
|
||
|
|
Loads the OptionList from the associated file, if it exists.
|
||
|
|
"""
|
||
|
|
try:
|
||
|
|
p = make_parser()
|
||
|
|
p.setContentHandler(OptionParser(self))
|
||
|
|
p.parse('file://' + self.filename)
|
||
|
|
except (IOError,OSError,SAXParseException):
|
||
|
|
pass
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------
|
||
|
|
#
|
||
|
|
# OptionParser
|
||
|
|
#
|
||
|
|
#-------------------------------------------------------------------------
|
||
|
|
class OptionParser(handler.ContentHandler):
|
||
|
|
"""
|
||
|
|
SAX parsing class for the OptionListCollection XML file.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(self,collection):
|
||
|
|
"""
|
||
|
|
Creates a OptionParser class that populates the passed collection.
|
||
|
|
|
||
|
|
collection: OptionListCollection to be loaded from the file.
|
||
|
|
"""
|
||
|
|
handler.ContentHandler.__init__(self)
|
||
|
|
self.collection = collection
|
||
|
|
|
||
|
|
self.mname = None
|
||
|
|
self.option_list = None
|
||
|
|
self.oname = None
|
||
|
|
self.o = None
|
||
|
|
self.an_o = None
|
||
|
|
self.list_class = OptionList
|
||
|
|
|
||
|
|
def startElement(self,tag,attrs):
|
||
|
|
"""
|
||
|
|
Overridden class that handles the start of a XML element
|
||
|
|
"""
|
||
|
|
if tag in ("report","module"):
|
||
|
|
self.mname = attrs['name']
|
||
|
|
self.option_list = self.list_class()
|
||
|
|
self.o = {}
|
||
|
|
elif tag == "option":
|
||
|
|
self.oname = attrs['name']
|
||
|
|
if attrs.has_key('length'):
|
||
|
|
self.an_o = []
|
||
|
|
else:
|
||
|
|
self.an_o = attrs['value']
|
||
|
|
elif tag == "listitem":
|
||
|
|
self.an_o.append(attrs['value'])
|
||
|
|
|
||
|
|
def endElement(self,tag):
|
||
|
|
"Overridden class that handles the end of a XML element"
|
||
|
|
if tag == "option":
|
||
|
|
self.o[self.oname] = self.an_o
|
||
|
|
elif tag in ("report","module"):
|
||
|
|
self.option_list.set_options(self.o)
|
||
|
|
self.collection.set_option_list(self.mname,self.option_list)
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------
|
||
|
|
#
|
||
|
|
# Class handling options for plugins
|
||
|
|
#
|
||
|
|
#-------------------------------------------------------------------------
|
||
|
|
class OptionHandler:
|
||
|
|
"""
|
||
|
|
Implements handling of the options for the plugins.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(self,module_name,options_dict,person_id=None):
|
||
|
|
self.module_name = module_name
|
||
|
|
self.default_options_dict = options_dict.copy()
|
||
|
|
self.options_dict = options_dict
|
||
|
|
|
||
|
|
# Retrieve our options from whole collection
|
||
|
|
self.init_subclass()
|
||
|
|
self.option_list_collection = self.collection_class(self.filename)
|
||
|
|
self.init_common()
|
||
|
|
self.saved_option_list = self.option_list_collection.get_option_list(module_name)
|
||
|
|
self.person_id = person_id
|
||
|
|
|
||
|
|
# Whatever was found should override the defaults
|
||
|
|
if self.saved_option_list:
|
||
|
|
self.set_options()
|
||
|
|
else:
|
||
|
|
# If nothing was found, set up the option list
|
||
|
|
self.saved_option_list = self.list_class()
|
||
|
|
self.option_list_collection.set_option_list(module_name,
|
||
|
|
self.saved_option_list)
|
||
|
|
|
||
|
|
def init_subclass(self):
|
||
|
|
self.collection_class = OptionListCollection
|
||
|
|
self.list_class = OptionList
|
||
|
|
self.filename = None
|
||
|
|
|
||
|
|
def init_common(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def set_options(self):
|
||
|
|
"""
|
||
|
|
Sets options to be used in this plugin according to the passed
|
||
|
|
options dictionary.
|
||
|
|
|
||
|
|
Dictionary values are all strings, since they were read from XML.
|
||
|
|
Here we need to convert them to the needed types. We use default
|
||
|
|
values to determine the type.
|
||
|
|
"""
|
||
|
|
# First we set options_dict values based on the saved options
|
||
|
|
options = self.saved_option_list.get_options()
|
||
|
|
bad_opts = []
|
||
|
|
for option_name in options.keys():
|
||
|
|
if not self.options_dict.has_key(option_name):
|
||
|
|
print "Option %s is present in the %s but is not known "\
|
||
|
|
"to the module." % (option_name,
|
||
|
|
self.option_list_collection.filename)
|
||
|
|
print "Ignoring..."
|
||
|
|
bad_opts.append(option_name)
|
||
|
|
continue
|
||
|
|
try:
|
||
|
|
converter = Utils.get_type_converter(self.options_dict[option_name])
|
||
|
|
self.options_dict[option_name] = converter(options[option_name])
|
||
|
|
except ValueError:
|
||
|
|
pass
|
||
|
|
|
||
|
|
for option_name in bad_opts:
|
||
|
|
options.pop(option_name)
|
||
|
|
|
||
|
|
# Then we set common options from whatever was found
|
||
|
|
self.set_common_options()
|
||
|
|
|
||
|
|
def set_common_options(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def save_options(self):
|
||
|
|
"""
|
||
|
|
Saves options to file.
|
||
|
|
|
||
|
|
We need to only store non-default options. Therefore, we remove all
|
||
|
|
options whose values are the defaults prior to saving.
|
||
|
|
"""
|
||
|
|
|
||
|
|
# First we save options from options_dict
|
||
|
|
for option_name in self.options_dict.keys():
|
||
|
|
if self.options_dict[option_name] == self.default_options_dict[option_name]:
|
||
|
|
self.saved_option_list.remove_option(option_name)
|
||
|
|
else:
|
||
|
|
self.saved_option_list.set_option(option_name,self.options_dict[option_name])
|
||
|
|
|
||
|
|
# Handle common options
|
||
|
|
self.save_common_options()
|
||
|
|
|
||
|
|
# Finally, save the whole collection into file
|
||
|
|
self.option_list_collection.save()
|
||
|
|
|
||
|
|
def save_common_options(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def get_filter_number(self):
|
||
|
|
if self.default_options_dict.has_key('filter'):
|
||
|
|
return self.options_dict.get('filter',
|
||
|
|
self.default_options_dict['filter'])
|
||
|
|
else:
|
||
|
|
return None
|
||
|
|
|
||
|
|
def set_filter_number(self,val):
|
||
|
|
self.options_dict['filter'] = val
|
||
|
|
|
||
|
|
def get_person_id(self):
|
||
|
|
return self.person_id
|
||
|
|
|
||
|
|
def set_person_id(self,val):
|
||
|
|
self.person_id = val
|
||
|
|
|
||
|
|
#------------------------------------------------------------------------
|
||
|
|
#
|
||
|
|
# Base Options class
|
||
|
|
#
|
||
|
|
#------------------------------------------------------------------------
|
||
|
|
class Options:
|
||
|
|
|
||
|
|
"""
|
||
|
|
Defines options and provides handling interface.
|
||
|
|
|
||
|
|
This is a base Options class for the modules. All modules' options
|
||
|
|
classes should derive from it.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(self,name,person_id=None):
|
||
|
|
"""
|
||
|
|
Initializes the class, performing usual house-keeping tasks.
|
||
|
|
Subclasses MUST call this in their __init__() method.
|
||
|
|
"""
|
||
|
|
self.set_new_options()
|
||
|
|
self.enable_options()
|
||
|
|
|
||
|
|
if self.enable_dict:
|
||
|
|
self.options_dict.update(self.enable_dict)
|
||
|
|
self.handler = OptionHandler(name,self.options_dict,person_id)
|
||
|
|
|
||
|
|
def set_new_options(self):
|
||
|
|
"""
|
||
|
|
Sets options specific for this module.
|
||
|
|
|
||
|
|
Modules that need custom options need to override this method.
|
||
|
|
Two dictionaries MUST be defined here:
|
||
|
|
|
||
|
|
self.options_dict
|
||
|
|
This is a dictionary whose keys are option names
|
||
|
|
and values are the default option values.
|
||
|
|
|
||
|
|
self.options_help
|
||
|
|
This is a dictionary whose keys are option names
|
||
|
|
and values are 3- or 4- lists or tuples:
|
||
|
|
('=example','Short description',VALUES,DO_PREPEND)
|
||
|
|
The VALUES is either a single string (in that case
|
||
|
|
the DO_PREPEND does not matter) or a list/tuple of
|
||
|
|
strings to list. In that case, if DO_PREPEND evaluates
|
||
|
|
as True then each string will be preneded with the ordinal
|
||
|
|
number when help is printed on the command line.
|
||
|
|
|
||
|
|
NOTE: Both dictionaries must have identical keys.
|
||
|
|
|
||
|
|
NOTE: If a particular module does not use custom options,
|
||
|
|
then it should not override this method.
|
||
|
|
"""
|
||
|
|
self.options_dict = {}
|
||
|
|
self.options_help = {}
|
||
|
|
|
||
|
|
def enable_options(self):
|
||
|
|
"""
|
||
|
|
Enables semi-common options for this module.
|
||
|
|
|
||
|
|
The semi-common option is the option which GRAMPS is aware of,
|
||
|
|
but not common enough to be present in all modules. Here's the list
|
||
|
|
of possible keys for semi-commons:
|
||
|
|
|
||
|
|
'filter' - Filter number, selected among filters
|
||
|
|
available for this module. If defined,
|
||
|
|
get_module_filters() method must be defined
|
||
|
|
which returns the list of available filters.
|
||
|
|
|
||
|
|
A self.enable_dict dictionary MUST be defined here, whose keys
|
||
|
|
are the valid semi-common keys above, and whose values are the
|
||
|
|
desired default values for semi-commons.
|
||
|
|
|
||
|
|
NOTE: If a particular module does not use semi-common options,
|
||
|
|
then it should not override this method.
|
||
|
|
"""
|
||
|
|
self.enable_dict = {}
|
||
|
|
|
||
|
|
def add_user_options(self,dialog):
|
||
|
|
"""
|
||
|
|
Sets up UI controls (widgets) for the options specific for this modul.
|
||
|
|
|
||
|
|
This method MUST be overridden by modules that define new options.
|
||
|
|
The single argument 'dialog' is the Report.BareReportDialog instance.
|
||
|
|
Any attribute of the dialog is available.
|
||
|
|
|
||
|
|
After the widgets are defined, they MUST be added to the dialog
|
||
|
|
using the following call:
|
||
|
|
dialog.add_options(LABEL,widget)
|
||
|
|
|
||
|
|
NOTE: To really have any effect besides looking pretty, each widget
|
||
|
|
set up here must be also parsed in the parse_user_options()
|
||
|
|
method below.
|
||
|
|
"""
|
||
|
|
pass
|
||
|
|
|
||
|
|
def parse_user_options(self,dialog):
|
||
|
|
"""
|
||
|
|
Parses UI controls (widgets) for the options specific for this module.
|
||
|
|
|
||
|
|
This method MUST be overridden by modules that define new options.
|
||
|
|
The single argument 'dialog' is the Report.BareReportDialog instance.
|
||
|
|
Any attribute of the dialog is available.
|
||
|
|
|
||
|
|
After obtaining values from the widgets, they MUST be used to set the
|
||
|
|
appropriate options_dict values. Otherwise the values will not have
|
||
|
|
any user-visible effect.
|
||
|
|
|
||
|
|
NOTE: Any widget parsed here MUST be defined and added to the dialog
|
||
|
|
in the add_user_options() method above.
|
||
|
|
"""
|
||
|
|
pass
|
||
|
|
|
||
|
|
def get_filter_number(self):
|
||
|
|
"""
|
||
|
|
Return number of a filter to use.
|
||
|
|
|
||
|
|
This method MUST NOT be overridden by subclasses.
|
||
|
|
"""
|
||
|
|
return self.handler.get_filter_number()
|