233 lines
7.8 KiB
Python
233 lines
7.8 KiB
Python
#
|
|
# Gramps - a GTK+/GNOME based genealogy program
|
|
#
|
|
# Copyright (C) 2000-2006 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$
|
|
|
|
"Utilities/Relationship calculator"
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# Standard python modules
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
import os
|
|
from gettext import gettext as _
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# GNOME libraries
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
import gtk.glade
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# GRAMPS modules
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
import RelLib
|
|
from BasicUtils import NameDisplay
|
|
import ManagedWindow
|
|
import ListModel
|
|
import DateHandler
|
|
from DisplayModels import PeopleModel
|
|
|
|
from QuestionDialog import ErrorDialog
|
|
from PluginUtils import Tool, relationship_class, register_tool
|
|
|
|
column_names = [
|
|
_('Name'),
|
|
_('ID') ,
|
|
_('Gender'),
|
|
_('Birth Date'),
|
|
_('Birth Place'),
|
|
_('Death Date'),
|
|
_('Death Place'),
|
|
_('Spouse'),
|
|
_('Last Change'),
|
|
_('Cause of Death'),
|
|
]
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
#
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
class RelCalc(Tool.Tool, ManagedWindow.ManagedWindow):
|
|
|
|
def __init__(self, dbstate, uistate, options_class, name, callback=None):
|
|
"""
|
|
Relationship calculator class.
|
|
"""
|
|
|
|
Tool.Tool.__init__(self, dbstate, options_class, name)
|
|
ManagedWindow.ManagedWindow.__init__(self,uistate,[],self.__class__)
|
|
|
|
if not self.person:
|
|
ErrorDialog(_('Active person has not been set'),
|
|
_('You must select an active person for this '
|
|
'tool to work properly.'))
|
|
return
|
|
|
|
self.RelClass = relationship_class
|
|
self.relationship = self.RelClass(self.db)
|
|
|
|
base = os.path.dirname(__file__)
|
|
glade_file = "%s/relcalc.glade" % base
|
|
self.glade = gtk.glade.XML(glade_file,"relcalc","gramps")
|
|
|
|
name = NameDisplay.displayer.display(self.person)
|
|
self.title = _('Relationship calculator: %(person_name)s'
|
|
) % {'person_name' : name}
|
|
window = self.glade.get_widget('relcalc')
|
|
self.set_window(window,self.glade.get_widget('title'),
|
|
_('Relationship to %(person_name)s'
|
|
) % {'person_name' : name },
|
|
self.title)
|
|
|
|
self.tree = self.glade.get_widget("peopleList")
|
|
|
|
self.model = PeopleModel(self.db,None)
|
|
self.tree.set_model(self.model)
|
|
|
|
column = gtk.TreeViewColumn(_('Name'), gtk.CellRendererText(),text=0)
|
|
column.set_resizable(True)
|
|
column.set_min_width(225)
|
|
column.set_sizing(gtk.TREE_VIEW_COLUMN_FIXED)
|
|
self.tree.append_column(column)
|
|
|
|
index = 1
|
|
for pair in self.db.get_person_column_order():
|
|
if not pair[0]:
|
|
continue
|
|
name = column_names[pair[1]]
|
|
column = gtk.TreeViewColumn(name, gtk.CellRendererText(),
|
|
markup=pair[1])
|
|
column.set_resizable(True)
|
|
column.set_min_width(60)
|
|
column.set_sizing(gtk.TREE_VIEW_COLUMN_GROW_ONLY)
|
|
self.tree.append_column(column)
|
|
index += 1
|
|
|
|
self.tree.get_selection().connect('changed',self.on_apply_clicked)
|
|
|
|
self.glade.signal_autoconnect({
|
|
"on_close_clicked" : self.close,
|
|
})
|
|
|
|
self.show()
|
|
|
|
def build_menu_names(self,obj):
|
|
return (_("Relationship Calculator tool"),None)
|
|
|
|
def on_apply_clicked(self,obj):
|
|
model,node = self.tree.get_selection().get_selected()
|
|
if not node:
|
|
return
|
|
|
|
handle = model.get_value(node,len(PeopleModel.COLUMN_DEFS)-1)
|
|
other_person = self.db.get_person_from_handle(handle)
|
|
|
|
if other_person != None:
|
|
(rel_string,common) = self.relationship.get_relationship(
|
|
self.person,other_person)
|
|
# A quick way to get unique list
|
|
common = list(set(common))
|
|
length = len(common)
|
|
else:
|
|
length = 0
|
|
|
|
if other_person == None:
|
|
commontext = ""
|
|
elif length == 1:
|
|
person = self.db.get_person_from_handle(common[0])
|
|
name = NameDisplay.displayer.display(person)
|
|
commontext = " " + _("Their common ancestor is %s.") % name
|
|
elif length == 2:
|
|
p1 = self.db.get_person_from_handle(common[0])
|
|
p2 = self.db.get_person_from_handle(common[1])
|
|
p1str = NameDisplay.displayer.display(p1)
|
|
p2str = NameDisplay.displayer.display(p2)
|
|
commontext = " " + _("Their common ancestors are %s and %s."
|
|
) % (p1str,p2str)
|
|
elif length > 2:
|
|
index = 0
|
|
commontext = " " + _("Their common ancestors are: ")
|
|
for person_handle in common:
|
|
person = self.db.get_person_from_handle(person_handle)
|
|
if index != 0:
|
|
commontext = commontext + ", "
|
|
commontext = commontext + NameDisplay.displayer.display(person)
|
|
index = index + 1
|
|
commontext = commontext + "."
|
|
else:
|
|
commontext = ""
|
|
|
|
text1 = self.glade.get_widget("text1").get_buffer()
|
|
|
|
if other_person:
|
|
p1 = NameDisplay.displayer.display(self.person)
|
|
p2 = NameDisplay.displayer.display(other_person)
|
|
|
|
if other_person == None:
|
|
rstr = ""
|
|
elif self.person.handle == other_person.handle:
|
|
rstr = _("%s and %s are the same person.") % (p1,p2)
|
|
elif rel_string == "":
|
|
rstr = _("%(person)s and %(active_person)s are not related.") % {
|
|
'person' : p2, 'active_person' : p1 }
|
|
else:
|
|
rstr = _("%(person)s is the %(relationship)s of %(active_person)s."
|
|
) % {'person' : p2, 'relationship' : rel_string,
|
|
'active_person' : p1 }
|
|
|
|
text1.set_text("%s %s" % (rstr, commontext))
|
|
|
|
#------------------------------------------------------------------------
|
|
#
|
|
#
|
|
#
|
|
#------------------------------------------------------------------------
|
|
class RelCalcOptions(Tool.ToolOptions):
|
|
"""
|
|
Defines options and provides handling interface.
|
|
"""
|
|
|
|
def __init__(self,name,person_id=None):
|
|
Tool.ToolOptions.__init__(self,name,person_id)
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
#
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
register_tool(
|
|
name = 'relcalc',
|
|
category = Tool.TOOL_UTILS,
|
|
tool_class = RelCalc,
|
|
options_class = RelCalcOptions,
|
|
modes = Tool.MODE_GUI,
|
|
translated_name = _("Relationship calculator"),
|
|
status=(_("Stable")),
|
|
author_name = "Donald N. Allingham",
|
|
author_email = "don@gramps-project.org",
|
|
description=_("Calculates the relationship between two people")
|
|
)
|