gramps/src/Merge/_MergeSource.py
Don Allingham 1772ac7d01 * src/Merge/_MergeSource.py: note handling on source merge
* src/glade/mergedata.glade: note handling on source merge


svn: r6786
2006-05-26 03:07:48 +00:00

223 lines
8.3 KiB
Python

#
# 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: MergeData.py 6777 2006-05-25 20:35:04Z dallingham $
#-------------------------------------------------------------------------
#
# Standard python modules
#
#-------------------------------------------------------------------------
from gettext import gettext as _
#-------------------------------------------------------------------------
#
# GNOME
#
#-------------------------------------------------------------------------
import gtk
#-------------------------------------------------------------------------
#
# GRAMPS modules
#
#-------------------------------------------------------------------------
import const
import GrampsDisplay
import ManagedWindow
#-------------------------------------------------------------------------
#
# Merge Sources
#
#-------------------------------------------------------------------------
class MergeSources(ManagedWindow.ManagedWindow):
"""
Merges to sources into a single source. Displays a dialog box that
allows the sources to be combined into one.
"""
def __init__(self, dbstate, uistate, new_handle, old_handle):
ManagedWindow.ManagedWindow.__init__(self, uistate, [], self.__class__)
self.db = dbstate.db
self.new_handle = new_handle
self.old_handle = old_handle
self.s1 = self.db.get_source_from_handle(self.new_handle)
self.s2 = self.db.get_source_from_handle(self.old_handle)
self.glade = gtk.glade.XML(const.merge_glade,"merge_sources","gramps")
self.set_window(self.glade.get_widget("merge_sources"),
self.glade.get_widget('title'),
_("Merge Sources"))
self.title1 = self.glade.get_widget("title1")
self.title2 = self.glade.get_widget("title2")
self.title1.set_text(self.s1.get_title())
self.title2.set_text(self.s2.get_title())
self.author1 = self.glade.get_widget("author1")
self.author2 = self.glade.get_widget("author2")
self.author1.set_text(self.s1.get_author())
self.author2.set_text(self.s2.get_author())
self.abbrev1 = self.glade.get_widget("abbrev1")
self.abbrev2 = self.glade.get_widget("abbrev2")
self.abbrev1.set_text(self.s1.get_abbreviation())
self.abbrev2.set_text(self.s2.get_abbreviation())
self.pub1 = self.glade.get_widget("pub1")
self.pub2 = self.glade.get_widget("pub2")
self.pub1.set_text(self.s1.get_publication_info())
self.pub2.set_text(self.s2.get_publication_info())
self.gramps1 = self.glade.get_widget("gramps1")
self.gramps2 = self.glade.get_widget("gramps2")
self.gramps1.set_text(self.s1.get_gramps_id())
self.gramps2.set_text(self.s2.get_gramps_id())
self.note_s1 = self.glade.get_widget('note_s1')
self.note_s2 = self.glade.get_widget('note_s2')
self.note_merge = self.glade.get_widget('note_merge')
self.note_title = self.glade.get_widget('note_title')
self.note_conflict = self.s1.get_note() and self.s2.get_note()
if self.note_conflict:
self.note_title.show()
self.note_s1.show()
self.note_s2.show()
self.note_merge.show()
self.glade.get_widget('ok').connect('clicked',self.merge)
self.glade.get_widget('cancel').connect('clicked',self.close_window)
self.glade.get_widget('help').connect('clicked',self.help)
self.show()
def close_window(self,obj):
self.close()
def help(self,obj):
"""Display the relevant portion of GRAMPS manual"""
GrampsDisplay.help('adv-merge-sources')
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.s1.set_title(self.s2.get_title())
if not use_author1:
self.s1.set_author(self.s2.get_author())
if not use_abbrev1:
self.s1.set_abbreviation(self.s2.get_abbreviation())
if not use_pub1:
self.s1.set_publication_info(self.s2.get_publication_info())
if not use_gramps1:
self.s1.set_gramps_id(self.s2.get_gramps_id())
# Copy photos from src2 to src1
for photo in self.s2.get_media_list():
self.s1.add_media_reference(photo)
# Add notes from S2 to S1
if self.note_conflict:
note1 = self.s1.get_note()
note2 = self.s2.get_note()
if self.note_s2.get_active():
self.s1.set_note(note2)
elif self.note_merge.get_active():
self.s1.set_note("%s\n\n%s" % (note1,note2))
else:
note = self.s2.get_note()
if note != "" and self.s1.get_note() == "":
self.s1.set_note(note)
src2_map = self.s2.get_data_map()
src1_map = self.s1.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
trans = self.db.transaction_begin()
self.db.remove_source(self.old_handle,trans)
self.db.commit_source(self.s1,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,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,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,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,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,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,trans)
self.db.transaction_commit(trans,_("Merge Sources"))
self.close()