# -*- coding: utf-8 -*- # # Gramps - a GTK+/GNOME based genealogy program # # Copyright (C) 2000-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$ #------------------------------------------------------------------------- # # internationalization # #------------------------------------------------------------------------- from gettext import gettext as _ #------------------------------------------------------------------------- # # GTK/Gnome modules # #------------------------------------------------------------------------- import gtk.glade import gnome #------------------------------------------------------------------------- # # gramps modules # #------------------------------------------------------------------------- import RelLib import const import Utils import PeopleModel import NameDisplay import AutoComp from QuestionDialog import ErrorDialog #------------------------------------------------------------------------- # # Constants # #------------------------------------------------------------------------- UNKNOWN_REL = (RelLib.Person.CHILD_UNKNOWN, Utils.child_relations[RelLib.Person.CHILD_UNKNOWN]) #------------------------------------------------------------------------- # # SelectChild # #------------------------------------------------------------------------- class SelectChild: def __init__(self,parent,db,family,person,callback): self.parent = parent self.db = db self.callback = callback self.person = person self.family = family self.renderer = gtk.CellRendererText() self.xml = gtk.glade.XML(const.gladeFile,"select_child","gramps") if person: self.default_name = person.get_primary_name().get_surname().upper() else: self.default_name = "" self.xml.signal_autoconnect({ "on_save_child_clicked" : self.on_save_child_clicked, "on_child_help_clicked" : self.on_child_help_clicked, "on_show_toggled" : self.on_show_toggled, "destroy_passed_object" : self.close, "on_select_child_delete_event" : self.on_delete_event, }) self.select_child_list = {} self.top = self.xml.get_widget("select_child") title_label = self.xml.get_widget('title') Utils.set_titles(self.top,title_label,_('Add Child to Family')) self.add_child = self.xml.get_widget("childlist") self.mrel = self.xml.get_widget("mrel_combo") self.frel = self.xml.get_widget("frel_combo") self.frel_selector = AutoComp.StandardCustomSelector( Utils.child_relations,self.frel, RelLib.Person.CHILD_CUSTOM,RelLib.Person.CHILD_BIRTH) self.mrel_selector = AutoComp.StandardCustomSelector( Utils.child_relations,self.mrel, RelLib.Person.CHILD_CUSTOM,RelLib.Person.CHILD_BIRTH) #self.build_list(self.mrel,RelLib.Person.CHILD_BIRTH) #self.build_list(self.frel,RelLib.Person.CHILD_BIRTH) if self.family: father = self.db.get_person_from_handle(self.family.get_father_handle()) mother = self.db.get_person_from_handle(self.family.get_mother_handle()) else: if self.person.get_gender() == RelLib.Person.MALE: self.mrel.set_sensitive(False) else: self.frel.set_sensitive(False) self.refmodel = PeopleModel.PeopleModel(self.db) self.add_child.set_model(self.refmodel) self.redraw_child_list(2) self.add_itself_to_menu() self.add_columns(self.add_child) self.top.show() def build_list(self,opt_menu,sel): cell = gtk.CellRendererText() opt_menu.pack_start(cell,True) opt_menu.add_attribute(cell,'text',0) store = gtk.ListStore(str) for val in Utils.child_relations.values(): store.append(row=[val]) opt_menu.set_model(store) opt_menu.set_active(sel) def add_columns(self,tree): column = gtk.TreeViewColumn(_('Name'), self.renderer,text=0) column.set_resizable(True) #column.set_clickable(True) column.set_min_width(225) tree.append_column(column) column = gtk.TreeViewColumn(_('ID'), self.renderer,text=1) column.set_resizable(True) #column.set_clickable(True) column.set_min_width(75) tree.append_column(column) column = gtk.TreeViewColumn(_('Birth date'), self.renderer,text=3) #column.set_resizable(True) column.set_clickable(True) tree.append_column(column) def on_delete_event(self,obj,b): self.remove_itself_from_menu() def close(self,obj): self.remove_itself_from_menu() self.top.destroy() def add_itself_to_menu(self): self.parent.child_windows[self] = self self.parent_menu_item = gtk.MenuItem(_('Add Child to Family')) self.parent_menu_item.connect("activate",self.present) self.parent_menu_item.show() self.parent.winsmenu.append(self.parent_menu_item) def remove_itself_from_menu(self): del self.parent.child_windows[self] self.parent_menu_item.destroy() def present(self,obj): self.top.present() def on_child_help_clicked(self,obj): """Display the relevant portion of GRAMPS manual""" gnome.help_display('gramps-manual','gramps-edit-quick') def redraw_child_list(self,filter): return birth = self.db.get_event_from_handle(self.person.get_birth_handle()) death = self.db.get_event_from_handle(self.person.get_death_handle()) if birth: bday = birth.get_date_object() else: bday = None if death: dday = death.get_date_object() else: dday = None slist = {} for f in self.person.get_parent_family_handle_list(): if f: family = self.db.get_family_from_handle(f[0]) if family.get_father_handle(): slist[family.get_father_handle()] = 1 elif family.get_mother_handle(): slist[family.get_mother_handle()] = 1 for c in family.get_child_handle_list(): slist[c] = 1 person_list = [] for key in self.db.get_person_handles(sort_handles=True): person = self.db.get_person_from_handle(key) if filter: if slist.has_key(key) or person.get_main_parents_family_handle(): continue birth_event = self.db.get_event_from_handle(person.get_birth_handle()) if birth_event: pbday = birth_event.get_date_object() else: pbday = None death_event = self.db.get_event_from_handle(person.get_death_handle()) if death_event: pdday = death_event.get_date_object() else: pdday = None if bday and bday.getYearValid(): if pbday and pbday.getYearValid(): # reject if child birthdate < parents birthdate + 10 if pbday.getLowYear() < bday.getHighYear()+10: continue # reject if child birthdate > parents birthdate + 90 if pbday.getLowYear() > bday.getHighYear()+90: continue if pdday and pdday.getYearValid(): # reject if child deathdate < parents birthdate+ 10 if pdday.getLowYear() < bday.getHighYear()+10: continue if dday and dday.getYearValid(): if pbday and pbday.getYearValid(): # reject if childs birth date > parents deathday + 3 if pbday.getLowYear() > dday.getHighYear()+3: continue if pdday and pdday.getYearValid(): # reject if childs death date > parents deathday + 150 if pdday.getLowYear() > dday.getHighYear() + 150: continue person_list.append(person.get_handle()) node = None for idval in person_list: person = self.db.get_person_from_handle(idval) name = NameDisplay.displayer.display(person) if person.gender == RelLib.Person.MALE: gender = _("male") elif person.gender == RelLib.Person.FEMALE: gender = _("female") else: gender = _("unknown") bh = person.get_birth_handle() dh = person.get_death_handle() if bh: bdate = self.db.get_event_from_handle(bh).get_date() else: bdate = "" if dh: ddate = self.db.get_event_from_handle(bh).get_date() else: ddate = "" rdata = [name,person.get_gramps_id(),gender,bdate,ddate] node = self.refmodel.add(rdata) self.refmodel.connect_model() if node: self.refmodel.selection.select_iter(node) path = self.refmodel.model.get_path(node) col = self.add_child.get_column(0) self.add_child.scroll_to_cell(path,col,1,0.5,0.0) def select_function(self,store,path,node,id_list): id_list.append(self.refmodel.get_value(node,PeopleModel.COLUMN_INT_ID)) def get_selected_ids(self): mlist = [] self.add_child.get_selection().selected_foreach(self.select_function,mlist) return mlist def on_save_child_clicked(self,obj): idlist = self.get_selected_ids() if not idlist or not idlist[0]: return handle = idlist[0] select_child = self.db.get_person_from_handle(handle) if self.person.get_handle() == handle: ErrorDialog(_("Error selecting a child"), _("A person cannot be linked as his/her own child"), self.top) return trans = self.db.transaction_begin() if self.family == None: self.family = RelLib.Family() self.db.add_family(self.family,trans) self.person.add_family_handle(self.family.get_handle()) self.db.commit_person(self.person,trans) if self.person.get_gender() == RelLib.Person.MALE: self.family.set_father_handle(self.person.get_handle()) else: self.family.set_mother_handle(self.person.get_handle()) self.db.commit_family(self.family,trans) if handle in (self.family.get_father_handle(),self.family.get_mother_handle()): ErrorDialog(_("Error selecting a child"), _("A person cannot be linked as his/her own child"), self.top) return # TODO: Add child ordered by birth day self.family.add_child_handle(select_child.get_handle()) mrel = self.mrel_selector.get_values() mother = self.db.get_person_from_handle(self.family.get_mother_handle()) if mother and mother.get_gender() != RelLib.Person.FEMALE: if mrel[0] == RelLib.Person.CHILD_BIRTH: mrel = UNKNOWN_REL frel = self.frel_selector.get_values() father = self.db.get_person_from_handle(self.family.get_father_handle()) if father and father.get_gender() != RelLib.Person.MALE: if frel[0] == RelLib.Person.CHILD_BIRTH: frel = UNKNOWN_REL select_child.add_parent_family_handle(self.family.get_handle(), mrel,frel) self.db.commit_person(select_child,trans) self.db.commit_family(self.family,trans) n = select_child.get_primary_name().get_regular_name() self.db.transaction_commit(trans,_("Add Child to Family (%s)") % n) self.close(obj) self.callback() def on_show_toggled(self,obj): self.redraw_child_list(not obj.get_active()) def north_american(self,val): if self.person.get_gender() == RelLib.Person.MALE: return self.person.get_primary_name().get_surname() elif self.family: f = self.family.get_father_handle() if f: pname = f.get_primary_name() return (pname.get_surname_prefix(),pname.get_surname()) return ("","") def no_name(self,val): return ("","") def latin_american(self,val): if self.family: father = self.family.get_father_handle() mother = self.family.get_mother_handle() if not father or not mother: return ("","") fsn = father.get_primary_name().get_surname() msn = mother.get_primary_name().get_surname() if not father or not mother: return ("","") try: return ("","%s %s" % (fsn.split()[0],msn.split()[0])) except: return ("","") else: return ("","") def icelandic(self,val): fname = "" if self.person.get_gender() == RelLib.Person.MALE: fname = self.person.get_primary_name().get_first_name() elif self.family: f = self.family.get_father_handle() if f: fname = f.get_primary_name().get_first_name() if fname: fname = fname.split()[0] if val == 0: return ("","%ssson" % fname) elif val == 1: return ("","%sdóttir" % fname) else: return ("","")