gramps/gramps2/src/AddSpouse.py

#
# Gramps - a GTK+/GNOME based genealogy program
#
# Copyright (C) 2000-2004  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$

"""
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
import gnome

#-------------------------------------------------------------------------
#
# gramps modules
#
#-------------------------------------------------------------------------
import RelLib
import const
import Utils
import GrampsCfg
import ListModel
import Date

#-------------------------------------------------------------------------
#
# 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,parent,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.parent = parent
        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_spouse_help_clicked"   : self.on_spouse_help_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 on_spouse_help_clicked(self,obj):
        """Display the relevant portion of GRAMPS manual"""
        gnome.help_display('gramps-manual','gramps-edit-quick')

    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(unicode(self.relation_type.get_text()))
        if relation == "Partners":
            if self.person.get_gender() == RelLib.Person.male:
                gen = RelLib.Person.male
            else:
                gen = RelLib.Person.female
        elif self.person.get_gender() == RelLib.Person.male:
            gen = RelLib.Person.female
        else:
            gen = RelLib.Person.male

        person = RelLib.Person()
        person.set_gender(gen)
        EditPerson.EditPerson(self.parent,person,self.db,self.update_list)

    def update_list(self,epo):
        """
        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.get_id() == "":
            self.db.add_person(person)
        else:
            self.db.add_person_no_map(person,person.get_id())
        self.db.build_person_display(person.get_id())
        self.addperson(person)
        self.update_data(person.get_id())
        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.get_person(id)

        # don't do anything if the marriage already exists
        for f in self.person.get_family_id_list():
            fam = self.db.find_family_from_id(f)
            if spouse.get_id() == fam.get_mother_id() or spouse.get_id() == fam.get_father_id():
                Utils.destroy_passed_object(obj)
                return

        if not self.active_family:
            self.active_family = self.db.new_family()
            self.person.add_family_id(self.active_family.get_id())
            self.db.commit_person(self.person)
        spouse.add_family_id(self.active_family.get_id())
        self.db.commit_person(spouse)

        if self.person.get_gender() == RelLib.Person.male:
            self.active_family.set_mother_id(spouse.get_id())
            self.active_family.set_father_id(self.person.get_id())
        else:	
            self.active_family.set_father_id(spouse.get_id())
            self.active_family.set_mother_id(self.person.get_id())

        self.active_family.set_relationship(const.save_frel(unicode(self.relation_type.get_text())))
        self.db.commit_family(self.active_family)
        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 = unicode(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.get_gender()

        birth_id = self.person.get_birth_id()
        death_id = self.person.get_death_id()
        
        bday = self.db.find_event_from_id(birth_id)
        dday = self.db.find_event_from_id(death_id)
        if birth_id:
            bday = self.db.find_event_from_id(birth_id).get_date_object()
        else:
            bday = Date.Date()
            
        if death_id:
            dday = self.db.find_event_from_id(death_id).get_date_object()
        else:
            dday = Date.Date()

        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.sort_person_keys():
            data = self.db.get_person_display(key)
            if data[2] == sgender:
                continue

            if not self.showall.get_active():
                pd_id = self.db.get_person(key).get_death_id()
                pb_id = self.db.get_person(key).get_birth_id()
                
                if pd_id:
                    pdday = self.db.find_event_from_id(pd_id).get_date_object()
                else:
                    pdday = Date.Date()
                if pb_id:
                    pbday = self.db.find_event_from_id(pb_id).get_date_object()
                else:
                    pbday = Date.Date()
                    
                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()