# # Gramps - a GTK+/GNOME based genealogy program # # Copyright (C) 2000-2003 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 # """ The AddSpouse module provides the AddSpouse class that allows the user to add a new spouse to the active person. """ __author__ = "Donald N. Allingham" __version__ = "$Revision$" #------------------------------------------------------------------------- # # internationalization # #------------------------------------------------------------------------- from gettext import gettext as _ #------------------------------------------------------------------------- # # GTK/Gnome modules # #------------------------------------------------------------------------- import gtk.glade #------------------------------------------------------------------------- # # gramps modules # #------------------------------------------------------------------------- import RelLib import const import Utils import GrampsCfg import ListModel #------------------------------------------------------------------------- # # AddSpouse # #------------------------------------------------------------------------- class AddSpouse: """ Displays the AddSpouse dialog, allowing the user to create a new family with the passed person as one spouse, and another person to be selected. """ def __init__(self,db,person,update,addperson,family=None): """ Displays the AddSpouse dialog box. db - database to which to add the new family person - the current person, will be one of the parents update - function that updates the family display addperson - function that adds a person to the person view """ self.db = db self.update = update self.person = person self.addperson = addperson self.active_family = family self.glade = gtk.glade.XML(const.gladeFile, "spouseDialog","gramps") self.rel_combo = self.glade.get_widget("rel_combo") self.relation_type = self.glade.get_widget("rel_type") self.spouse_list = self.glade.get_widget("spouse_list") self.showall = self.glade.get_widget('showall') titles = [ (_('Name'),3,200), (_('ID'),1,50), (_('Birth date'),4,50), ('',0,50), ('',0,0)] self.slist = ListModel.ListModel(self.spouse_list, titles, self.select_row ) self.relation_def = self.glade.get_widget("reldef") self.ok = self.glade.get_widget('spouse_ok') self.ok.set_sensitive(0) self.rel_combo.set_popdown_strings(const.familyRelations) title = _("Choose Spouse/Partner of %s") % GrampsCfg.nameof(person) Utils.set_titles(self.glade.get_widget('spouseDialog'), self.glade.get_widget('title'),title, _('Choose Spouse/Partner')) self.glade.signal_autoconnect({ "on_select_spouse_clicked" : self.select_spouse_clicked, "on_show_toggled" : self.on_show_toggled, "on_new_spouse_clicked" : self.new_spouse_clicked, "on_rel_type_changed" : self.relation_type_changed, "destroy_passed_object" : Utils.destroy_passed_object }) self.relation_type.set_text(_("Married")) self.relation_type_changed(None) def select_row(self,obj): """ Called with a row has be unselected. Used to enable the OK button when a row has been selected. """ model,iter = self.slist.get_selected() if iter: self.ok.set_sensitive(1) else: self.ok.set_sensitive(0) def new_spouse_clicked(self,obj): """ Called when the spouse to be added does not exist, and needs to be created and added to the database """ import EditPerson relation = const.save_frel(self.relation_type.get_text()) if relation == "Partners": if self.person.getGender() == RelLib.Person.male: gen = RelLib.Person.male else: gen = RelLib.Person.female elif self.person.getGender() == RelLib.Person.male: gen = RelLib.Person.female else: gen = RelLib.Person.male person = RelLib.Person() person.setGender(gen) EditPerson.EditPerson(person,self.db,self.update_list) def update_list(self,epo,list): """ Updates the potential spouse list after a person has been added to database. Called by the QuickAdd class when the dialog has been closed. """ person = epo.person if person.getId() == "": self.db.addPerson(person) else: self.db.addPersonNoMap(person,person.getId()) self.db.buildPersonDisplay(person.getId()) self.addperson(person) self.update_data(person.getId()) self.slist.center_selected() def select_spouse_clicked(self,obj): """ Called when the spouse to be added already exists and has been selected from the list. """ model,iter = self.slist.get_selected() if not iter: return id = self.slist.get_object(iter) spouse = self.db.getPerson(id) # don't do anything if the marriage already exists for f in self.person.getFamilyList(): if spouse == f.getMother() or spouse == f.getFather(): Utils.destroy_passed_object(obj) return Utils.modified() if not self.active_family: self.active_family = self.db.newFamily() self.person.addFamily(self.active_family) spouse.addFamily(self.active_family) if self.person.getGender() == RelLib.Person.male: self.active_family.setMother(spouse) self.active_family.setFather(self.person) else: self.active_family.setFather(spouse) self.active_family.setMother(self.person) self.active_family.setRelationship(const.save_frel(self.relation_type.get_text())) Utils.destroy_passed_object(obj) self.update(self.active_family) def relation_type_changed(self,obj): self.update_data() def update_data(self,person = None): """ Called whenever the relationship type changes. Rebuilds the the potential spouse list. """ text = self.relation_type.get_text() self.relation_def.set_text(const.relationship_def(text)) # determine the gender of the people to be loaded into # the potential spouse list. If Partners is selected, use # the same gender as the current person. gender = self.person.getGender() bday = self.person.getBirth().getDateObj() dday = self.person.getDeath().getDateObj() if text == _("Partners"): if gender == RelLib.Person.male: sgender = const.female else: sgender = const.male else: if gender == RelLib.Person.male: sgender = const.male else: sgender = const.female self.entries = [] self.slist.clear() self.slist.new_model() for key in self.db.sortPersonKeys(): data = self.db.getPersonDisplay(key) if data[2] == sgender: continue if not self.showall.get_active(): pdday = self.db.getPerson(key).getDeath().getDateObj() pbday = self.db.getPerson(key).getBirth().getDateObj() if bday.getYearValid(): if pbday.getYearValid(): # reject if person birthdate differs more than # 100 years from spouse birthdate if abs(pbday.getYear() - bday.getYear()) > 100: continue if pdday.getYearValid(): # reject if person birthdate is after the spouse deathdate if bday.getLowYear() + 10 > pdday.getHighYear(): continue # reject if person birthdate is more than 100 years # before the spouse deathdate if bday.getHighYear() + 100 < pdday.getLowYear(): continue if dday.getYearValid(): if pbday.getYearValid(): # reject if person deathdate was prior to # the spouse birthdate if dday.getHighYear() < pbday.getLowYear() + 10: continue if pdday.getYearValid(): # reject if person deathdate differs more than # 100 years from spouse deathdate if abs(pdday.getYear() - dday.getYear()) > 100: continue self.slist.add([data[0],data[1],data[3],data[5],data[6]],key,person==key) self.slist.connect_model() def on_show_toggled(self,obj): self.update_data()