# # 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$ #------------------------------------------------------------------------- # # Standard python modules # #------------------------------------------------------------------------- from gettext import gettext as _ #------------------------------------------------------------------------- # # GNOME # #------------------------------------------------------------------------- import gtk #------------------------------------------------------------------------- # # GRAMPS modules # #------------------------------------------------------------------------- import Utils import const #------------------------------------------------------------------------- # # 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,new_handle,old_handle,update): self.db = database self.new_handle = new_handle self.old_handle = old_handle self.p1 = self.db.get_place_from_handle(self.new_handle) self.p2 = self.db.get_place_from_handle(self.old_handle) self.update = update self.trans = self.db.transaction_begin() self.glade = gtk.glade.XML(const.mergeFile,"merge_places","gramps") self.top = self.glade.get_widget("merge_places") Utils.set_titles(self.top,self.glade.get_widget('title'), _("Select title")) self.glade.get_widget("title1_text").set_text(self.p1.get_title()) self.glade.get_widget("title2_text").set_text(self.p2.get_title()) self.t3 = self.glade.get_widget("title3_text") self.t3.set_text(self.p1.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() 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) # remove old and commit new source self.db.remove_place(self.old_handle,self.trans) self.db.commit_place(self.p1,self.trans) # replace references in other objetcs # events for handle in self.db.get_event_handles(): event = self.db.get_event_from_handle(handle) if event.get_place_handle() == self.old_handle: event.set_place_handle(self.new_handle) self.db.commit_event(event,self.trans) # personal LDS ordinances for handle in self.db.get_person_handles(sort_handles=False): person = self.db.get_person_from_handle(handle) ord_list = [ordinance for ordinance \ in [person.lds_bapt,person.lds_endow,person.lds_seal] if (ordinance and \ ordinance.get_place_handle() == self.old_handle) ] if ord_list: for ordinance in ord_list: ordinance.set_place_handle(self.new_handle) self.db.commit_person(person,self.trans) # family LDS ordinance for handle in self.db.get_family_handles(): family = self.db.get_family_from_handle(handle) if family.lds_seal and \ family.lds_seal.get_place_handle() == self.old_handle: family.lds_seal.set_place_handle(self.new_handle) self.db.commit_family(family,self.trans) self.db.transaction_commit(self.trans,_("Merge Places")) self.update() 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,new_handle,old_handle,update): self.db = database self.new_handle = new_handle self.old_handle = old_handle self.src1 = self.db.get_source_from_handle(self.new_handle) self.src2 = self.db.get_source_from_handle(self.old_handle) 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(self.src1.get_title()) self.title2.set_text(self.src2.get_title()) self.author1 = self.glade.get_widget("author1") self.author2 = self.glade.get_widget("author2") self.author1.set_text(self.src1.get_author()) self.author2.set_text(self.src2.get_author()) self.abbrev1 = self.glade.get_widget("abbrev1") self.abbrev2 = self.glade.get_widget("abbrev2") self.abbrev1.set_text(self.src1.get_abbreviation()) self.abbrev2.set_text(self.src2.get_abbreviation()) self.pub1 = self.glade.get_widget("pub1") self.pub2 = self.glade.get_widget("pub2") self.pub1.set_text(self.src1.get_publication_info()) self.pub2.set_text(self.src2.get_publication_info()) self.gramps1 = self.glade.get_widget("gramps1") self.gramps2 = self.glade.get_widget("gramps2") self.gramps1.set_text(self.src1.get_gramps_id()) self.gramps2.set_text(self.src2.get_gramps_id()) self.glade.get_widget('ok').connect('clicked',self.merge) self.glade.get_widget('cancel').connect('clicked',self.close) self.trans = self.db.transaction_begin() self.top.show() def close(self,obj): self.top.destroy() def merge(self,obj): """ Performs the merge of the sources 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() 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 references in other objetcs self.db.remove_source(self.old_handle,self.trans) self.db.commit_source(self.src1,self.trans) # replace handles # people for handle in self.db.get_person_handles(sort_handles=False): person = self.db.get_person_from_handle(handle) if person.has_source_reference(self.old_handle): person.replace_source_references(self.old_handle,self.new_handle) self.db.commit_person(person,self.trans) # family for handle in self.db.get_family_handles(): family = self.db.get_family_from_handle(handle) if family.has_source_reference(self.old_handle): family.replace_source_references(self.old_handle,self.new_handle) self.db.commit_family(family,self.trans) # events for handle in self.db.get_event_handles(): event = self.db.get_event_from_handle(handle) if event.has_source_reference(self.old_handle): event.replace_source_references(self.old_handle,self.new_handle) self.db.commit_event(event,self.trans) # sources for handle in self.db.get_source_handles(): source = self.db.get_source_from_handle(handle) if source.has_source_reference(self.old_handle): source.replace_source_references(self.old_handle,self.new_handle) self.db.commit_source(source,self.trans) # places for handle in self.db.get_place_handles(): place = self.db.get_place_from_handle(handle) if place.has_source_reference(self.old_handle): place.replace_source_references(self.old_handle,self.new_handle) self.db.commit_place(place,self.trans) # media for handle in self.db.get_media_object_handles(): obj = self.db.get_object_from_handle(handle) if obj.has_source_reference(self.old_handle): obj.replace_source_references(self.old_handle,self.new_handle) self.db.commit_media_object(obj,self.trans) self.db.transaction_commit(self.trans,_("Merge Sources")) self.top.destroy()