441 lines
16 KiB
Python
441 lines
16 KiB
Python
|
#
|
||
|
# Gramps - a GTK+/GNOME based genealogy program
|
||
|
#
|
||
|
# Copyright (C) 2000 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
|
||
|
#
|
||
|
|
||
|
"""
|
||
|
ChooseParents interface allows users to select the paretns of an
|
||
|
individual.
|
||
|
"""
|
||
|
|
||
|
__author__ = "Donald N. Allingham"
|
||
|
__version__ = "$Revision$"
|
||
|
|
||
|
#-------------------------------------------------------------------------
|
||
|
#
|
||
|
# internationalization
|
||
|
#
|
||
|
#-------------------------------------------------------------------------
|
||
|
from intl import gettext as _
|
||
|
|
||
|
#-------------------------------------------------------------------------
|
||
|
#
|
||
|
# GTK/Gnome modules
|
||
|
#
|
||
|
#-------------------------------------------------------------------------
|
||
|
import gobject
|
||
|
import gtk.glade
|
||
|
|
||
|
#-------------------------------------------------------------------------
|
||
|
#
|
||
|
# gramps modules
|
||
|
#
|
||
|
#-------------------------------------------------------------------------
|
||
|
import RelLib
|
||
|
import const
|
||
|
import sort
|
||
|
import Utils
|
||
|
import GrampsCfg
|
||
|
|
||
|
#-------------------------------------------------------------------------
|
||
|
#
|
||
|
# ChooseParents
|
||
|
#
|
||
|
#-------------------------------------------------------------------------
|
||
|
class ChooseParents:
|
||
|
"""
|
||
|
Displays the Choose Parents dialog box, allowing the parents
|
||
|
to be edited.
|
||
|
"""
|
||
|
def __init__(self,db,person,family,family_update,full_update):
|
||
|
"""
|
||
|
Creates a ChoosePerson dialog box.
|
||
|
|
||
|
db - database associated the person
|
||
|
person - person whose parents we are selecting
|
||
|
family - current family
|
||
|
family_update - task that updates the family display
|
||
|
full_update - task that updates the main display
|
||
|
"""
|
||
|
self.db = db
|
||
|
self.person = person
|
||
|
self.family = family
|
||
|
self.family_update = family_update
|
||
|
self.full_update = full_update
|
||
|
self.old_type = ""
|
||
|
self.type = ""
|
||
|
|
||
|
if self.family:
|
||
|
self.father = self.family.getFather()
|
||
|
self.mother = self.family.getMother()
|
||
|
else:
|
||
|
self.mother = None
|
||
|
self.father = None
|
||
|
|
||
|
self.glade = gtk.glade.XML(const.gladeFile,"familyDialog")
|
||
|
self.top = self.glade.get_widget("familyDialog")
|
||
|
self.mother_rel = self.glade.get_widget("mrel")
|
||
|
self.father_rel = self.glade.get_widget("frel")
|
||
|
self.fcombo = self.glade.get_widget("prel_combo")
|
||
|
self.prel = self.glade.get_widget("prel")
|
||
|
self.title = self.glade.get_widget("chooseTitle")
|
||
|
self.father_list = self.glade.get_widget("father_list")
|
||
|
self.mother_list = self.glade.get_widget("mother_list")
|
||
|
self.flabel = self.glade.get_widget("flabel")
|
||
|
self.mlabel = self.glade.get_widget("mlabel")
|
||
|
self.fcombo.set_popdown_strings(const.familyRelations)
|
||
|
|
||
|
|
||
|
self.fmodel = gtk.ListStore(gobject.TYPE_STRING, gobject.TYPE_STRING,
|
||
|
gobject.TYPE_STRING, gobject.TYPE_STRING,
|
||
|
gobject.TYPE_STRING)
|
||
|
self.father_list.set_model(self.fmodel)
|
||
|
self.fselection = self.father_list.get_selection()
|
||
|
self.fselection.connect('changed',self.father_list_select_row)
|
||
|
|
||
|
self.mmodel = gtk.ListStore(gobject.TYPE_STRING, gobject.TYPE_STRING,
|
||
|
gobject.TYPE_STRING, gobject.TYPE_STRING,
|
||
|
gobject.TYPE_STRING)
|
||
|
self.mother_list.set_model(self.mmodel)
|
||
|
self.mselection = self.mother_list.get_selection()
|
||
|
self.mselection.connect('changed',self.mother_list_select_row)
|
||
|
|
||
|
self.build_list(self.father_list)
|
||
|
self.build_list(self.mother_list)
|
||
|
|
||
|
for (f,mr,fr) in self.person.getParentList():
|
||
|
if f == self.family:
|
||
|
self.mother_rel.set_text(_(mr))
|
||
|
self.father_rel.set_text(_(fr))
|
||
|
break
|
||
|
else:
|
||
|
self.mother_rel.set_text(_("Birth"))
|
||
|
self.father_rel.set_text(_("Birth"))
|
||
|
|
||
|
if self.family:
|
||
|
self.type = self.family.getRelationship()
|
||
|
else:
|
||
|
self.type = "Married"
|
||
|
|
||
|
self.prel.set_text(_(self.type))
|
||
|
self.redraw()
|
||
|
|
||
|
self.glade.signal_autoconnect({
|
||
|
"on_save_parents_clicked" : self.save_parents_clicked,
|
||
|
"on_add_parent_clicked" : self.add_parent_clicked,
|
||
|
"on_prel_changed" : self.parent_relation_changed,
|
||
|
"destroy_passed_object" : Utils.destroy_passed_object
|
||
|
})
|
||
|
|
||
|
text = _("Choose the Parents of %s") % GrampsCfg.nameof(self.person)
|
||
|
self.title.set_text(text)
|
||
|
|
||
|
def build_list(self,clist):
|
||
|
colno = 0
|
||
|
for title in [ (_('Name'),3,200),
|
||
|
(_('ID'),1,50),
|
||
|
(_('Birth Date'),4,50),
|
||
|
('',0,50),
|
||
|
('',0,0)]:
|
||
|
renderer = gtk.CellRendererText ()
|
||
|
column = gtk.TreeViewColumn (title[0], renderer, text=colno)
|
||
|
colno = colno + 1
|
||
|
column.set_clickable (gtk.TRUE)
|
||
|
if title[0] == '':
|
||
|
column.set_clickable(gtk.TRUE)
|
||
|
column.set_visible(gtk.FALSE)
|
||
|
else:
|
||
|
column.set_resizable(gtk.TRUE)
|
||
|
column.set_sort_column_id(title[1])
|
||
|
column.set_min_width(title[2])
|
||
|
clist.append_column(column)
|
||
|
if colno == 1:
|
||
|
column.clicked()
|
||
|
|
||
|
def redraw(self):
|
||
|
"""Redraws the potential father and mother lists"""
|
||
|
|
||
|
self.fmodel.clear()
|
||
|
self.mmodel.clear()
|
||
|
|
||
|
pkey = self.person.getId()
|
||
|
gender = self.person.getGender()
|
||
|
if self.father:
|
||
|
fid = self.father.getId()
|
||
|
else:
|
||
|
fid = None
|
||
|
if self.mother:
|
||
|
mid = self.mother.getId()
|
||
|
else:
|
||
|
mid = None
|
||
|
|
||
|
for key in self.db.getPersonKeys():
|
||
|
if pkey == key:
|
||
|
continue
|
||
|
if gender == const.unknown:
|
||
|
continue
|
||
|
|
||
|
dinfo = self.db.getPersonDisplay(key)
|
||
|
if self.type == "Partners":
|
||
|
self.set_data(self.fmodel,self.fselection,dinfo,fid)
|
||
|
self.set_data(self.mmodel,self.mselection,dinfo,mid)
|
||
|
elif dinfo[2] == const.male:
|
||
|
self.set_data(self.fmodel,self.fselection,dinfo,fid)
|
||
|
else:
|
||
|
self.set_data(self.mmodel,self.mselection,dinfo,mid)
|
||
|
|
||
|
if self.type == "Partners":
|
||
|
self.mlabel.set_label(_("Parent"))
|
||
|
self.flabel.set_label(_("Parent"))
|
||
|
else:
|
||
|
self.mlabel.set_label(_("Mother"))
|
||
|
self.flabel.set_label(_("Father"))
|
||
|
|
||
|
def set_data(self,model,selection,dinfo,key):
|
||
|
iter = model.append()
|
||
|
model.set(iter,0,dinfo[0],1,dinfo[1],2,dinfo[3],3,dinfo[5],4,dinfo[6])
|
||
|
if key == dinfo[1]:
|
||
|
selection.select_iter(iter)
|
||
|
|
||
|
def parent_relation_changed(self,obj):
|
||
|
"""Called everytime the parent relationship information is changegd"""
|
||
|
self.old_type = self.type
|
||
|
self.type = const.save_frel(obj.get_text())
|
||
|
if self.old_type == "Partners" or self.type == "Partners":
|
||
|
self.redraw()
|
||
|
|
||
|
def find_family(self,father,mother):
|
||
|
"""
|
||
|
Finds the family associated with the father and mother.
|
||
|
If one does not exist, it is created.
|
||
|
"""
|
||
|
if not father and not mother:
|
||
|
return None
|
||
|
|
||
|
families = self.db.getFamilyMap().values()
|
||
|
for family in families:
|
||
|
if family.getFather() == father and family.getMother() == mother:
|
||
|
return family
|
||
|
elif family.getFather() == mother and family.getMother() == father:
|
||
|
return family
|
||
|
|
||
|
family = self.db.newFamily()
|
||
|
family.setFather(father)
|
||
|
family.setMother(mother)
|
||
|
family.addChild(self.person)
|
||
|
|
||
|
if father:
|
||
|
father.addFamily(family)
|
||
|
if mother:
|
||
|
mother.addFamily(family)
|
||
|
return family
|
||
|
|
||
|
def mother_list_select_row(self,obj):
|
||
|
"""Called when a row is selected in the mother list. Sets the
|
||
|
active mother based off the id associated with the row."""
|
||
|
|
||
|
model, iter = self.mselection.get_selected()
|
||
|
if iter:
|
||
|
id = model.get_value(iter,1)
|
||
|
self.mother = self.db.getPerson(id)
|
||
|
else:
|
||
|
self.mother = None
|
||
|
|
||
|
def father_list_select_row(self,obj):
|
||
|
"""Called when a row is selected in the father list. Sets the
|
||
|
active father based off the id associated with the row."""
|
||
|
model, iter = self.fselection.get_selected()
|
||
|
if iter:
|
||
|
id = model.get_value(iter,1)
|
||
|
self.father = self.db.getPerson(id)
|
||
|
else:
|
||
|
self.father = None
|
||
|
|
||
|
def save_parents_clicked(self,obj):
|
||
|
"""
|
||
|
Called with the OK button nis pressed. Saves the selected people as parents
|
||
|
of the main perosn.
|
||
|
"""
|
||
|
mother_rel = const.childRelations[self.mother_rel.get_text()]
|
||
|
father_rel = const.childRelations[self.father_rel.get_text()]
|
||
|
|
||
|
if self.father or self.mother:
|
||
|
if self.mother and not self.father:
|
||
|
if self.mother.getGender() == RelLib.Person.male:
|
||
|
self.father = self.mother
|
||
|
self.mother = None
|
||
|
self.family = self.find_family(self.father,self.mother)
|
||
|
elif self.father and not self.mother:
|
||
|
if self.father.getGender() == RelLib.Person.female:
|
||
|
self.mother = self.father
|
||
|
self.father = None
|
||
|
self.family = self.find_family(self.father,self.mother)
|
||
|
elif self.mother.getGender() != self.father.getGender():
|
||
|
if self.type == "Partners":
|
||
|
self.type = "Unknown"
|
||
|
if self.father.getGender() == RelLib.Person.female:
|
||
|
x = self.father
|
||
|
self.father = self.mother
|
||
|
self.mother = x
|
||
|
self.family = self.find_family(self.father,self.mother)
|
||
|
else:
|
||
|
self.type = "Partners"
|
||
|
self.family = self.find_family(self.father,self.mother)
|
||
|
else:
|
||
|
self.family = None
|
||
|
|
||
|
Utils.destroy_passed_object(obj)
|
||
|
if self.family:
|
||
|
self.family.setRelationship(self.type)
|
||
|
self.change_family_type(self.family,mother_rel,father_rel)
|
||
|
self.family_update(None)
|
||
|
|
||
|
def add_new_parent(self,person):
|
||
|
"""Adds a new person to either the father list or the mother list,
|
||
|
depending on the gender of the person."""
|
||
|
id = person.getId()
|
||
|
self.type = const.save_frel(self.prel.get_text())
|
||
|
dinfo = self.db.getPersonDisplay(id)
|
||
|
rdata = [dinfo[0],dinfo[3],dinfo[5],dinfo[6]]
|
||
|
|
||
|
if self.type == "Partners":
|
||
|
self.parent_relation_changed(self.prel)
|
||
|
elif person.getGender() == RelLib.Person.male:
|
||
|
self.father_list.insert(0,rdata)
|
||
|
self.father_list.set_row_data(0,id)
|
||
|
self.father_list.select_row(0,0)
|
||
|
self.father_list.sort()
|
||
|
self.father_list.moveto(self.father_list.selection[0],0)
|
||
|
else:
|
||
|
self.mother_list.insert(0,rdata)
|
||
|
self.mother_list.set_row_data(0,id)
|
||
|
self.mother_list.select_row(0,0)
|
||
|
self.mother_list.moveto(0,0)
|
||
|
self.mother_list.sort()
|
||
|
self.mother_list.moveto(self.mother_list.selection[0],0)
|
||
|
self.full_update()
|
||
|
|
||
|
def add_parent_clicked(self,obj):
|
||
|
"""Called with the Add New Person button is pressed. Calls the QuickAdd
|
||
|
class to create a new person."""
|
||
|
|
||
|
import QuickAdd
|
||
|
QuickAdd.QuickAdd(self.db,"male",self.add_new_parent)
|
||
|
|
||
|
def change_family_type(self,family,mother_rel,father_rel):
|
||
|
"""
|
||
|
Changes the family type of the specified family. If the family
|
||
|
is None, the the relationship type shoud be deleted.
|
||
|
"""
|
||
|
if self.person not in family.getChildList():
|
||
|
family.addChild(self.person)
|
||
|
for fam in self.person.getParentList():
|
||
|
if family == fam[0]:
|
||
|
if mother_rel == fam[1] and father_rel == fam[2]:
|
||
|
return
|
||
|
if mother_rel != fam[1] or father_rel != fam[2]:
|
||
|
self.person.removeAltFamily(family)
|
||
|
self.person.addAltFamily(family,mother_rel,father_rel)
|
||
|
break
|
||
|
else:
|
||
|
self.person.addAltFamily(family,mother_rel,father_rel)
|
||
|
Utils.modified()
|
||
|
|
||
|
|
||
|
class ModifyParents:
|
||
|
def __init__(self,db,person,family,family_update,full_update):
|
||
|
"""
|
||
|
Creates a ChoosePerson dialog box.
|
||
|
|
||
|
db - database associated the person
|
||
|
person - person whose parents we are selecting
|
||
|
family - current family
|
||
|
family_update - task that updates the family display
|
||
|
full_update - task that updates the main display
|
||
|
"""
|
||
|
self.db = db
|
||
|
self.person = person
|
||
|
self.family = family
|
||
|
self.family_update = family_update
|
||
|
self.full_update = full_update
|
||
|
|
||
|
self.father = self.family.getFather()
|
||
|
self.mother = self.family.getMother()
|
||
|
|
||
|
self.glade = gtk.glade.XML(const.gladeFile,"modparents")
|
||
|
self.top = self.glade.get_widget("modparents")
|
||
|
self.mother_rel = self.glade.get_widget("mrel")
|
||
|
self.father_rel = self.glade.get_widget("frel")
|
||
|
self.title = self.glade.get_widget("title")
|
||
|
self.flabel = self.glade.get_widget("flabel")
|
||
|
self.mlabel = self.glade.get_widget("mlabel")
|
||
|
|
||
|
self.orig_mrel = _("Birth")
|
||
|
self.orig_frel = _("Birth")
|
||
|
for (f,mr,fr) in self.person.getParentList():
|
||
|
if f == self.family:
|
||
|
self.orig_mrel = _(mr)
|
||
|
self.orig_frel = _(fr)
|
||
|
|
||
|
self.mother_rel.set_text(self.orig_mrel)
|
||
|
self.father_rel.set_text(self.orig_frel)
|
||
|
|
||
|
self.glade.signal_autoconnect({
|
||
|
"on_save_parents_clicked" : self.save_parents_clicked,
|
||
|
"destroy_passed_object" : Utils.destroy_passed_object
|
||
|
})
|
||
|
|
||
|
text = _("<b>Modify the Parents of %s</b>") % GrampsCfg.nameof(self.person)
|
||
|
self.title.set_text(text)
|
||
|
self.title.set_use_markup(gtk.TRUE)
|
||
|
|
||
|
if self.family.getRelationship() == "Partners":
|
||
|
self.mlabel.set_label(_("Parent"))
|
||
|
self.flabel.set_label(_("Parent"))
|
||
|
else:
|
||
|
self.mlabel.set_label(_("Mother"))
|
||
|
self.flabel.set_label(_("Father"))
|
||
|
|
||
|
|
||
|
if self.father:
|
||
|
self.glade.get_widget("fname").set_text(self.father.getPrimaryName().getName())
|
||
|
else:
|
||
|
self.father_rel.set_senstive(0)
|
||
|
|
||
|
if self.father:
|
||
|
self.glade.get_widget("mname").set_text(self.mother.getPrimaryName().getName())
|
||
|
else:
|
||
|
self.mother_rel.set_senstive(0)
|
||
|
|
||
|
|
||
|
def save_parents_clicked(self,obj):
|
||
|
"""
|
||
|
Called with the OK button nis pressed. Saves the selected people as parents
|
||
|
of the main perosn.
|
||
|
"""
|
||
|
mother_rel = const.childRelations[self.mother_rel.get_text()]
|
||
|
father_rel = const.childRelations[self.father_rel.get_text()]
|
||
|
|
||
|
Utils.destroy_passed_object(self.top)
|
||
|
if mother_rel != self.orig_mrel or father_rel != self.orig_frel:
|
||
|
self.person.removeAltFamily(self.family)
|
||
|
self.person.addAltFamily(self.family,mother_rel,father_rel)
|
||
|
self.family_update(None)
|
||
|
Utils.modified()
|
||
|
|