2059: need a way to compare and merge all objects, by MD Nauta

svn: r15645
This commit is contained in:
Doug Blank
2010-07-22 02:16:32 +00:00
parent 59403835fe
commit 3e28ee67ec
67 changed files with 7835 additions and 109 deletions

View File

@@ -7,9 +7,14 @@ pkgdatadir = $(datadir)/@PACKAGE@/Merge
pkgdata_PYTHON = \
__init__.py \
_MergePerson.py \
_MergePlace.py \
_MergeSource.py
mergeperson.py \
mergefamily.py \
mergeevent.py \
mergeplace.py \
mergesource.py \
mergerepository.py \
mergemedia.py \
mergenote.py
pkgpyexecdir = @pkgpyexecdir@/Merge
pkgpythondir = @pkgpythondir@/Merge

View File

@@ -23,6 +23,11 @@
"""
"""
from _MergePerson import *
from _MergePlace import *
from _MergeSource import *
from mergeperson import *
from mergefamily import *
from mergeevent import *
from mergeplace import *
from mergesource import *
from mergerepository import *
from mergemedia import *
from mergenote import *

259
src/Merge/mergeevent.py Normal file
View File

@@ -0,0 +1,259 @@
#
# Gramps - a GTK+/GNOME based genealogy program
#
# Copyright (C) 2010 Michiel D. Nauta
#
# 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$
"""
Provide merge capabilities for events.
"""
#-------------------------------------------------------------------------
#
# Gramps modules
#
#-------------------------------------------------------------------------
from gen.ggettext import sgettext as _
import const
import GrampsDisplay
import ManagedWindow
import DateHandler
import Utils
#-------------------------------------------------------------------------
#
# Gramps constants
#
#-------------------------------------------------------------------------
WIKI_HELP_PAGE = '%s_-_Entering_and_Editing_Data:_Detailed_-_part_3' % \
const.URL_MANUAL_PAGE
WIKI_HELP_SEC = _('manual|Merge_Events')
_GLADE_FILE = 'mergeevent.glade'
#-------------------------------------------------------------------------
#
# Merge Events
#
#-------------------------------------------------------------------------
class MergeEvents(ManagedWindow.ManagedWindow):
"""
Displays a dialog box that allows the events to be combined into one.
"""
def __init__(self, dbstate, uistate, handle1, handle2):
ManagedWindow.ManagedWindow.__init__(self, uistate, [], self.__class__)
self.dbstate = dbstate
database = dbstate.db
self.ev1 = database.get_event_from_handle(handle1)
self.ev2 = database.get_event_from_handle(handle2)
self.define_glade('mergeevent', _GLADE_FILE)
self.set_window(self._gladeobj.toplevel,
self.get_widget("event_title"),
_("Merge Events"))
# Detailed selection widgets
type1 = str(self.ev1.get_type())
type2 = str(self.ev2.get_type())
entry1 = self.get_widget("type1")
entry2 = self.get_widget("type2")
entry1.set_text(type1)
entry2.set_text(type2)
if entry1.get_text() == entry2.get_text():
for widget_name in ('type1', 'type2', 'type_btn1', 'type_btn2'):
self.get_widget(widget_name).set_sensitive(False)
entry1 = self.get_widget("date1")
entry2 = self.get_widget("date2")
entry1.set_text(DateHandler.get_date(self.ev1))
entry2.set_text(DateHandler.get_date(self.ev2))
if entry1.get_text() == entry2.get_text():
for widget_name in ('date1', 'date2', 'date_btn1', 'date_btn2'):
self.get_widget(widget_name).set_sensitive(False)
place1 = database.get_place_from_handle(
self.ev1.get_place_handle())
place2 = database.get_place_from_handle(
self.ev2.get_place_handle())
place1 = place1.get_title() if place1 else ""
place2 = place2.get_title() if place2 else ""
entry1 = self.get_widget("place1")
entry2 = self.get_widget("place2")
entry1.set_text(place1)
entry2.set_text(place2)
if entry1.get_text() == entry2.get_text():
for widget_name in ('place1', 'place2', 'place_btn1', 'place_btn2'):
self.get_widget(widget_name).set_sensitive(False)
entry1 = self.get_widget("desc1")
entry2 = self.get_widget("desc2")
entry1.set_text(self.ev1.get_description())
entry2.set_text(self.ev2.get_description())
if entry1.get_text() == entry2.get_text():
for widget_name in ('desc1', 'desc2', 'desc_btn1', 'desc_btn2'):
self.get_widget(widget_name).set_sensitive(False)
entry1 = self.get_widget("marker1")
entry2 = self.get_widget("marker2")
entry1.set_text(str(self.ev1.get_marker()))
entry2.set_text(str(self.ev2.get_marker()))
if entry1.get_text() == entry2.get_text():
for widget_name in ('marker1', 'marker2', 'marker_btn1',
'marker_btn2'):
self.get_widget(widget_name).set_sensitive(False)
gramps1 = self.ev1.get_gramps_id()
gramps2 = self.ev2.get_gramps_id()
entry1 = self.get_widget("gramps1")
entry2 = self.get_widget("gramps2")
entry1.set_text(gramps1)
entry2.set_text(gramps2)
if entry1.get_text() == entry2.get_text():
for widget_name in ('gramps1', 'gramps2', 'gramps_btn1',
'gramps_btn2'):
self.get_widget(widget_name).set_sensitive(False)
# Main window widgets that determine which handle survives
ppant1 = Utils.get_participant_from_event(database, handle1)
ppant2 = Utils.get_participant_from_event(database, handle2)
rbutton1 = self.get_widget("handle_btn1")
rbutton_label1 = self.get_widget("label_handle_btn1")
rbutton_label2 = self.get_widget("label_handle_btn2")
rbutton_label1.set_label("%s %s [%s]" % (type1, ppant1, gramps1))
rbutton_label2.set_label("%s %s [%s]" % (type2, ppant2, gramps2))
rbutton1.connect("toggled", self.on_handle1_toggled)
self.connect_button("event_help", self.cb_help)
self.connect_button("event_ok", self.cb_merge)
self.connect_button("event_cancel", self.close)
self.show()
def on_handle1_toggled(self, obj):
"""Preferred event changes"""
if obj.get_active():
self.get_widget("type_btn1").set_active(True)
self.get_widget("date_btn1").set_active(True)
self.get_widget("place_btn1").set_active(True)
self.get_widget("desc_btn1").set_active(True)
self.get_widget("marker_btn1").set_active(True)
self.get_widget("gramps_btn1").set_active(True)
else:
self.get_widget("type_btn2").set_active(True)
self.get_widget("date_btn2").set_active(True)
self.get_widget("place_btn2").set_active(True)
self.get_widget("desc_btn2").set_active(True)
self.get_widget("marker_btn2").set_active(True)
self.get_widget("gramps_btn2").set_active(True)
def cb_help(self, obj):
"""Display the relevant portion of the Gramps manual"""
GrampsDisplay.help(webpage = WIKI_HELP_PAGE, section = WIKI_HELP_SEC)
def cb_merge(self, obj):
"""
Perform the merge of the events when the merge button is clicked.
"""
self.uistate.set_busy_cursor(True)
use_handle1 = self.get_widget("handle_btn1").get_active()
if use_handle1:
phoenix = self.ev1
titanic = self.ev2
unselect_path = (1,)
else:
phoenix = self.ev2
titanic = self.ev1
unselect_path = (0,)
if self.get_widget("type_btn1").get_active() ^ use_handle1:
phoenix.set_type(titanic.get_type())
if self.get_widget("date_btn1").get_active() ^ use_handle1:
phoenix.set_date_object(titanic.get_date_object())
if self.get_widget("place_btn1").get_active() ^ use_handle1:
phoenix.set_place_handle(titanic.get_place_handle())
if self.get_widget("desc_btn1").get_active() ^ use_handle1:
phoenix.set_description(titanic.get_description())
if self.get_widget("marker_btn1").get_active() ^ use_handle1:
phoenix.set_marker(titanic.get_marker())
if self.get_widget("gramps_btn1").get_active() ^ use_handle1:
phoenix.set_gramps_id(titanic.get_gramps_id())
# cause is deprecated.
query = MergeEventQuery(self.dbstate, phoenix, titanic)
query.execute()
self.uistate.viewmanager.active_page.selection.unselect_path(
unselect_path)
self.uistate.set_busy_cursor(False)
self.close()
#-------------------------------------------------------------------------
#
# Merge Event Query
#
#-------------------------------------------------------------------------
class MergeEventQuery(object):
"""
Create database query to merge two events.
"""
def __init__(self, dbstate, phoenix, titanic):
self.database = dbstate.db
self.phoenix = phoenix
self.titanic = titanic
def execute(self):
"""
Merges two events into a single event.
"""
new_handle = self.phoenix.get_handle()
old_handle = self.titanic.get_handle()
self.phoenix.merge(self.titanic)
trans = self.database.transaction_begin()
for person in self.database.iter_people():
if person.has_handle_reference("Event", old_handle):
bri = person.birth_ref_index
dri = person.death_ref_index
person.replace_handle_reference("Event", old_handle, new_handle)
if person.birth_ref_index != bri and person.birth_ref_index==-1:
for index, ref in enumerate(person.get_event_ref_list()):
if ref.ref == new_handle:
event = self.phoenix
else:
event = self.database.get_event_from_handle(ref.ref)
if event.type.is_birth() and ref.role.is_primary():
person.birth_ref_index = index
break
if person.death_ref_index != dri and person.death_ref_index==-1:
for index, ref in enumerate(person.get_event_ref_list()):
if ref.ref == new_handle:
event = self.phoenix
else:
event = self.database.get_event_from_handle(ref.ref)
if event.type.is_death() and ref.role.is_primary():
person.death_ref_index = index
break
self.database.commit_person(person, trans)
for family in self.database.iter_families():
if family.has_handle_reference("Event", old_handle):
family.replace_handle_reference("Event", old_handle, new_handle)
self.database.commit_family(family, trans)
self.database.remove_event(old_handle, trans)
self.database.commit_event(self.phoenix, trans)
self.database.transaction_commit(trans, _("Merge Event Objects"))

261
src/Merge/mergefamily.py Normal file
View File

@@ -0,0 +1,261 @@
#
# Gramps - a GTK+/GNOME based genealogy program
#
# Copyright (C) 2010 Michiel D. Nauta
#
# 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$
"""
Provide merge capabilities for families.
"""
#-------------------------------------------------------------------------
#
# Gramps modules
#
#-------------------------------------------------------------------------
from gen.ggettext import sgettext as _
from gen.display.name import displayer as name_displayer
import const
import GrampsDisplay
from QuestionDialog import ErrorDialog
from Errors import MergeError
import ManagedWindow
from Merge.mergeperson import MergePersonQuery
#-------------------------------------------------------------------------
#
# Gramps constants
#
#-------------------------------------------------------------------------
WIKI_HELP_PAGE = '%s_-_Entering_and_Editing_Data:_Detailed_-_part_3' % \
const.URL_MANUAL_PAGE
WIKI_HELP_SEC = _('manual|Merge_Families')
_GLADE_FILE = 'mergefamily.glade'
#-------------------------------------------------------------------------
#
# Merge Families
#
#-------------------------------------------------------------------------
class MergeFamilies(ManagedWindow.ManagedWindow):
"""
Merges two families into a single family. Displays a dialog box that allows
the families to be combined into one.
"""
def __init__(self, dbstate, uistate, handle1, handle2):
ManagedWindow.ManagedWindow.__init__(self, uistate, [], self.__class__)
self.dbstate = dbstate
database = dbstate.db
self.fy1 = database.get_family_from_handle(handle1)
self.fy2 = database.get_family_from_handle(handle2)
self.define_glade('mergefamily', _GLADE_FILE)
self.set_window(self._gladeobj.toplevel,
self.get_widget("family_title"),
_("Merge Families"))
# Detailed selection widgets
father1 = self.fy1.get_father_handle()
father2 = self.fy2.get_father_handle()
father1 = database.get_person_from_handle(father1)
father2 = database.get_person_from_handle(father2)
father_id1 = father1.get_gramps_id()
father_id2 = father2.get_gramps_id()
father1 = name_displayer.display(father1) if father1 else ""
father2 = name_displayer.display(father2) if father2 else ""
entry1 = self.get_widget("father1")
entry2 = self.get_widget("father2")
entry1.set_text("%s [%s]" % (father1, father_id1))
entry2.set_text("%s [%s]" % (father2, father_id2))
if entry1.get_text() == entry2.get_text():
for widget_name in ('father1', 'father2', 'father_btn1',
'father_btn2'):
self.get_widget(widget_name).set_sensitive(False)
mother1 = self.fy1.get_mother_handle()
mother2 = self.fy2.get_mother_handle()
mother1 = database.get_person_from_handle(mother1)
mother2 = database.get_person_from_handle(mother2)
mother_id1 = mother1.get_gramps_id()
mother_id2 = mother2.get_gramps_id()
mother1 = name_displayer.display(mother1) if mother1 else ""
mother2 = name_displayer.display(mother2) if mother2 else ""
entry1 = self.get_widget("mother1")
entry2 = self.get_widget("mother2")
entry1.set_text("%s [%s]" % (mother1, mother_id1))
entry2.set_text("%s [%s]" % (mother2, mother_id2))
if entry1.get_text() == entry2.get_text():
for widget_name in ('mother1', 'mother2', 'mother_btn1',
'mother_btn2'):
self.get_widget(widget_name).set_sensitive(False)
entry1 = self.get_widget("rel1")
entry2 = self.get_widget("rel2")
entry1.set_text(str(self.fy1.get_relationship()))
entry2.set_text(str(self.fy2.get_relationship()))
if entry1.get_text() == entry2.get_text():
for widget_name in ('rel1', 'rel2', 'rel_btn1', 'rel_btn2'):
self.get_widget(widget_name).set_sensitive(False)
entry1 = self.get_widget("marker1")
entry2 = self.get_widget("marker2")
entry1.set_text(str(self.fy1.get_marker()))
entry2.set_text(str(self.fy2.get_marker()))
if entry1.get_text() == entry2.get_text():
for widget_name in ('marker1', 'marker2', 'marker_btn1',
'marker_btn2'):
self.get_widget(widget_name).set_sensitive(False)
gramps1 = self.fy1.get_gramps_id()
gramps2 = self.fy2.get_gramps_id()
entry1 = self.get_widget("gramps1")
entry2 = self.get_widget("gramps2")
entry1.set_text(gramps1)
entry2.set_text(gramps2)
if entry1.get_text() == entry2.get_text():
for widget_name in ('gramps1', 'gramps2', 'gramps_btn1',
'gramps_btn2'):
self.get_widget(widget_name).set_sensitive(False)
# Main window widgets that determine which handle survives
rbutton1 = self.get_widget("handle_btn1")
rbutton_label1 = self.get_widget("label_handle_btn1")
rbutton_label2 = self.get_widget("label_handle_btn2")
rbutton_label1.set_label("%s and %s [%s]" %(father1, mother1, gramps1))
rbutton_label2.set_label("%s and %s [%s]" %(father2, mother2, gramps2))
rbutton1.connect("toggled", self.on_handle1_toggled)
self.connect_button("family_help", self.cb_help)
self.connect_button("family_ok", self.cb_merge)
self.connect_button("family_cancel", self.close)
self.show()
def on_handle1_toggled(self, obj):
"""Preferred family changes"""
if obj.get_active():
self.get_widget("father_btn1").set_active(True)
self.get_widget("mother_btn1").set_active(True)
self.get_widget("rel_btn1").set_active(True)
self.get_widget("marker_btn1").set_active(True)
self.get_widget("gramps_btn1").set_active(True)
else:
self.get_widget("father_btn2").set_active(True)
self.get_widget("mother_btn2").set_active(True)
self.get_widget("rel_btn2").set_active(True)
self.get_widget("marker_btn2").set_active(True)
self.get_widget("gramps_btn2").set_active(True)
def cb_help(self, obj):
"""Display the relevant portion of the Gramps manual"""
GrampsDisplay.help(webpage = WIKI_HELP_PAGE, section = WIKI_HELP_SEC)
def cb_merge(self, obj):
"""
Perform the merge of the families when the merge button is clicked.
"""
self.uistate.set_busy_cursor(True)
need_commit = False
database = self.dbstate.db
use_handle1 = self.get_widget("handle_btn1").get_active()
if use_handle1:
phoenix = self.fy1
titanic = self.fy2
unselect_path = (1,)
else:
phoenix = self.fy2
titanic = self.fy1
unselect_path = (0,)
phoenix_father = database.get_person_from_handle(
phoenix.get_father_handle())
phoenix_mother = database.get_person_from_handle(
phoenix.get_mother_handle())
titanic_father = database.get_person_from_handle(
titanic.get_father_handle())
titanic_mother = database.get_person_from_handle(
titanic.get_mother_handle())
trans = database.transaction_begin("", True)
# Use merge persons on father and mother to merge a family; The merge
# person routine also merges families if necessary. Merging is not
# an equal operation, there is one preferred family over the other.
# The preferred family is the first listed in a persons
# family_handle_list. Since the GUI allows users to chose the
# preferred father, mother and family independent of each other, while
# the merge person routine fixes preferred family with respect to
# father and mother, the father and mother need first to be swapped
# into the right family, before the merge person routines can be called.
if self.get_widget("father_btn1").get_active() ^ use_handle1:
father_handle = phoenix.get_father_handle()
phoenix.set_father_handle(titanic.get_father_handle())
titanic.set_father_handle(father_handle)
phoenix_father.replace_handle_reference('Family',
phoenix.get_handle(), titanic.get_handle())
titanic_father.replace_handle_reference('Family',
titanic.get_handle(), phoenix.get_handle())
phoenix_father, titanic_father = titanic_father, phoenix_father
database.commit_person(phoenix_father, trans)
database.commit_person(titanic_father, trans)
database.commit_family(phoenix, trans)
database.commit_family(titanic, trans)
if self.get_widget("mother_btn1").get_active() ^ use_handle1:
mother_handle = phoenix.get_mother_handle()
phoenix.set_mother_handle(titanic.get_mother_handle())
titanic.set_mother_handle(mother_handle)
phoenix_mother.replace_handle_reference('Family',
phoenix.get_handle(), titanic.get_handle())
titanic_mother.replace_handle_reference('Family',
titanic.get_handle(), phoenix.get_handle())
phoenix_mother, titanic_mother = titanic_mother, phoenix_mother
database.commit_person(phoenix_mother, trans)
database.commit_person(titanic_mother, trans)
database.commit_family(phoenix, trans)
database.commit_family(titanic, trans)
if self.get_widget("rel_btn1").get_active() ^ use_handle1:
phoenix.set_relationship(titanic.get_relationship())
need_commit = True
if self.get_widget("marker_btn1").get_active() ^ use_handle1:
phoenix.set_marker(titanic.get_marker())
need_commit = True
if self.get_widget("gramps_btn1").get_active() ^ use_handle1:
phoenix.set_gramps_id(titanic.get_gramps_id())
need_commit = True
if need_commit:
database.commit_family(phoenix, trans)
try:
if phoenix_father != titanic_father:
query = MergePersonQuery(self.dbstate, phoenix_father,
titanic_father)
query.execute(trans)
if phoenix_mother != titanic_mother:
query = MergePersonQuery(self.dbstate, phoenix_mother,
titanic_mother)
query.execute(trans)
except MergeError, e:
ErrorDialog( _("Cannot merge people"), str(e))
# TODO: rollback
else:
database.transaction_commit(trans, _('Merge family'))
self.uistate.viewmanager.active_page.selection.unselect_path(
unselect_path)
self.uistate.set_busy_cursor(False)
self.close()

215
src/Merge/mergemedia.py Normal file
View File

@@ -0,0 +1,215 @@
#
# Gramps - a GTK+/GNOME based genealogy program
#
# Copyright (C) 2010 Michiel D. Nauta
#
# 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$
"""
Provide merge capabilities for media objects.
"""
#-------------------------------------------------------------------------
#
# Gramps modules
#
#-------------------------------------------------------------------------
from gen.ggettext import sgettext as _
import const
import GrampsDisplay
import ManagedWindow
import DateHandler
#-------------------------------------------------------------------------
#
# Gramps constants
#
#-------------------------------------------------------------------------
WIKI_HELP_PAGE = '%s_-_Entering_and_Editing_Data:_Detailed_-_part_3' % \
const.URL_MANUAL_PAGE
WIKI_HELP_SEC = _('manual|Merge_Media_Objects')
_GLADE_FILE = 'mergemedia.glade'
#-------------------------------------------------------------------------
#
# Merge Media Objects
#
#-------------------------------------------------------------------------
class MergeMediaObjects(ManagedWindow.ManagedWindow):
"""
Displays a dialog box that allows the media objects to be combined into one.
"""
def __init__(self, dbstate, uistate, handle1, handle2):
ManagedWindow.ManagedWindow.__init__(self, uistate, [], self.__class__)
self.dbstate = dbstate
database = dbstate.db
self.mo1 = database.get_object_from_handle(handle1)
self.mo2 = database.get_object_from_handle(handle2)
self.define_glade('mergeobject', _GLADE_FILE)
self.set_window(self._gladeobj.toplevel,
self.get_widget('object_title'),
_("Merge Media Objects"))
# Detailed selection Widgets
desc1 = self.mo1.get_description()
desc2 = self.mo2.get_description()
entry1 = self.get_widget("desc1")
entry2 = self.get_widget("desc2")
entry1.set_text(desc1)
entry2.set_text(desc2)
if entry1.get_text() == entry2.get_text():
for widget_name in ('desc1', 'desc2', 'desc_btn1', 'desc_btn2'):
self.get_widget(widget_name).set_sensitive(False)
entry1 = self.get_widget("path1")
entry2 = self.get_widget("path2")
entry1.set_text(self.mo1.get_path())
entry2.set_text(self.mo2.get_path())
entry1.set_position(-1)
entry2.set_position(-1)
if entry1.get_text() == entry2.get_text():
for widget_name in ('path1', 'path2', 'path_btn1', 'path_btn2'):
self.get_widget(widget_name).set_sensitive(False)
entry1 = self.get_widget("date1")
entry2 = self.get_widget("date2")
entry1.set_text(DateHandler.get_date(self.mo1))
entry2.set_text(DateHandler.get_date(self.mo2))
if entry1.get_text() == entry2.get_text():
for widget_name in ('date1', 'date2', 'date_btn1', 'date_btn2'):
self.get_widget(widget_name).set_sensitive(False)
gramps1 = self.mo1.get_gramps_id()
gramps2 = self.mo2.get_gramps_id()
entry1 = self.get_widget("gramps1")
entry2 = self.get_widget("gramps2")
entry1.set_text(gramps1)
entry2.set_text(gramps2)
if entry1.get_text() == entry2.get_text():
for widget_name in ('gramps1', 'gramps2', 'gramps_btn1',
'gramps_btn2'):
self.get_widget(widget_name).set_sensitive(False)
# Main window widgets that determine which handle survives
rbutton1 = self.get_widget("handle_btn1")
rbutton_label1 = self.get_widget("label_handle_btn1")
rbutton_label2 = self.get_widget("label_handle_btn2")
rbutton_label1.set_label("%s [%s]" % (desc1, gramps1))
rbutton_label2.set_label("%s [%s]" % (desc2, gramps2))
rbutton1.connect('toggled', self.on_handle1_toggled)
self.connect_button('object_help', self.cb_help)
self.connect_button('object_ok', self.cb_merge)
self.connect_button('object_cancel', self.close)
self.show()
def on_handle1_toggled(self, obj):
""" first chosen media object changes"""
if obj.get_active():
self.get_widget("path_btn1").set_active(True)
self.get_widget("desc_btn1").set_active(True)
self.get_widget("date_btn1").set_active(True)
self.get_widget("gramps_btn1").set_active(True)
else:
self.get_widget("path_btn2").set_active(True)
self.get_widget("desc_btn2").set_active(True)
self.get_widget("date_btn2").set_active(True)
self.get_widget("gramps_btn2").set_active(True)
def cb_help(self, obj):
"""Display the relevant portion of the Gramps manual"""
GrampsDisplay.help(webpage = WIKI_HELP_PAGE, section = WIKI_HELP_SEC)
def cb_merge(self, obj):
"""
Perform the merge of the media objects when the merge button is clicked.
"""
use_handle1 = self.get_widget("handle_btn1").get_active()
if use_handle1:
phoenix = self.mo1
titanic = self.mo2
unselect_path = (1,)
else:
phoenix = self.mo2
titanic = self.mo1
unselect_path = (0,)
if self.get_widget("path_btn1").get_active() ^ use_handle1:
phoenix.set_path(titanic.get_path())
phoenix.set_mime_type(titanic.get_mime_type())
if self.get_widget("desc_btn1").get_active() ^ use_handle1:
phoenix.set_description(titanic.get_description())
if self.get_widget("date_btn1").get_active() ^ use_handle1:
phoenix.set_date_object(titanic.get_date_object())
if self.get_widget("gramps_btn1").get_active() ^ use_handle1:
phoenix.set_gramps_id(titanic.get_gramps_id())
query = MergeMediaQuery(self.dbstate, phoenix, titanic)
query.execute()
self.uistate.viewmanager.active_page.selection.unselect_path(
unselect_path)
self.close()
class MergeMediaQuery(object):
"""
Create datqabase query to merge two media objects.
"""
def __init__(self, dbstate, phoenix, titanic):
self.database = dbstate.db
self.phoenix = phoenix
self.titanic = titanic
def execute(self):
"""
Merges two media objects into a single object.
"""
new_handle = self.phoenix.get_handle()
old_handle = self.titanic.get_handle()
self.phoenix.merge(self.titanic)
trans = self.database.transaction_begin()
for person in self.database.iter_people():
if person.has_media_reference(old_handle):
person.replace_media_references(old_handle, new_handle)
self.database.commit_person(person, trans)
for family in self.database.iter_families():
if family.has_media_reference(old_handle):
family.replace_media_references(old_handle, new_handle)
self.database.commit_family(family, trans)
for event in self.database.iter_events():
if event.has_media_reference(old_handle):
event.replace_media_references(old_handle, new_handle)
self.database.commit_event(event, trans)
for source in self.database.iter_sources():
if source.has_media_reference(old_handle):
source.replace_media_references(old_handle, new_handle)
self.database.commit_source(source, trans)
for place in self.database.iter_places():
if place.has_media_reference(old_handle):
place.replace_media_references(old_handle, new_handle)
self.database.commit_place(place, trans)
self.database.remove_object(old_handle, trans)
self.database.commit_media_object(self.phoenix, trans)
self.database.transaction_commit(trans, _("Merge Media Objects"))

251
src/Merge/mergenote.py Normal file
View File

@@ -0,0 +1,251 @@
#
# Gramps - a GTK+/GNOME based genealogy program
#
# Copyright (C) 2010 Michiel D. Nauta
#
# 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$
"""
Provide merge capabilities for notes.
"""
#-------------------------------------------------------------------------
#
# Gramps modules
#
#-------------------------------------------------------------------------
from gen.ggettext import sgettext as _
import const
import GrampsDisplay
import ManagedWindow
from gui.widgets.styledtextbuffer import StyledTextBuffer
#-------------------------------------------------------------------------
#
# Gramps constants
#
#-------------------------------------------------------------------------
WIKI_HELP_PAGE = '%s_-_Entering_and_Editing_Data:_Detailed_-_part_3' % \
const.URL_MANUAL_PAGE
WIKI_HELP_SEC = _('manual|Merge_Notes')
_GLADE_FILE = 'mergenote.glade'
#-------------------------------------------------------------------------
#
# Merge Notes
#
#-------------------------------------------------------------------------
class MergeNotes(ManagedWindow.ManagedWindow):
"""
Displays a dialog box that allows two notes to be combined into one.
"""
def __init__(self, dbstate, uistate, handle1, handle2):
ManagedWindow.ManagedWindow.__init__(self, uistate, [], self.__class__)
self.dbstate = dbstate
database = dbstate.db
self.no1 = database.get_note_from_handle(handle1)
self.no2 = database.get_note_from_handle(handle2)
self.define_glade('mergenote', _GLADE_FILE)
self.set_window(self._gladeobj.toplevel,
self.get_widget("note_title"),
_("Merge Notes"))
# Detailed selection widgets
text1 = self.no1.get_styledtext()
tv1 = self.get_widget("text1")
tb1 = StyledTextBuffer()
tv1.set_buffer(tb1)
tb1.set_text(text1)
text2 = self.no2.get_styledtext()
tv2 = self.get_widget("text2")
tb2 = StyledTextBuffer()
tv2.set_buffer(tb2)
tb2.set_text(text2)
if text1 == text2:
for widget_name in ('text1', 'text2', 'text_btn1', 'text_btn2'):
self.get_widget(widget_name).set_sensitive(False)
entry1 = self.get_widget("type1")
entry2 = self.get_widget("type2")
entry1.set_text(str(self.no1.get_type()))
entry2.set_text(str(self.no2.get_type()))
if entry1.get_text() == entry2.get_text():
for widget_name in ('type1', 'type2', 'type_btn1', 'type_btn2'):
self.get_widget(widget_name).set_sensitive(False)
format_names = (_('flowed'), _('preformatted'))
entry1 = self.get_widget("format1")
entry2 = self.get_widget("format2")
entry1.set_text(format_names[self.no1.get_format()])
entry2.set_text(format_names[self.no2.get_format()])
if entry1.get_text() == entry2.get_text():
for widget_name in ('format1', 'format2', 'format_btn1',
'format_btn2'):
self.get_widget(widget_name).set_sensitive(False)
entry1 = self.get_widget("marker1")
entry2 = self.get_widget("marker2")
entry1.set_text(str(self.no1.get_marker()))
entry2.set_text(str(self.no2.get_marker()))
if entry1.get_text() == entry2.get_text():
for widget_name in ('marker1', 'marker2', 'marker_btn1',
'marker_btn2'):
self.get_widget(widget_name).set_sensitive(False)
gramps1 = self.no1.get_gramps_id()
gramps2 = self.no2.get_gramps_id()
entry1 = self.get_widget("gramps1")
entry2 = self.get_widget("gramps2")
entry1.set_text(gramps1)
entry2.set_text(gramps2)
if entry1.get_text() == entry2.get_text():
for widget_name in ('gramps1', 'gramps2', 'gramps_btn1',
'gramps_btn2'):
self.get_widget(widget_name).set_sensitive(False)
# Main window widgets that determine which handle survives
rbutton1 = self.get_widget("handle_btn1")
rbutton_label1 = self.get_widget("label_handle_btn1")
rbutton_label2 = self.get_widget("label_handle_btn2")
text1short = self.no1.get()
if len(text1short) > 50:
text1short = text1short[0:47] + "..."
text2short = self.no2.get()
if len(text2short) > 50:
text2short = text2short[0:47] + "..."
rbutton_label1.set_label("%s [%s]" % (text1short, gramps1))
rbutton_label2.set_label("%s [%s]" % (text2short, gramps2))
rbutton1.connect("toggled", self.on_handle1_toggled)
self.connect_button("note_help", self.cb_help)
self.connect_button("note_ok", self.cb_merge)
self.connect_button("note_cancel", self.close)
self.show()
def on_handle1_toggled(self, obj):
""" preferred note changes"""
if obj.get_active():
self.get_widget("text_btn1").set_active(True)
self.get_widget("type_btn1").set_active(True)
self.get_widget("format_btn1").set_active(True)
self.get_widget("marker_btn1").set_active(True)
self.get_widget("gramps_btn1").set_active(True)
else:
self.get_widget("text_btn2").set_active(True)
self.get_widget("type_btn2").set_active(True)
self.get_widget("format_btn2").set_active(True)
self.get_widget("marker_btn2").set_active(True)
self.get_widget("gramps_btn2").set_active(True)
def cb_help(self, obj):
"""Display the relevant portion of the Gramps manual"""
GrampsDisplay.help(webpage = WIKI_HELP_PAGE, section= WIKI_HELP_SEC)
def cb_merge(self, obj):
"""
Perform the merge of the notes when the merge button is clicked.
"""
use_handle1 = self.get_widget("handle_btn1").get_active()
if use_handle1:
phoenix = self.no1
titanic = self.no2
unselect_path = (1,)
else:
phoenix = self.no2
titanic = self.no1
unselect_path = (0,)
if self.get_widget("text_btn1").get_active() ^ use_handle1:
phoenix.set_styledtext(titanic.get_styledtext())
if self.get_widget("type_btn1").get_active() ^ use_handle1:
phoenix.set_type(titanic.get_type())
if self.get_widget("format_btn1").get_active() ^ use_handle1:
phoenix.set_format(titanic.get_format())
if self.get_widget("marker_btn1").get_active() ^ use_handle1:
phoenix.set_marker(titanic.get_marker())
if self.get_widget("gramps_btn1").get_active() ^ use_handle1:
phoenix.set_gramps_id(titanic.get_gramps_id())
query = MergeNoteQuery(self.dbstate, phoenix, titanic)
query.execute()
self.uistate.viewmanager.active_page.selection.unselect_path(
unselect_path)
self.close()
#-------------------------------------------------------------------------
#
# Merge Note Query
#
#-------------------------------------------------------------------------
class MergeNoteQuery(object):
"""
Create database query to merge two notes.
"""
def __init__(self, dbstate, phoenix, titanic):
self.database = dbstate.db
self.phoenix = phoenix
self.titanic = titanic
def execute(self):
"""
Merges two notes into a single note.
"""
new_handle = self.phoenix.get_handle()
old_handle = self.titanic.get_handle()
self.phoenix.merge(self.titanic)
trans = self.database.transaction_begin()
for person in self.database.iter_people():
if person.has_note_reference(old_handle):
person.replace_note_references(old_handle, new_handle)
self.database.commit_person(person, trans)
for family in self.database.iter_families():
if family.has_note_reference(old_handle):
family.replace_note_references(old_handle, new_handle)
self.database.commit_family(family, trans)
for event in self.database.iter_events():
if event.has_note_reference(old_handle):
event.replace_note_references(old_handle, new_handle)
self.database.commit_event(event, trans)
for source in self.database.iter_sources():
if source.has_note_reference(old_handle):
source.replace_note_references(old_handle, new_handle)
self.database.commit_source(source, trans)
for place in self.database.iter_places():
if place.has_note_reference(old_handle):
place.replace_note_references(old_handle, new_handle)
self.database.commit_place(place, trans)
for obj in self.database.iter_media_objects():
if obj.has_note_reference(old_handle):
obj.replace_note_references(old_handle, new_handle)
self.database.commit_media_object(obj, trans)
for repo in self.database.iter_repositories():
if repo.has_note_reference(old_handle):
repo.replace_note_references(old_handle, new_handle)
self.database.commit_repository(repo, trans)
self.database.remove_note(old_handle, trans)
self.database.commit_note(self.phoenix, trans)
self.database.transaction_commit(trans, _("Merge Notes"))

463
src/Merge/mergeperson.py Normal file
View File

@@ -0,0 +1,463 @@
#
# Gramps - a GTK+/GNOME based genealogy program
#
# Copyright (C) 2000-2007 Donald N. Allingham
# Copyright (C) 2010 Michiel D. Nauta
# Copyright (C) 2010 Jakim Friant
#
# 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$
"""
Provide merge capabilities for persons.
"""
#-------------------------------------------------------------------------
#
# GTK/Gnome modules
#
#-------------------------------------------------------------------------
import pango
#-------------------------------------------------------------------------
#
# Gramps modules
#
#-------------------------------------------------------------------------
from gen.ggettext import sgettext as _
from gen.plug.report import utils as ReportUtils
from gen.display.name import displayer as name_displayer
import const
import GrampsDisplay
import DateHandler
from QuestionDialog import ErrorDialog
from Errors import MergeError
import ManagedWindow
#-------------------------------------------------------------------------
#
# Gramps constants
#
#-------------------------------------------------------------------------
WIKI_HELP_PAGE = "%s_-_Entering_and_Editing_Data:_Detailed_-_part_3" % \
const.URL_MANUAL_PAGE
WIKI_HELP_SEC = _("manual|Merge_People")
_GLADE_FILE = "mergeperson.glade"
sex = ( _("female"), _("male"), _("unknown") )
def name_of(person):
"""Return string with name and ID of a person."""
if not person:
return ""
return "%s [%s]" % (name_displayer.display(person), person.get_gramps_id())
class MergePeople(ManagedWindow.ManagedWindow):
"""
Displays a dialog box that allows the persons to be combined into one.
"""
def __init__(self, dbstate, uistate, handle1, handle2):
ManagedWindow.ManagedWindow.__init__(self, uistate, [], self.__class__)
self.dbstate = dbstate
database = dbstate.db
self.pr1 = database.get_person_from_handle(handle1)
self.pr2 = database.get_person_from_handle(handle2)
self.define_glade('mergeperson', _GLADE_FILE)
self.set_window(self._gladeobj.toplevel,
self.get_widget("person_title"),
_("Merge People"))
# Detailed selection widgets
name1 = name_displayer.display_name(self.pr1.get_primary_name())
name2 = name_displayer.display_name(self.pr2.get_primary_name())
entry1 = self.get_widget("name1")
entry2 = self.get_widget("name2")
entry1.set_text(name1)
entry2.set_text(name2)
if entry1.get_text() == entry2.get_text():
for widget_name in ('name1', 'name2', 'name_btn1', 'name_btn2'):
self.get_widget(widget_name).set_sensitive(False)
entry1 = self.get_widget("gender1")
entry2 = self.get_widget("gender2")
entry1.set_text(sex[self.pr1.get_gender()])
entry2.set_text(sex[self.pr2.get_gender()])
if entry1.get_text() == entry2.get_text():
for widget_name in ('gender1', 'gender2', 'gender_btn1',
'gender_btn2'):
self.get_widget(widget_name).set_sensitive(False)
entry1 = self.get_widget("marker1")
entry2 = self.get_widget("marker2")
entry1.set_text(str(self.pr1.get_marker()))
entry2.set_text(str(self.pr2.get_marker()))
if entry1.get_text() == entry2.get_text():
for widget_name in ('marker1', 'marker2', 'marker_btn1',
'marker_btn2'):
self.get_widget(widget_name).set_sensitive(False)
gramps1 = self.pr1.get_gramps_id()
gramps2 = self.pr2.get_gramps_id()
entry1 = self.get_widget("gramps1")
entry2 = self.get_widget("gramps2")
entry1.set_text(gramps1)
entry2.set_text(gramps2)
if entry1.get_text() == entry2.get_text():
for widget_name in ('gramps1', 'gramps2', 'gramps_btn1',
'gramps_btn2'):
self.get_widget(widget_name).set_sensitive(False)
# Main window widgets that determine which handle survives
rbutton1 = self.get_widget("handle_btn1")
rbutton_label1 = self.get_widget("label_handle_btn1")
rbutton_label2 = self.get_widget("label_handle_btn2")
rbutton_label1.set_label(name1 + " [" + gramps1 + "]")
rbutton_label2.set_label(name2 + " [" + gramps2 + "]")
rbutton1.connect("toggled", self.on_handle1_toggled)
expander2 = self.get_widget("expander2")
self.expander_handler = \
expander2.connect("activate", self.on_expander2_activated)
self.connect_button("person_help", self.cb_help)
self.connect_button("person_ok", self.cb_merge)
self.connect_button("person_cancel", self.close)
self.show()
def on_handle1_toggled(self, obj):
"""Preferred person changes"""
if obj.get_active():
self.get_widget("name_btn1").set_active(True)
self.get_widget("gender_btn1").set_active(True)
self.get_widget("marker_btn1").set_active(True)
self.get_widget("gramps_btn1").set_active(True)
else:
self.get_widget("name_btn2").set_active(True)
self.get_widget("gender_btn2").set_active(True)
self.get_widget("marker_btn2").set_active(True)
self.get_widget("gramps_btn2").set_active(True)
def on_expander2_activated(self, obj):
"""Context Information expander is activated"""
text1 = self.get_widget('text1')
text2 = self.get_widget('text2')
self.display(text1.get_buffer(), self.pr1)
self.display(text2.get_buffer(), self.pr2)
expander2 = self.get_widget("expander2")
expander2.disconnect(self.expander_handler)
def add(self, tobj, tag, text):
"""Add text text to text buffer tobj with formatting tag."""
text += "\n"
tobj.insert_with_tags(tobj.get_end_iter(), text, tag)
def display(self, tobj, person):
"""Fill text buffer tobj with detailed info on person person."""
database = self.dbstate.db
normal = tobj.create_tag()
normal.set_property('indent', 10)
normal.set_property('pixels-above-lines', 1)
normal.set_property('pixels-below-lines', 1)
indent = tobj.create_tag()
indent.set_property('indent', 30)
indent.set_property('pixels-above-lines', 1)
indent.set_property('pixels-below-lines', 1)
title = tobj.create_tag()
title.set_property('weight', pango.WEIGHT_BOLD)
title.set_property('scale', pango.SCALE_LARGE)
self.add(tobj, title, name_displayer.display(person))
self.add(tobj, normal, "%s:\t%s" % (_('ID'),
person.get_gramps_id()))
self.add(tobj, normal, "%s:\t%s" % (_('Gender'),
sex[person.get_gender()]))
bref = person.get_birth_ref()
if bref:
self.add(tobj, normal, "%s:\t%s" % (_('Birth'),
self.get_event_info(bref.ref)))
dref = person.get_death_ref()
if dref:
self.add(tobj, normal, "%s:\t%s" % (_('Death'),
self.get_event_info(dref.ref)))
nlist = person.get_alternate_names()
if len(nlist) > 0:
self.add(tobj, title, _("Alternate Names"))
for name in nlist:
self.add(tobj, normal,
name_displayer.display_name(name))
elist = person.get_event_ref_list()
if len(elist) > 0:
self.add(tobj, title, _("Events"))
for event_ref in person.get_event_ref_list():
event_handle = event_ref.ref
name = str(
database.get_event_from_handle(event_handle).get_type())
self.add(tobj, normal, "%s:\t%s" %
(name, self.get_event_info(event_handle)))
plist = person.get_parent_family_handle_list()
if len(plist) > 0:
self.add(tobj, title, _("Parents"))
for fid in person.get_parent_family_handle_list():
(fname, mname, gid) = self.get_parent_info(fid)
self.add(tobj, normal, "%s:\t%s" % (_('Family ID'), gid))
if fname:
self.add(tobj, indent, "%s:\t%s" % (_('Father'), fname))
if mname:
self.add(tobj, indent, "%s:\t%s" % (_('Mother'), mname))
else:
self.add(tobj, normal, _("No parents found"))
self.add(tobj, title, _("Spouses"))
slist = person.get_family_handle_list()
if len(slist) > 0:
for fid in slist:
(fname, mname, pid) = self.get_parent_info(fid)
family = database.get_family_from_handle(fid)
self.add(tobj, normal, "%s:\t%s" % (_('Family ID'), pid))
spouse_id = ReportUtils.find_spouse(person, family)
if spouse_id:
spouse = database.get_person_from_handle(spouse_id)
self.add(tobj, indent, "%s:\t%s" % (_('Spouse'),
name_of(spouse)))
relstr = str(family.get_relationship())
self.add(tobj, indent, "%s:\t%s" % (_('Type'), relstr))
event = ReportUtils.find_marriage(database, family)
if event:
self.add(tobj, indent, "%s:\t%s" % (
_('Marriage'),
self.get_event_info(event.get_handle())))
for child_ref in family.get_child_ref_list():
child = database.get_person_from_handle(child_ref.ref)
self.add(tobj, indent, "%s:\t%s" % (_('Child'),
name_of(child)))
else:
self.add(tobj, normal, _("No spouses or children found"))
alist = person.get_address_list()
if len(alist) > 0:
self.add(tobj, title, _("Addresses"))
for addr in alist:
location = ", ".join([addr.get_street(), addr.get_city(),
addr.get_state(), addr.get_country(),
addr.get_postal_code(), addr.get_phone()])
self.add(tobj, normal, location.strip())
def get_parent_info(self, fid):
"""Return tuple of father name, mother name and family ID"""
database = self.dbstate.db
family = database.get_family_from_handle(fid)
father_id = family.get_father_handle()
mother_id = family.get_mother_handle()
if father_id:
father = database.get_person_from_handle(father_id)
fname = name_of(father)
else:
fname = u""
if mother_id:
mother = database.get_person_from_handle(mother_id)
mname = name_of(mother)
else:
mname = u""
return (fname, mname, family.get_gramps_id())
def get_event_info(self, handle):
"""Return date and place of an event as string."""
date = ""
place = ""
if handle:
event = self.dbstate.db.get_event_from_handle(handle)
date = DateHandler.get_date(event)
place = self.place_name(event)
if date:
return ("%s, %s" % (date, place)) if place else date
else:
return place or ""
else:
return ""
def place_name(self, event):
"""Return place name of an event as string."""
place_id = event.get_place_handle()
if place_id:
place = self.dbstate.db.get_place_from_handle(place_id)
return place.get_title()
else:
return ""
def cb_help(self, obj):
"""Display the relevant portion of Gramps manual"""
GrampsDisplay.help(webpage = WIKI_HELP_PAGE, section = WIKI_HELP_SEC)
def cb_merge(self, obj):
"""
Perform the merge of the persons when the merge button is clicked.
"""
self.uistate.set_busy_cursor(True)
use_handle1 = self.get_widget("handle_btn1").get_active()
if use_handle1:
phoenix = self.pr1
titanic = self.pr2
unselect_path = (1,)
else:
phoenix = self.pr2
titanic = self.pr1
unselect_path = (0,)
if self.get_widget("name_btn1").get_active() ^ use_handle1:
swapname = phoenix.get_primary_name()
phoenix.set_primary_name(titanic.get_primary_name())
titanic.set_primary_name(swapname)
if self.get_widget("gender_btn1").get_active() ^ use_handle1:
phoenix.set_gender(titanic.get_gender())
if self.get_widget("marker_btn1").get_active() ^ use_handle1:
phoenix.set_marker(titanic.get_marker())
if self.get_widget("gramps_btn1").get_active() ^ use_handle1:
swapid = phoenix.get_gramps_id()
phoenix.set_gramps_id(titanic.get_gramps_id())
titanic.set_gramps_id(swapid)
try:
query = MergePersonQuery(self.dbstate, phoenix, titanic)
query.execute()
except MergeError, errstr:
ErrorDialog( _("Cannot merge people"), errstr)
self.uistate.viewmanager.active_page.selection.unselect_path(
unselect_path)
self.uistate.set_busy_cursor(False)
self.close()
class MergePersonQuery(object):
"""
Create database query to merge two persons.
"""
def __init__(self, dbstate, phoenix, titanic):
self.database = dbstate.db
self.phoenix = phoenix
self.titanic = titanic
if self.check_for_spouse(self.phoenix, self.titanic):
raise MergeError(_("Spouses cannot be merged. To merge these "
"people, you must first break the relationship between them."))
if self.check_for_child(self.phoenix, self.titanic):
raise MergeError(_("A parent and child cannot be merged. To merge "
"these people, you must first break the relationship between "
"them"))
def check_for_spouse(self, person1, person2):
"""Return if person1 and person2 are spouses of eachother."""
fs1 = set(person1.get_family_handle_list())
fs2 = set(person2.get_family_handle_list())
return len(fs1.intersection(fs2)) != 0
def check_for_child(self, person1, person2):
"""Return if person1 and person2 have a child-parent relationship."""
fs1 = set(person1.get_family_handle_list())
fp1 = set(person1.get_parent_family_handle_list())
fs2 = set(person2.get_family_handle_list())
fp2 = set(person2.get_parent_family_handle_list())
return len(fs1.intersection(fp2)) != 0 or len(fs2.intersection(fp1))
def merge_families(self, main_family_handle, family, trans):
new_handle = self.phoenix.get_handle()
family_handle = family.get_handle()
main_family = self.database.get_family_from_handle(main_family_handle)
main_family.merge(family)
for childref in family.get_child_ref_list():
child = self.database.get_person_from_handle(
childref.get_reference_handle())
if main_family_handle in child.parent_family_list:
child.remove_handle_references('Family', [family_handle])
else:
child.replace_handle_reference('Family', family_handle,
main_family_handle)
self.database.commit_person(child, trans)
self.phoenix.remove_family_handle(family_handle)
family_father_handle = family.get_father_handle()
spouse_handle = family.get_mother_handle() if \
new_handle == family_father_handle else family_father_handle
spouse = self.database.get_person_from_handle(spouse_handle)
spouse.remove_family_handle(family_handle)
self.database.commit_person(spouse, trans)
self.database.remove_family(family_handle, trans)
self.database.commit_family(main_family, trans)
def execute(self, trans=None):
"""
Merges two persons into a single person.
"""
new_handle = self.phoenix.get_handle()
old_handle = self.titanic.get_handle()
self.phoenix.merge(self.titanic)
# For now use a batch transaction, because merger of persons is
# complicated, thus is done in several steps and the database should
# be updated after each step for the calculation of the next step.
# Normal Gramps transactions only touch the database upon
# transaction_commit, not after each commit_person/commit_family.
# Unfortunately batch transactions are no transactions at all, so there
# is not possibility of rollback in case of trouble.
if trans is None:
need_commit = True
trans = self.database.transaction_begin("", True)
else:
need_commit = False
commit_persons = []
for person in self.database.iter_people():
if person.has_handle_reference('Person', old_handle):
person.replace_handle_reference('Person', old_handle,new_handle)
#self.database.commit_person(person, trans) # DEADLOCK
person_handle = person.get_handle()
if person_handle == new_handle:
self.phoenix.replace_handle_reference('Person', old_handle,
new_handle)
elif person_handle != old_handle:
commit_persons.append(person)
for person in commit_persons:
self.database.commit_person(person, trans)
for family_handle in self.phoenix.get_parent_family_handle_list():
family = self.database.get_family_from_handle(family_handle)
if family.has_handle_reference('Person', old_handle):
family.replace_handle_reference('Person', old_handle,new_handle)
self.database.commit_family(family, trans)
parent_list = []
family_handle_list = self.phoenix.get_family_handle_list()[:]
for family_handle in family_handle_list:
family = self.database.get_family_from_handle(family_handle)
parents = (family.get_father_handle(), family.get_mother_handle())
if family.has_handle_reference('Person', old_handle):
family.replace_handle_reference('Person', old_handle,new_handle)
parents = (family.get_father_handle(),
family.get_mother_handle())
# prune means merging families in this case.
if parents in parent_list:
# also merge when father_handle or mother_handle == None!
idx = parent_list.index(parents)
main_family_handle = family_handle_list[idx]
self.merge_families(main_family_handle, family, trans)
continue
self.database.commit_family(family, trans)
parent_list.append(parents)
self.database.remove_person(old_handle, trans)
self.database.commit_person(self.phoenix, trans)
if need_commit:
self.database.transaction_commit(trans, _('Merge Person'))

231
src/Merge/mergeplace.py Normal file
View File

@@ -0,0 +1,231 @@
#
# Gramps - a GTK+/GNOME based genealogy program
#
# Copyright (C) 2000-2007 Donald N. Allingham
# Copyright (C) 2010 Michiel D. Nauta
#
# 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: _MergePlace.py 14135 2010-01-25 17:45:21Z gbritton $
"""
Provide merge capabilities for places.
"""
#-------------------------------------------------------------------------
#
# GTK/Gnome modules
#
#-------------------------------------------------------------------------
import gtk
#-------------------------------------------------------------------------
#
# Gramps modules
#
#-------------------------------------------------------------------------
from gen.ggettext import sgettext as _
import const
import GrampsDisplay
import ManagedWindow
#-------------------------------------------------------------------------
#
# Gramps constants
#
#-------------------------------------------------------------------------
WIKI_HELP_PAGE = '%s_-_Entering_and_Editing_Data:_Detailed_-_part_3' % \
const.URL_MANUAL_PAGE
WIKI_HELP_SEC = _('manual|Merge_Places')
_GLADE_FILE = 'mergeplace.glade'
#-------------------------------------------------------------------------
#
# Merge Places
#
#-------------------------------------------------------------------------
class MergePlaces(ManagedWindow.ManagedWindow):
"""
Displays a dialog box that allows the places to be combined into one.
"""
def __init__(self, dbstate, uistate, handle1, handle2):
ManagedWindow.ManagedWindow.__init__(self, uistate, [], self.__class__)
self.dbstate = dbstate
database = dbstate.db
self.pl1 = database.get_place_from_handle(handle1)
self.pl2 = database.get_place_from_handle(handle2)
self.define_glade('mergeplace', _GLADE_FILE)
self.set_window(self._gladeobj.toplevel,
self.get_widget('place_title'),
_("Merge Places"))
# Detailed selection widgets
title1 = self.pl1.get_title()
title2 = self.pl2.get_title()
entry1 = self.get_widget("title1")
entry2 = self.get_widget("title2")
entry1.set_text(title1)
entry2.set_text(title2)
if entry1.get_text() == entry2.get_text():
for widget_name in ('title1', 'title2', 'title_btn1', 'title_btn2'):
self.get_widget(widget_name).set_sensitive(False)
entry1 = self.get_widget("lat1")
entry2 = self.get_widget("lat2")
entry1.set_text(self.pl1.get_latitude())
entry2.set_text(self.pl2.get_latitude())
if entry1.get_text() == entry2.get_text():
for widget_name in ('lat1', 'lat2', 'lat_btn1', 'lat_btn2'):
self.get_widget(widget_name).set_sensitive(False)
entry1 = self.get_widget("long1")
entry2 = self.get_widget("long2")
entry1.set_text(self.pl1.get_longitude())
entry2.set_text(self.pl2.get_longitude())
if entry1.get_text() == entry2.get_text():
for widget_name in ('long1', 'long2', 'long_btn1', 'long_btn2'):
self.get_widget(widget_name).set_sensitive(False)
loc1 = self.pl1.get_main_location().get_text_data_list()
loc2 = self.pl2.get_main_location().get_text_data_list()
tv1 = self.get_widget("loc1")
tv2 = self.get_widget("loc2")
tb1 = gtk.TextBuffer()
tb2 = gtk.TextBuffer()
tv1.set_buffer(tb1)
tv2.set_buffer(tb2)
tb1.set_text("\n".join(loc1))
tb2.set_text("\n".join(loc2))
if loc1 == loc2:
for widget_name in ('loc1', 'loc2', 'loc_btn1', 'loc_btn2'):
self.get_widget(widget_name).set_sensitive(False)
gramps1 = self.pl1.get_gramps_id()
gramps2 = self.pl2.get_gramps_id()
entry1 = self.get_widget("gramps1")
entry2 = self.get_widget("gramps2")
entry1.set_text(gramps1)
entry2.set_text(gramps2)
if entry1.get_text() == entry2.get_text():
for widget_name in ('gramps1', 'gramps2', 'gramps_btn1',
'gramps_btn2'):
self.get_widget(widget_name).set_sensitive(False)
# Main window widgets that determine which handle survives
rbutton1 = self.get_widget("handle_btn1")
rbutton_label1 = self.get_widget("label_handle_btn1")
rbutton_label2 = self.get_widget("label_handle_btn2")
rbutton_label1.set_label(title1 + " [" + gramps1 + "]")
rbutton_label2.set_label(title2 + " [" + gramps2 + "]")
rbutton1.connect("toggled", self.on_handle1_toggled)
self.connect_button('place_help', self.cb_help)
self.connect_button('place_ok', self.cb_merge)
self.connect_button('place_cancel', self.close)
self.show()
def on_handle1_toggled(self, obj):
"""first chosen place changes"""
if obj.get_active():
self.get_widget("title_btn1").set_active(True)
self.get_widget("lat_btn1").set_active(True)
self.get_widget("long_btn1").set_active(True)
self.get_widget("loc_btn1").set_active(True)
self.get_widget("gramps_btn1").set_active(True)
else:
self.get_widget("title_btn2").set_active(True)
self.get_widget("lat_btn2").set_active(True)
self.get_widget("long_btn2").set_active(True)
self.get_widget("loc_btn2").set_active(True)
self.get_widget("gramps_btn2").set_active(True)
def cb_help(self, obj):
"""Display the relevant portion of Gramps manual"""
GrampsDisplay.help(webpage = WIKI_HELP_PAGE, section = WIKI_HELP_SEC)
def cb_merge(self, obj):
"""
Performs the merge of the places when the merge button is clicked.
"""
self.uistate.set_busy_cursor(True)
use_handle1 = self.get_widget("handle_btn1").get_active()
if use_handle1:
phoenix = self.pl1
titanic = self.pl2
unselect_path = (1,)
else:
phoenix = self.pl2
titanic = self.pl1
unselect_path = (0,)
if self.get_widget("title_btn1").get_active() ^ use_handle1:
phoenix.set_title(titanic.get_title())
if self.get_widget("lat_btn1").get_active() ^ use_handle1:
phoenix.set_latitude(titanic.get_latitude())
if self.get_widget("long_btn1").get_active() ^ use_handle1:
phoenix.set_longitude(titanic.get_longitude())
if self.get_widget("loc_btn1").get_active() ^ use_handle1:
swaploc = phoenix.get_main_location()
phoenix.set_main_location(titanic.get_main_location())
titanic.set_main_location(swaploc)
if self.get_widget("gramps_btn1").get_active() ^ use_handle1:
phoenix.set_gramps_id(titanic.get_gramps_id())
query = MergePlaceQuery(self.dbstate, phoenix, titanic)
query.execute()
self.uistate.viewmanager.active_page.selection.unselect_path(
unselect_path)
self.uistate.set_busy_cursor(False)
self.close()
class MergePlaceQuery(object):
"""
Create database query to merge two places.
"""
def __init__(self, dbstate, phoenix, titanic):
self.database = dbstate.db
self.phoenix = phoenix
self.titanic = titanic
def execute(self):
"""
Merges to places into a single place.
"""
new_handle = self.phoenix.get_handle()
old_handle = self.titanic.get_handle()
self.phoenix.merge(self.titanic)
trans = self.database.transaction_begin()
for person in self.database.iter_people():
if person.has_handle_reference('Place', old_handle):
person.replace_handle_reference('Place', old_handle, new_handle)
self.database.commit_person(person, trans)
for family in self.database.iter_families():
if family.has_handle_reference('Place', old_handle):
family.replace_handle_reference('Place', old_handle, new_handle)
self.database.commit_family(family, trans)
for event in self.database.iter_events():
if event.has_handle_reference('Place', old_handle):
event.replace_handle_reference('Place', old_handle, new_handle)
self.database.commit_event(event, trans)
self.database.remove_place(old_handle, trans)
self.database.commit_place(self.phoenix, trans)
self.database.transaction_commit(trans, _("Merge Places"))

View File

@@ -0,0 +1,181 @@
#
# Gramps - a GTK+/GNOME based genealogy program
#
# Copyright (C) 2010 Michiel D. Nauta
#
# 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$
"""
Provide merge capabilities for repositories.
"""
#-------------------------------------------------------------------------
#
# Gramps modules
#
#-------------------------------------------------------------------------
from gen.ggettext import sgettext as _
import const
import GrampsDisplay
import ManagedWindow
#-------------------------------------------------------------------------
#
# Gramps constants
#
#-------------------------------------------------------------------------
WIKI_HELP_PAGE = '%s_-_Entering_and_Editing_Data:_Detailed_-_part_3' % \
const.URL_MANUAL_PAGE
WIKI_HELP_SEC = _('manual|Merge_Repositories')
_GLADE_FILE = 'mergerepository.glade'
#-------------------------------------------------------------------------
#
# Merge Repositories
#
#-------------------------------------------------------------------------
class MergeRepositories(ManagedWindow.ManagedWindow):
"""
Displays a dialog box that allows two repositories to be combined into one.
"""
def __init__(self, dbstate, uistate, handle1, handle2):
ManagedWindow.ManagedWindow.__init__(self, uistate, [], self.__class__)
self.dbstate = dbstate
database = dbstate.db
self.rp1 = database.get_repository_from_handle(handle1)
self.rp2 = database.get_repository_from_handle(handle2)
self.define_glade('mergerepository', _GLADE_FILE)
self.set_window(self._gladeobj.toplevel,
self.get_widget('repository_title'),
_("Merge Repositories"))
# Detailed selection widgets
name1 = self.rp1.get_name()
name2 = self.rp2.get_name()
entry1 = self.get_widget('name1')
entry2 = self.get_widget('name2')
entry1.set_text(name1)
entry2.set_text(name2)
if entry1.get_text() == entry2.get_text():
for widget_name in ('name1', 'name2', 'name_btn1', 'name_btn2'):
self.get_widget(widget_name).set_sensitive(False)
entry1 = self.get_widget('type1')
entry2 = self.get_widget('type2')
entry1.set_text(str(self.rp1.get_type()))
entry2.set_text(str(self.rp2.get_type()))
if entry1.get_text() == entry2.get_text():
for widget_name in ('type1', 'type2', 'type_btn1', 'type_btn2'):
self.get_widget(widget_name).set_sensitive(False)
gramps1 = self.rp1.get_gramps_id()
gramps2 = self.rp2.get_gramps_id()
entry1 = self.get_widget('gramps1')
entry2 = self.get_widget('gramps2')
entry1.set_text(gramps1)
entry2.set_text(gramps2)
if entry1.get_text() == entry2.get_text():
for widget_name in ('gramps1', 'gramps2', 'gramps_btn1',
'gramps_btn2'):
self.get_widget(widget_name).set_sensitive(False)
# Main window widgets that determine which handle survives
rbutton1 = self.get_widget("handle_btn1")
rbutton_label1 = self.get_widget("label_handle_btn1")
rbutton_label2 = self.get_widget("label_handle_btn2")
rbutton_label1.set_label("%s [%s]" % (name1, gramps1))
rbutton_label2.set_label("%s [%s]" % (name2, gramps2))
rbutton1.connect('toggled', self.on_handle1_toggled)
self.connect_button('repository_help', self.cb_help)
self.connect_button('repository_ok', self.cb_merge)
self.connect_button('repository_cancel', self.close)
self.show()
def on_handle1_toggled(self, obj):
""" preferred repository changes"""
if obj.get_active():
self.get_widget('name_btn1').set_active(True)
self.get_widget('type_btn1').set_active(True)
self.get_widget('gramps_btn1').set_active(True)
else:
self.get_widget('name_btn2').set_active(True)
self.get_widget('type_btn2').set_active(True)
self.get_widget('gramps_btn2').set_active(True)
def cb_help(self, obj):
"""Display the relevant portion of the Gramps manual"""
GrampsDisplay.help(webpage = WIKI_HELP_PAGE, section = WIKI_HELP_SEC)
def cb_merge(self, obj):
"""
Perform the merge of the repositories when the merge button is clicked.
"""
self.uistate.set_busy_cursor(True)
use_handle1 = self.get_widget("handle_btn1").get_active()
if use_handle1:
phoenix = self.rp1
titanic = self.rp2
unselect_path = (1,)
else:
phoenix = self.rp2
titanic = self.rp1
unselect_path = (0,)
if self.get_widget("name_btn1").get_active() ^ use_handle1:
phoenix.set_name(titanic.get_name())
if self.get_widget("type_btn1").get_active() ^ use_handle1:
phoenix.set_type(titanic.get_type())
if self.get_widget("gramps_btn1").get_active() ^ use_handle1:
phoenix.set_gramps_id(titanic.get_gramps_id())
query = MergeRepoQuery(self.dbstate, phoenix, titanic)
query.execute()
self.uistate.viewmanager.active_page.selection.unselect_path(
unselect_path)
self.uistate.set_busy_cursor(False)
self.close()
class MergeRepoQuery(object):
"""
Create database query to merge two repositories.
"""
def __init__(self, dbstate, phoenix, titanic):
self.database = dbstate.db
self.phoenix = phoenix
self.titanic = titanic
def execute(self):
"""
Merges two repositories into a single repository.
"""
new_handle = self.phoenix.get_handle()
old_handle = self.titanic.get_handle()
self.phoenix.merge(self.titanic)
trans = self.database.transaction_begin()
for source in self.database.iter_sources():
if source.has_repo_reference(old_handle):
source.replace_repo_references(old_handle, new_handle)
self.database.commit_source(source, trans)
self.database.remove_repository(old_handle, trans)
self.database.commit_repository(self.phoenix, trans)
self.database.transaction_commit(trans, _("Merge Repositories"))

238
src/Merge/mergesource.py Normal file
View File

@@ -0,0 +1,238 @@
#
# Gramps - a GTK+/GNOME based genealogy program
#
# Copyright (C) 2000-2005 Donald N. Allingham
# Copyright (C) 2010 Michiel D. Nauta
#
# 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: _MergeSource.py 14135 2010-01-25 17:45:21Z gbritton $
"""
Provide merge capabilities for sources.
"""
#-------------------------------------------------------------------------
#
# Gramps modules
#
#-------------------------------------------------------------------------
from gen.ggettext import sgettext as _
import const
import GrampsDisplay
import ManagedWindow
#-------------------------------------------------------------------------
#
# Gramps constants
#
#-------------------------------------------------------------------------
WIKI_HELP_PAGE = '%s_-_Entering_and_Editing_Data:_Detailed_-_part_3' % \
const.URL_MANUAL_PAGE
WIKI_HELP_SEC = _('manual|Merge_Sources')
_GLADE_FILE = 'mergesource.glade'
#-------------------------------------------------------------------------
#
# Merge Sources
#
#-------------------------------------------------------------------------
class MergeSources(ManagedWindow.ManagedWindow):
"""
Displays a dialog box that allows the sources to be combined into one.
"""
def __init__(self, dbstate, uistate, handle1, handle2):
ManagedWindow.ManagedWindow.__init__(self, uistate, [], self.__class__)
self.dbstate = dbstate
database = dbstate.db
self.src1 = database.get_source_from_handle(handle1)
self.src2 = database.get_source_from_handle(handle2)
self.define_glade('mergesource', _GLADE_FILE)
self.set_window(self._gladeobj.toplevel,
self.get_widget('source_title'),
_("Merge Sources"))
# Detailed Selection widgets
title1 = self.src1.get_title()
title2 = self.src2.get_title()
entry1 = self.get_widget("title1")
entry2 = self.get_widget("title2")
entry1.set_text(title1)
entry2.set_text(title2)
if entry1.get_text() == entry2.get_text():
for widget_name in ('title1', 'title2', 'title_btn1', 'title_btn2'):
self.get_widget(widget_name).set_sensitive(False)
entry1 = self.get_widget("author1")
entry2 = self.get_widget("author2")
entry1.set_text(self.src1.get_author())
entry2.set_text(self.src2.get_author())
if entry1.get_text() == entry2.get_text():
for widget_name in ('author1', 'author2', 'author_btn1',
'author_btn2'):
self.get_widget(widget_name).set_sensitive(False)
entry1 = self.get_widget("abbrev1")
entry2 = self.get_widget("abbrev2")
entry1.set_text(self.src1.get_abbreviation())
entry2.set_text(self.src2.get_abbreviation())
if entry1.get_text() == entry2.get_text():
for widget_name in ('abbrev1', 'abbrev2', 'abbrev_btn1',
'abbrev_btn2'):
self.get_widget(widget_name).set_sensitive(False)
entry1 = self.get_widget("pub1")
entry2 = self.get_widget("pub2")
entry1.set_text(self.src1.get_publication_info())
entry2.set_text(self.src2.get_publication_info())
if entry1.get_text() == entry2.get_text():
for widget_name in ('pub1', 'pub2', 'pub_btn1', 'pub_btn2'):
self.get_widget(widget_name).set_sensitive(False)
gramps1 = self.src1.get_gramps_id()
gramps2 = self.src2.get_gramps_id()
entry1 = self.get_widget("gramps1")
entry2 = self.get_widget("gramps2")
entry1.set_text(gramps1)
entry2.set_text(gramps2)
if entry1.get_text() == entry2.get_text():
for widget_name in ('gramps1', 'gramps2', 'gramps_btn1',
'gramps_btn2'):
self.get_widget(widget_name).set_sensitive(False)
# Main window widgets that determine which handle survives
rbutton1 = self.get_widget("handle_btn1")
rbutton_label1 = self.get_widget("label_handle_btn1")
rbutton_label2 = self.get_widget("label_handle_btn2")
rbutton_label1.set_label(title1 + " [" + gramps1 + "]")
rbutton_label2.set_label(title2 + " [" + gramps2 + "]")
rbutton1.connect("toggled", self.on_handle1_toggled)
self.connect_button('source_help', self.cb_help)
self.connect_button('source_ok', self.cb_merge)
self.connect_button('source_cancel', self.close)
self.show()
def on_handle1_toggled(self, obj):
"""first chosen source changes"""
if obj.get_active():
self.get_widget("title_btn1").set_active(True)
self.get_widget("author_btn1").set_active(True)
self.get_widget("abbrev_btn1").set_active(True)
self.get_widget("pub_btn1").set_active(True)
self.get_widget("gramps_btn1").set_active(True)
else:
self.get_widget("title_btn2").set_active(True)
self.get_widget("author_btn2").set_active(True)
self.get_widget("abbrev_btn2").set_active(True)
self.get_widget("pub_btn2").set_active(True)
self.get_widget("gramps_btn2").set_active(True)
def cb_help(self, obj):
"""Display the relevant portion of Gramps manual"""
GrampsDisplay.help(webpage = WIKI_HELP_PAGE, section = WIKI_HELP_SEC)
def cb_merge(self, obj):
"""
Performs the merge of the sources when the merge button is clicked.
"""
self.uistate.set_busy_cursor(True)
use_handle1 = self.get_widget("handle_btn1").get_active()
if use_handle1:
phoenix = self.src1
titanic = self.src2
unselect_path = (1,)
else:
phoenix = self.src2
titanic = self.src1
unselect_path = (0,)
if self.get_widget("title_btn1").get_active() ^ use_handle1:
phoenix.set_title(titanic.get_title())
if self.get_widget("author_btn1").get_active() ^ use_handle1:
phoenix.set_author(titanic.get_author())
if self.get_widget("abbrev_btn1").get_active() ^ use_handle1:
phoenix.set_abbreviation(titanic.get_abbreviation())
if self.get_widget("pub_btn1").get_active() ^ use_handle1:
phoenix.set_publication_info(titanic.get_publication_info())
if self.get_widget("gramps_btn1").get_active() ^ use_handle1:
phoenix.set_gramps_id(titanic.get_gramps_id())
query = MergeSourceQuery(self.dbstate, phoenix, titanic)
query.execute()
self.uistate.viewmanager.active_page.selection.unselect_path(
unselect_path)
self.uistate.set_busy_cursor(False)
self.close()
class MergeSourceQuery(object):
"""
Create database query to merge two sources.
"""
def __init__(self, dbstate, phoenix, titanic):
self.database = dbstate.db
self.phoenix = phoenix
self.titanic = titanic
def execute(self):
"""
Merges to sources into a single source.
"""
new_handle = self.phoenix.get_handle()
old_handle = self.titanic.get_handle()
self.phoenix.merge(self.titanic)
trans = self.database.transaction_begin()
for person in self.database.iter_people():
if person.has_source_reference(old_handle):
person.replace_source_references(old_handle, new_handle)
self.database.commit_person(person, trans)
for family in self.database.iter_families():
if family.has_source_reference(old_handle):
family.replace_source_references(old_handle, new_handle)
self.database.commit_family(family, trans)
for event in self.database.iter_events():
if event.has_source_reference(old_handle):
event.replace_source_references(old_handle, new_handle)
self.database.commit_event(event, trans)
for source in self.database.iter_sources():
if source.has_source_reference(old_handle):
source.replace_source_references(old_handle, new_handle)
self.database.commit_source(source, trans)
for place in self.database.iter_places():
if place.has_source_reference(old_handle):
place.replace_source_references(old_handle, new_handle)
self.database.commit_place(place, trans)
for obj in self.database.iter_media_objects():
if obj.has_source_reference(old_handle):
obj.replace_source_references(old_handle, new_handle)
self.database.commit_media_object(obj, trans)
for repo in self.database.iter_repositories():
if repo.has_source_reference(old_handle):
repo.replace_source_references(old_handle, new_handle)
self.database.commit_repository(repo, trans)
self.database.remove_source(old_handle, trans)
self.database.commit_source(self.phoenix, trans)
self.database.transaction_commit(trans, _("Merge Sources"))