# # 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$ #------------------------------------------------------------------------- # # Standard python modules # #------------------------------------------------------------------------- import string #------------------------------------------------------------------------- # # GNOME # #------------------------------------------------------------------------- import gtk.glade #------------------------------------------------------------------------- # # GRAMPS modules # #------------------------------------------------------------------------- import RelLib import Utils import ListModel import NameDisplay import const from gettext import gettext as _ #------------------------------------------------------------------------- # # Merge People # #------------------------------------------------------------------------- class MergePeople: def __init__(self,parent,db,person1,person2,update,ep_update=None): self.parent = parent self.db = db self.p1 = person1 self.p2 = person2 self.update = update self.ep_update = ep_update self.glade = gtk.glade.XML(const.mergeFile,"merge","gramps") self.top = self.glade.get_widget("merge") Utils.set_titles(self.top,self.glade.get_widget('title'), _('Merge Places'), _('Select the title for the merged place')) self.altname = self.glade.get_widget("altname") self.altbirth = self.glade.get_widget("altbirth") self.altdeath = self.glade.get_widget("altdeath") self.glade.signal_autoconnect({ "on_merge_clicked" : self.on_merge_clicked, "on_next_clicked" : self.on_merge_edit_clicked, "destroy_passed_object" : Utils.destroy_passed_object, }) fname = NameDisplay.displayer.display(person1) mname = NameDisplay.displayer.display(person2) Utils.set_titles(self.top, self.glade.get_widget('title'), _("Merge %s and %s") % (fname,mname), _("Merge people")) f1_handle = person1.get_main_parents_family_handle() f2_handle = person2.get_main_parents_family_handle() name1 = NameDisplay.displayer.display(person1) death1_handle = person1.get_death_handle() if death1_handle: death1 = death1_handle.get_date() dplace1 = self.place_name(death1_handle) else: death1 = "" dplace1 = "" birth1_handle = person1.get_birth_handle() if birth1_handle: birth1 = birth1_handle.get_date() bplace1 = self.place_name(birth1_handle) else: birth1 = "" bplace1 = "" name2 = NameDisplay.displayer.display(person2) death2_handle = person2.get_death_handle() if death2_handle: death2 = death2_handle.get_date() dplace2 = self.place_name(death2_handle) else: death2 = "" dplace2 = "" birth2_handle = person2.get_birth_handle() if birth2_handle: birth2 = birth2_handle.get_date() bplace2 = self.place_name(birth2_handle) else: birth2 = "" bplace2 = "" if f2_handle and not f1_handle: self.glade.get_widget("bfather2").set_active(1) else: self.glade.get_widget("bfather1").set_active(1) if f1_handle: f1 = self.db.get_family_from_handle(f1_handle) father1 = name_of(self.db.get_person_from_handle(f1.get_father_handle())) mother1 = name_of(self.db.get_person_from_handle(f1.get_mother_handle())) else: father1 = "" mother1 = "" if f2_handle: f2 = self.db.get_family_from_handle(f2_handle) father2 = name_of(self.db.get_person_from_handle(f2.get_father_handle())) mother2 = name_of(self.db.get_person_from_handle(f2.get_mother_handle())) else: father2 = "" mother2 = "" self.set_field(self.glade.get_widget("id1_text"),person1.get_gramps_id()) self.set_field(self.glade.get_widget("id2_text"),person2.get_gramps_id()) self.set_field(self.glade.get_widget("name1_text"),name1) self.set_field(self.glade.get_widget("name2_text"),name2) self.bname1 = self.glade.get_widget("bname1") self.bname1.set_active(1) self.set_field(self.glade.get_widget("birth1_text"),birth1) self.set_field(self.glade.get_widget("birth2_text"),birth2) self.set_field(self.glade.get_widget("bplace1_text"),bplace1) self.set_field(self.glade.get_widget("bplace2_text"),bplace2) if ((not birth1 and not bplace1) and (birth2 or bplace2) or (not birth1 or not bplace1) and (birth2 and bplace2)): self.glade.get_widget('bbirth2').set_active(1) else: self.glade.get_widget('bbirth1').set_active(1) if ((not death1 and not dplace1) and (death2 or dplace2) or (not death1 or not dplace1) and (death2 and dplace2)): self.glade.get_widget('death2').set_active(1) else: self.glade.get_widget('death1').set_active(1) self.set_field(self.glade.get_widget("death1_text"),death1) self.set_field(self.glade.get_widget("dplace1_text"),dplace1) self.set_field(self.glade.get_widget("death2_text"),death2) self.set_field(self.glade.get_widget("dplace2_text"),dplace2) self.set_field(self.glade.get_widget("father1"),father1) self.set_field(self.glade.get_widget("father2"),father2) self.set_field(self.glade.get_widget("mother1"),mother1) self.set_field(self.glade.get_widget("mother2"),mother2) sp1_list = [('-',0,100)] self.sp1 = ListModel.ListModel(self.glade.get_widget('spouse1'),sp1_list) self.sp2 = ListModel.ListModel(self.glade.get_widget('spouse2'),sp1_list) self.build_spouse_list(person1,self.sp1) self.build_spouse_list(person2,self.sp2) if name1 != name2: self.altname.set_sensitive(1) self.altname.set_active(1) else: self.altname.set_sensitive(0) self.altname.set_active(0) if birth1 and birth2 and birth1 != birth2: self.altbirth.set_active(1) if bplace1 and bplace2 or bplace1 != bplace2: self.altbirth.set_active(1) else: self.altbirth.set_active(0) if death1 and death2 and death1 != death2: self.altdeath.set_active(1) if dplace1 and dplace2 or dplace1 != dplace2: self.altdeath.set_active(1) else: self.altdeath.set_active(0) def build_spouse_list(self,person,widget): widget.clear() for fam_handle in person.get_family_handle_list(): fam = self.db.get_family_from_handle(fam_handle) if person.get_gender() == RelLib.Person.MALE: spouse_handle = fam.get_mother_handle() else: spouse_handle = fam.get_father_handle() if spouse_handle: spouse = self.db.get_person_from_handle(spouse_handle) else: spouse = None if spouse == None: name = "unknown" else: sname = NameDisplay.displayer.display(spouse) name = "%s [%s]" % (sname,spouse.get_gramps_id()) widget.add([name]) def set_field(self,widget,value): """Sets the string of the entry field at positions it a space 0""" widget.set_text(value) def place_name(self,event): place = event.get_place_handle() if place: return "%s (%s)" % (place.get_title(),place.get_handle()) else: return "" def empty(self,junk): pass def on_merge_edit_clicked(self,obj): import EditPerson self.on_merge_clicked(obj) # This needs to be fixed to provide an update call EditPerson.EditPerson(self.parent,self.p1,self.db,self.ep_update) def copy_note(self,one,two): if one.get_note() != two.get_note(): one.set_note("%s\n\n%s" % (one.get_note(),two.get_note())) def copy_sources(self,one,two): slist = one.get_source_references()[:] for xsrc in two.get_source_references(): for src in slist: if src.are_equal(xsrc): break else: one.add_source_reference(xsrc) def on_merge_clicked(self,obj): lst = self.p1.get_alternate_names()[:] for xdata in self.p2.get_alternate_names(): for data in lst: if data.are_equal(xdata): self.copy_note(data,xdata) self.copy_sources(data,xdata) break else: self.p1.add_alternate_name(xdata) lst = self.p1.get_attribute_list()[:] for xdata in self.p2.get_attribute_list(): for data in lst: if data.get_type() == xdata.get_type() and \ data.getValue() == xdata.get_value(): self.copy_note(data,xdata) self.copy_sources(data,xdata) break else: self.p1.add_attribute(xdata) handle_lst = self.p1.get_event_list()[:] lst = [] for data_handle in handle_lst: lst.append(self.db.get_event_from_handle(data_handle)) for xdata_handle in self.p2.get_event_list(): xdata = self.db.get_event_from_handle(xdata_handle) for data in lst: if data.are_equal(xdata): self.copy_note(data,xdata) self.copy_sources(data,xdata) break else: self.p1.add_event_handle(xdata_handle) lst = self.p1.get_address_list() for xdata in self.p2.get_address_list(): for data in lst: if data.are_equal(xdata): self.copy_note(data,xdata) self.copy_sources(data,xdata) break else: self.p1.addAddress(xdata) lst = self.p1.get_url_list()[:] for xdata in self.p2.get_url_list(): for data in lst: if data.are_equal(xdata): break else: self.p1.add_url(xdata) self.id2 = self.glade.get_widget("id2") old_id = self.p1.get_handle() if self.id2.get_active(): self.p1.set_handle(self.p2.get_handle()) if self.bname1.get_active(): if self.altname.get_active(): self.p1.add_alternate_name(self.p2.get_primary_name()) else: if self.altname.get_active(): self.p1.add_alternate_name(self.p1.get_primary_name()) self.p1.set_primary_name(self.p2.get_primary_name()) alt = self.glade.get_widget("altbirth").get_active() if self.glade.get_widget("bbirth2").get_active(): if alt: event = self.p1.get_birth_handle() event.set_name("Alternate Birth") self.p1.add_event(event) self.p1.set_birth(self.p2.get_birth_handle()) else: if alt: event = self.p2.get_birth_handle() event.set_name("Alternate Birth") self.p1.add_event(event) alt = self.glade.get_widget("altdeath").get_active() if self.glade.get_widget("bbirth2").get_active(): if alt: event = self.p1.get_death_handle() event.set_name("Alternate Death") self.p1.add_event(event) self.p1.set_death(self.p2.get_death_handle()) else: if alt: event = self.p2.get_death_handle() event.set_name("Alternate Death") self.p1.add_event(event) if self.glade.get_widget("bfather2").get_active(): orig_family = self.p1.get_main_parents_family_handle() if orig_family: orig_family.remove_child_handle(self.p1) self.p1.remove_parent_family_handle(orig_family) (source_family,mrel,frel) = self.p2.get_main_parents_family_handle() self.p1.set_main_parent_family_handle(source_family) if source_family: if self.p2 in source_family.get_child_handle_list(): source_family.remove_child_handle(self.p2) self.p2.remove_parent_family_handle(source_family) if self.p1 not in source_family.get_child_handle_list(): source_family.add_child_handle(self.p1) self.p1.add_parent_family_handle(source_family.get_handle(),mrel,frel) else: source_family = self.p2.get_main_parents_family_handle() if source_family: source_family.remove_child_handle(self.p2) self.p2.set_main_parent_family_handle(None) self.merge_families() for photo in self.p2.get_media_list(): self.p1.add_media_reference(photo) if self.p1.get_nick_name() == "": self.p1.set_nick_name(self.p2.get_nick_name()) if self.p2.get_note() != "": old_note = self.p1.get_note() if old_note: old_note = old_note + "\n\n" self.p1.set_note(old_note + self.p2.get_note()) self.copy_sources(self.p1,self.p2) try: self.db.remove_person(self.p2.get_handle()) self.db.personMap[self.p1.get_handle()] = self.p1 except: print "%s is not in the person map!" % (NameDisplay.displayer.display(self.p2)) self.update(self.p1,self.p2,old_id) Utils.destroy_passed_object(self.top) def find_family(self,family_handle): family = self.db.get_family_from_handle(family_handle) if self.p1.get_gender() == RelLib.Person.MALE: mother = family.get_mother_handle() father = self.p1.get_handle() else: father = family.get_father_handle() mother = self.p1.get_handle() for myfamily_handle in self.db.get_family_handles(): myfamily = self.db.get_family_from_handle(myfamily_handle) if myfamily.get_father_handle() == father and myfamily.get_mother_handle() == mother: return myfamily_handle return None def merge_family_pair(self,tgt_family,src_family): # copy children from source to target for child in src_family.get_child_handle_list(): if child not in tgt_family.get_child_handle_list(): parents = child.get_parent_family_handle_list() tgt_family.add_child_handle(child) if child.get_main_parents_family_handle() == src_family: child.set_main_parent_family_handle(tgt_family) i = 0 for fam in parents[:]: if fam[0] == src_family: parents[i] = (tgt_family,fam[1],fam[2]) i = i + 1 # merge family events handle_lst = tgt_family.get_event_list()[:] lst = [] for data_handle in handle_lst: lst.append(self.db.get_event_from_handle(data_handle)) for xdata in src_family.get_event_list(): for data in lst: if data.are_equal(xdata): self.copy_note(data,xdata) self.copy_sources(data,xdata) break else: tgt_family.add_event_handle(xdata_handle) # merge family attributes lst = tgt_family.get_attribute_list()[:] for xdata in src_family.get_attribute_list(): for data in lst: if data.get_type() == xdata.get_type() and \ data.getValue() == xdata.get_value(): self.copy_note(data,xdata) self.copy_sources(data,xdata) break else: tgt_family.add_attribute(xdata) # merge family notes if src_family.get_note() != "": old_note = tgt_family.get_note() if old_note: old_note = old_note + "\n\n" tgt_family.set_note(old_note + src_family.get_note()) # merge family top-level sources self.copy_sources(tgt_family,src_family) # merge multimedia objects for photo in src_family.get_media_list(): tgt_family.add_media_reference(photo) def merge_families(self): family_num = 0 mylist = self.p2.get_family_handle_list()[:] for src_family_handle in mylist: family_num = family_num + 1 if not self.db.has_family_handle(src_family_handle): continue if src_family_handle in self.p1.get_family_handle_list(): continue src_family = self.db.get_family_from_handle(src_family_handle) tgt_family_handle = self.find_family(src_family_handle) tgt_family = self.db.get_family_from_handle(tgt_family_handle) # # This is the case where a new family to be added to the # p1 as a result of the merge already exists as a # family. In this case, we need to remove the old source # family (with the pre-merge identity of the p1) from # both the parents # if tgt_family_handle in self.p1.get_family_handle_list(): if tgt_family.get_father_handle() != None and \ src_family_handle in tgt_family.get_family_handle_list(): tgt_family.get_father_handle().remove_family_handle(src_family_handle) if tgt_family.get_mother_handle() != None and \ src_family in tgt_family.get_mother_handle().get_family_handle_list(): tgt_family.get_mother_handle().remove_family_handle(src_family) self.merge_family_pair(tgt_family,src_family) # delete the old source family del self.db.get_family_handles()[src_family.get_handle()] continue # This is the case where a new family to be added # and it is not already in the list. if tgt_family: # tgt_family a duplicate family, transfer children from # the p2 family, and delete the family. Not sure # what to do about marriage/divorce date/place yet. # transfer child to new family, alter children to # point to the correct family self.merge_family_pair(tgt_family,src_family) # change parents of the family to point to the new # family if src_family.get_father_handle(): src_family.get_father_handle().remove_family_handle(src_family.get_handle()) src_family.get_father_handle().add_family_handle(tgt_family.get_handle()) if src_family.get_mother_handle(): src_family.get_mother_handle().remove_family_handle(src_family.get_handle()) src_family.get_mother_handle().add_family_handle(tgt_family.get_handle()) del self.db.get_family_handles()[src_family.get_handle()] else: if src_family not in self.p1.get_family_handle_list(): self.p1.add_family_handle(src_family) if self.p1.get_gender() == RelLib.Person.MALE: src_family.set_father_handle(self.p1) else: src_family.set_mother_handle(self.p1) self.remove_marriage(src_family,self.p2) # a little debugging here for fam_handle in self.db.get_family_handles(): fam = self.db.get_family_from_handle(fam_handle) if self.p2.get_handle() in fam.get_child_handle_list(): fam.remove_child_handle(self.p2.get_handle()) fam.add_child_handle(self.p1.get_handle()) if self.p2.get_handle() == fam.get_father_handle(): fam.set_father_handle(self.p1.get_handle()) if self.p2.get_handle() == fam.get_mother_handle(): fam.set_mother_handle(self.p1) if fam.get_father_handle() == None and fam.get_mother_handle() == None: self.delete_empty_family(fam_handle) def remove_marriage(self,family,person): if person: person.remove_family_handle(family) if family.get_father_handle() == None and family.get_mother_handle() == None: self.delete_empty_family(family) def delete_empty_family(self,family_handle): family = self.db.get_family_from_handle(family_handle) for child in family.get_child_handle_list(): if child.get_main_parents_family_handle() == family_handle: child.set_main_parent_family_handle(None) else: child.remove_parent_family_handle(family_handle) self.db.delete_family(family_handle) def compare_people(p1,p2): name1 = p1.get_primary_name() name2 = p2.get_primary_name() chance = name_match(name1,name2) if chance == -1.0 : return -1.0 birth1 = p1.get_birth_handle() death1 = p1.get_death_handle() birth2 = p2.get_birth_handle() death2 = p2.get_death_handle() value = date_match(birth1.get_date_object(),birth2.get_date_object()) if value == -1.0 : return -1.0 chance = chance + value value = date_match(death1.get_date_object(),death2.get_date_object()) if value == -1.0 : return -1.0 chance = chance + value value = place_match(birth1.get_place_handle(),birth2.get_place_handle()) if value == -1.0 : return -1.0 chance = chance + value value = place_match(death1.get_place_handle(),death2.get_place_handle()) if value == -1.0 : return -1.0 chance = chance + value ancestors = [] ancestors_of(p1,ancestors) if p2 in ancestors: return -1.0 ancestors = [] ancestors_of(p2,ancestors) if p1 in ancestors: return -1.0 f1 = p1.get_main_parents_family_handle() f2 = p2.get_main_parents_family_handle() if f1 and f1.get_father_handle(): dad1 = f1.get_father_handle().get_primary_name() else: dad1 = None if f2 and f2.get_father_handle(): dad2 = f2.get_father_handle().get_primary_name() else: dad2 = None value = name_match(dad1,dad2) if value == -1.0: return -1.0 chance = chance + value if f1 and f1.get_mother_handle(): mom1 = f1.get_mother_handle().get_primary_name() else: mom1 = None if f2 and f2.get_mother_handle(): mom2 = f2.get_mother_handle().get_primary_name() else: mom2 = None value = name_match(mom1,mom2) if value == -1.0: return -1.0 chance = chance + value for f1 in p1.get_family_handle_list(): for f2 in p2.get_family_handle_list(): if p1.get_gender() == RelLib.Person.FEMALE: father1 = f1.get_father_handle() father2 = f2.get_father_handle() if father1 and father2: if father1 == father2: chance = chance + 1.0 else: fname1 = NameDisplay.displayer.display(father1) fname2 = NameDisplay.displayer.display(father2) value = name_match(fname1,fname2) if value != -1.0: chance = chance + value else: mother1 = f1.get_mother_handle() mother2 = f2.get_mother_handle() if mother1 and mother2: if mother1 == mother2: chance = chance + 1.0 else: mname1 = NameDisplay.displayer.display(mother1) mname2 = NameDisplay.displayer.display(mother2) value = name_match(mname1,mname2) if value != -1.0: chance = chance + value return chance #----------------------------------------------------------------- # # # #----------------------------------------------------------------- def name_compare(s1,s2): return s1 == s2 #----------------------------------------------------------------- # # # #----------------------------------------------------------------- def date_match(date1,date2): if date1.get_date() == "" or date2.get_date() == "": return 0.0 if date1.get_date() == date2.get_date(): return 1.0 if date1.isRange() or date2.isRange(): return range_compare(date1,date2) date1 = date1.get_start_date() date2 = date2.get_start_date() if date1.getYear() == date2.getYear(): if date1.getMonth() == date2.getMonth(): return 0.75 if not date1.getMonthValid() or not date2.getMonthValid(): return 0.75 else: return -1.0 else: return -1.0 #----------------------------------------------------------------- # # # #----------------------------------------------------------------- def range_compare(date1,date2): d1_start = date1.get_start_date() d2_start = date2.get_start_date() d1_stop = date1.get_stop_date() d2_stop = date2.get_stop_date() if date1.isRange() and date2.isRange(): if d1_start >= d2_start and d1_start <= d2_stop or \ d2_start >= d1_start and d2_start <= d1_stop or \ d1_stop >= d2_start and d1_stop <= d2_stop or \ d2_stop >= d1_start and d2_stop <= d1_stop: return 0.5 else: return -1.0 elif date2.isRange(): if d1_start >= d2_start and d1_start <= d2_stop: return 0.5 else: return -1.0 else: if d2_start >= d1_start and d2_start <= d1_stop: return 0.5 else: return -1.0 #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def name_match(name,name1): if not name1 or not name: return 0 srn1 = name.get_surname() sfx1 = name.get_suffix() srn2 = name1.get_surname() sfx2 = name1.get_suffix() if not name_compare(srn1,srn2): return -1 if sfx1 != sfx2: if sfx1 != "" and sfx2 != "": return -1 if name.get_first_name() == name1.get_first_name(): return 1 else: list1 = string.split(name.get_first_name()) list2 = string.split(name1.get_first_name()) if len(list1) < len(list2): return list_reduce(list1,list2) else: return list_reduce(list2,list1) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def list_reduce(list1,list2): value = 0 for name in list1: for name2 in list2: if is_initial(name) and name[0] == name2[0]: value = value + 0.25 break if is_initial(name2) and name2[0] == name[0]: value = value + 0.25 break if name == name2: value = value + 0.5 break if name[0] == name2[0] and name_compare(name,name2): value = value + 0.25 break if value == 0: return -1 else: return min(value,1) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def place_match(p1,p2): if p1 == p2: return 1 if p1 == None: name1 = "" else: name1 = p1.get_title() if p2 == None: name2 = "" else: name2 = p2.get_title() if name1 == "" or name2 == "": return 0 if name1 == name2: return 1 list1 = string.split(string.replace(name1,","," ")) list2 = string.split(string.replace(name2,","," ")) value = 0 for name in list1: for name2 in list2: if name == name2: value = value + 0.5 break if name[0] == name2[0] and name_compare(name,name2): value = value + 0.25 break if value == 0: return -1 else: return min(value,1) #------------------------------------------------------------------------- # # # #------------------------------------------------------------------------- def is_initial(name): if len(name) > 2: return 0 elif len(name) == 2: if name[0] in string.uppercase and name[1] == '.': return 1 else: return name[0] in string.uppercase #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def ancestors_of(p1,lst): if p1 == None: return lst.append(p1) f1 = p1.get_main_parents_family_handle() if f1 != None: ancestors_of(f1.get_father_handle(),lst) ancestors_of(f1.get_mother_handle(),lst) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def name_of(p): if not p: return "" return "%s (%s)" % (NameDisplay.displayer.display(p),p.get_gramps_id()) #------------------------------------------------------------------------- # # Merge Places # #------------------------------------------------------------------------- class MergePlaces: """ Merges to places into a single place. Displays a dialog box that allows the places to be combined into one. """ def __init__(self,database,place1,place2,update): self.db = database self.p1 = place1 self.p2 = place2 self.update = update self.glade = gtk.glade.XML(const.mergeFile,"merge_places","gramps") self.top = self.glade.get_widget("merge_places") self.glade.get_widget("title1_text").set_text(place1.get_title()) self.glade.get_widget("title2_text").set_text(place2.get_title()) self.t3 = self.glade.get_widget("title3_text") self.t3.set_text(place1.get_title()) self.glade.signal_autoconnect({ "destroy_passed_object" : Utils.destroy_passed_object, "on_merge_places_clicked" : self.on_merge_places_clicked, }) self.top.show() def on_merge_places_clicked(self,obj): """ Performs the merge of the places when the merge button is clicked. """ t2active = self.glade.get_widget("title2").get_active() old_id = self.p1.get_handle() if t2active: self.p1.set_title(self.p2.get_title()) elif self.glade.get_widget("title3").get_active(): self.p1.set_title(unicode(self.t3.get_text())) # Set longitude if self.p1.get_longitude() == "" and self.p2.get_longitude() != "": self.p1.set_longitude(self.p2.get_longitude()) # Set latitude if self.p1.get_latitude() == "" and self.p2.get_latitude() != "": self.p1.set_latitude(self.p2.get_latitude()) # Add URLs from P2 to P1 for url in self.p2.get_url_list(): self.p1.add_url(url) # Copy photos from P2 to P1 for photo in self.p2.get_media_list(): self.p1.add_media_reference(photo) # Copy sources from P2 to P1 for source in self.p2.get_source_references(): self.p1.add_source(source) # Add notes from P2 to P1 note = self.p2.get_note() if note != "": if self.p1.get_note() == "": self.p1.set_note(note) elif self.p1.get_note() != note: self.p1.set_note("%s\n\n%s" % (self.p1.get_note(),note)) if t2active: lst = [self.p1.get_main_location()] + self.p1.get_alternate_locations() self.p1.set_main_location(self.p2.get_main_location()) for l in lst: if not l.is_empty(): self.p1.add_alternate_locations(l) else: lst = [self.p2.get_main_location()] + self.p2.get_alternate_locations() for l in lst: if not l.is_empty(): self.p1.add_alternate_locations(l) # loop through people, changing event references to P2 to P1 for key in self.db.get_person_handles(sort_handles=False): p = self.db.get_person_from_handle(key) for event in [p.get_birth_handle(), p.get_death_handle()] + p.get_event_list(): if event.get_place_handle() == self.p2: event.set_place_handle(self.p1) # loop through families, changing event references to P2 to P1 for f_handle in self.db.get_family_handles(): f = self.db.get_event_from_handle(f_handle) for event in f.get_event_list(): if event.get_place_handle() == self.p2: event.set_place_handle(self.p1) self.db.remove_place(self.p2.get_handle()) self.db.build_place_display(self.p1.get_handle(),old_id) self.update(self.p1.get_handle()) Utils.destroy_passed_object(obj) #------------------------------------------------------------------------- # # Merge Sources # #------------------------------------------------------------------------- class MergeSources: """ Merges to places into a single place. Displays a dialog box that allows the places to be combined into one. """ def __init__(self,database,src1,src2,update): self.db = database self.p1 = src1 self.p2 = src2 self.update = update self.glade = gtk.glade.XML(const.mergeFile,"merge_sources","gramps") self.top = self.glade.get_widget("merge_sources") self.title1 = self.glade.get_widget("title1") self.title2 = self.glade.get_widget("title2") self.title1.set_text(src1.get_title()) self.title2.set_text(src2.get_title()) self.author1 = self.glade.get_widget("author1") self.author2 = self.glade.get_widget("author2") self.author1.set_text(src1.get_author()) self.author2.set_text(src2.get_author()) self.abbrev1 = self.glade.get_widget("abbrev1") self.abbrev2 = self.glade.get_widget("abbrev2") self.abbrev1.set_text(src1.get_abbreviation()) self.abbrev2.set_text(src2.get_abbreviation()) self.pub1 = self.glade.get_widget("pub1") self.pub2 = self.glade.get_widget("pub2") self.pub1.set_text(src1.get_publication_info()) self.pub2.set_text(src2.get_publication_info()) self.gramps1 = self.glade.get_widget("gramps1") self.gramps2 = self.glade.get_widget("gramps2") self.gramps1.set_text(src1.get_gramps_id()) self.gramps2.set_text(src2.get_gramps_id()) self.glade.get_widget('ok').connect('clicked',self.merge) self.glade.get_widget('close').connect('clicked',self.close) self.top.show() def close(self,obj): self.top.destroy() def merge(self,obj): """ Performs the merge of the places when the merge button is clicked. """ use_title1 = self.glade.get_widget("title_btn1").get_active() use_author1 = self.glade.get_widget("author_btn1").get_active() use_abbrev1 = self.glade.get_widget("abbrev_btn1").get_active() use_pub1 = self.glade.get_widget("pub_btn1").get_active() use_gramps1 = self.glade.get_widget("gramps_btn1").get_active() old_id = self.p1.get_handle() if not use_title1: self.src1.set_title(self.src2.get_title()) if not use_author1: self.src1.set_author(self.src2.get_author()) if not use_abbrev1: self.src1.set_abbreviation(self.src2.get_abbreviation()) if not use_pub1: self.src1.set_publication_info(self.src2.get_publication_info()) if not use_gramps1: self.src1.set_gramps_id(self.src2.get_gramps_id()) # Copy photos from src2 to src1 for photo in self.src2.get_media_list(): self.src1.add_media_reference(photo) # Add notes from P2 to P1 note = self.src2.get_note() if note != "": if self.src1.get_note() == "": self.src1.set_note(note) elif self.src1.get_note() != note: self.src1.set_note("%s\n\n%s" % (self.src1.get_note(),note)) src2_map = self.src2.get_data_map() src1_map = self.src1.get_data_map() for key in src2_map.keys(): if not src1_map.has_key(key): src1_map[key] = src2_map[key] # replace handles old_handle = self.src2.get_handle() new_handle = self.src1.get_handle() for key in self.db.get_place_handles(): p = self.db.get_place_from_handle(key) for sref in p.get_source_references(): if sref.get_base_handle() == old_handle: sref.set_base_handle(new_handle) for key in self.db.get_person_handles(sort_handles=False): p = self.db.get_person_from_handle(key) name = self.name_display(p) # Sources of person for sref in p.get_source_references(): if sref.get_base_handle() == self.source.get_handle(): person_list.append((name,'')) for event_handle in p.get_event_list() + [p.get_birth_handle(), p.get_death_handle()]: if event_handle: event = self.db.get_event_from_handle(event_handle) # Personal event sources for sref in event.get_source_references(): if sref.get_base_handle() == self.source.get_handle(): p_event_list.append((name,event.get_name())) # personal event's media for v in event.get_media_list(): # personal event's media's sources for sref in v.get_source_references(): if sref.get_base_handle() == self.source.get_handle(): o_handle = v.get_reference_handle() o = self.db.get_object_from_handle(o_handle) p_event_list_media.append((name,o.get_description())) for vv in v.get_attribute_list(): # personal event's media's attribute's sources for sref in vv.get_source_references(): if sref.get_base_handle() == self.source.get_handle(): p_event_list_media_attr.append((name,vv.get_type())) # personal LDS events Sources if p.get_lds_baptism(): for sref in p.get_lds_baptism().get_source_references(): if sref.get_base_handle() == self.source.get_handle(): p_lds_list.append((name,_('LDS Baptism'))) if p.get_lds_endowment(): for sref in p.get_lds_endowment().get_source_references(): if sref.get_base_handle() == self.source.get_handle(): p_lds_list.append((name,_('Endowment'))) if p.get_lds_sealing(): for sref in p.get_lds_sealing().get_source_references(): if sref.get_base_handle() == self.source.get_handle(): p_lds_list.append((name,_('Sealed to parents'))) # Personal attribute's sources for v in p.get_attribute_list(): for sref in v.get_source_references(): if sref.get_base_handle() == self.source.get_handle(): p_attr_list.append((name,v.get_type())) # personal Names' sources for v in p.get_alternate_names() + [p.get_primary_name()]: for sref in v.get_source_references(): if sref.get_base_handle() == self.source.get_handle(): p_name_list.append((name,v.get_name())) # personal addresses' sources for v in p.get_address_list(): for sref in v.get_source_references(): if sref.get_base_handle() == self.source.get_handle(): p_addr_list.append((name,v.get_street())) # personal media sources for v in p.get_media_list(): for sref in v.get_source_references(): if sref.get_base_handle() == self.source.get_handle(): o_handle = v.get_reference_handle() o = self.db.get_object_from_handle(o_handle) p_media_list.append((name,o.get_description())) for vv in v.get_attribute_list(): # personal media's attribute's sources for sref in vv.get_source_references(): if sref.get_base_handle() == self.source.get_handle(): p_media_attr_list.append((name,vv.get_type())) # personal media's sources for object_handle in self.db.get_media_object_handles(): obj = self.db.get_object_from_handle(object_handle) name = obj.get_description() for sref in obj.get_source_references(): if sref.get_base_handle() == self.source.get_handle(): m_list.append(name) for v in obj.get_attribute_list(): # personal media attribute's sources for sref in v.get_source_references(): if sref.get_base_handle() == self.source.get_handle(): m_attr_list.append((name,v.get_type())) for family_handle in self.db.get_family_handles(): family = self.db.get_family_from_handle(family_handle) f_id = family.get_father_handle() m_id = family.get_mother_handle() if f_id: f = self.db.get_person_from_handle(f_id) if m_id: m = self.db.get_person_from_handle(m_id) if f_id and m_id: name = _("%(father)s and %(mother)s") % { "father" : self.name_display(f), "mother" : self.name_display(m)} elif f_id: name = self.name_display(f) else: name = self.name_display(m) for v_id in family.get_event_list(): v = self.db.get_event_from_handle(v_id) if not v: continue # Family events sources for sref in v.get_source_references(): if sref.get_base_handle() == self.source.get_handle(): f_event_list.append((name,v.get_name())) # Family event's media for vv in v.get_media_list(): # Family event's media's sources for sref in vv.get_source_references(): if sref.get_base_handle() == self.source.get_handle(): o_handle = vv.get_reference_handle() o = self.db.get_object_from_handle(o_handle) f_event_list_media.append((name,o.get_description())) for vvv in vv.get_attribute_list(): # Family event's media's attribute's sources for sref in vvv.get_source_references(): if sref.get_base_handle() == self.source.get_handle(): f_event_list_media_attr.append((name,vvv.get_type())) # Family LDS events' sources if family.get_lds_sealing(): for sref in family.get_lds_sealing().get_source_references(): if sref.get_base_handle() == self.source.get_handle(): f_lds_list.append((name,_('Sealed to spouse'))) # Family sources for sref in family.get_source_references(): if sref.get_base_handle() == self.source.get_handle(): family_list.append((name,'')) # Family attributes for v in family.get_attribute_list(): for sref in v.get_source_references(): if sref.get_base_handle() == self.source.get_handle(): f_attr_list.append((name,v.get_type())) # Family media for v in family.get_media_list(): # Family media sources for sref in v.get_source_references(): if sref.get_base_handle() == self.source.get_handle(): o_handle = v.get_reference_handle() o = self.db.get_object_from_handle(o_handle) f_media_list.append((name,o.get_description())) for vv in v.get_attribute_list(): # Family media's attribute's sources for sref in vv.get_source_references(): if sref.get_base_handle() == self.source.get_handle(): f_media_attr_list.append((name,vv.get_type())) self.top.destroy()