6d596ad987
This is a major patch by Michael Nauta. It means all writes happen immediately to bsddb, and the bsddb rollback is used on a crash. Transaction is no longer used to store changes and do them on commit. Undo database is set on end. At the same time with statement is used throughout for transactions At the same time, the original bug in merge code should be fixed Still some issues, that will be ironed out svn: r16523
343 lines
15 KiB
Python
343 lines
15 KiB
Python
#
|
|
# 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.database = dbstate.db
|
|
self.fy1 = self.database.get_family_from_handle(handle1)
|
|
self.fy2 = self.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 = self.database.get_person_from_handle(father1)
|
|
father2 = self.database.get_person_from_handle(father2)
|
|
father_id1 = father1.get_gramps_id() if father1 else ""
|
|
father_id2 = father2.get_gramps_id() if father2 else ""
|
|
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))
|
|
deactivate = False
|
|
if father_id1 == "" and father_id2 == "":
|
|
deactivate = True
|
|
elif father_id2 == "":
|
|
self.get_widget("father_btn1").set_active(True)
|
|
deactivate = True
|
|
elif father_id1 == "":
|
|
self.get_widget("father_btn2").set_active(True)
|
|
deactivate = True
|
|
elif entry1.get_text() == entry2.get_text():
|
|
deactivate = True
|
|
if deactivate:
|
|
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 = self.database.get_person_from_handle(mother1)
|
|
mother2 = self.database.get_person_from_handle(mother2)
|
|
mother_id1 = mother1.get_gramps_id() if mother1 else ""
|
|
mother_id2 = mother2.get_gramps_id() if mother2 else ""
|
|
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))
|
|
deactivate = False
|
|
if mother_id1 == "" and mother_id2 == "":
|
|
deactivate = True
|
|
elif mother_id1 == "":
|
|
self.get_widget("mother_btn2").set_active(True)
|
|
deactivate = True
|
|
elif mother_id2 == "":
|
|
self.get_widget("mother_btn1").set_active(True)
|
|
deactivate = True
|
|
elif entry1.get_text() == entry2.get_text():
|
|
deactivate = True
|
|
if deactivate:
|
|
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)
|
|
|
|
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():
|
|
father1_text = self.get_widget("father1").get_text()
|
|
if father1_text != " []" or (father1_text == " []" and
|
|
self.get_widget("father2").get_text() == " []"):
|
|
self.get_widget("father_btn1").set_active(True)
|
|
mother1_text = self.get_widget("mother1").get_text()
|
|
if mother1_text != " []" or (mother1_text == " []" and
|
|
self.get_widget("mother2").get_text() == " []"):
|
|
self.get_widget("mother_btn1").set_active(True)
|
|
self.get_widget("rel_btn1").set_active(True)
|
|
self.get_widget("gramps_btn1").set_active(True)
|
|
else:
|
|
father2_text = self.get_widget("father2").get_text()
|
|
if father2_text != " []" or (father2_text == " []" and
|
|
self.get_widget("father1").get_text() == " []"):
|
|
self.get_widget("father_btn2").set_active(True)
|
|
mother2_text = self.get_widget("mother2").get_text()
|
|
if mother2_text != " []" or (mother2_text == " []" and
|
|
self.get_widget("mother1").get_text() == " []"):
|
|
self.get_widget("mother_btn2").set_active(True)
|
|
self.get_widget("rel_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)
|
|
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_fh = phoenix.get_father_handle()
|
|
phoenix_mh = phoenix.get_mother_handle()
|
|
|
|
if self.get_widget("father_btn1").get_active() ^ use_handle1:
|
|
phoenix_fh = titanic.get_father_handle()
|
|
if self.get_widget("mother_btn1").get_active() ^ use_handle1:
|
|
phoenix_mh = titanic.get_mother_handle()
|
|
if self.get_widget("rel_btn1").get_active() ^ use_handle1:
|
|
phoenix.set_relationship(titanic.get_relationship())
|
|
if self.get_widget("gramps_btn1").get_active() ^ use_handle1:
|
|
phoenix.set_gramps_id(titanic.get_gramps_id())
|
|
|
|
try:
|
|
query = MergeFamilyQuery(self.database, phoenix, titanic,
|
|
phoenix_fh, phoenix_mh)
|
|
query.execute()
|
|
except MergeError, err:
|
|
ErrorDialog( _("Cannot merge people"), str(err))
|
|
self.uistate.viewmanager.active_page.selection.unselect_path(
|
|
unselect_path)
|
|
self.uistate.set_busy_cursor(False)
|
|
self.close()
|
|
|
|
class MergeFamilyQuery(object):
|
|
"""
|
|
Create database query to merge two families.
|
|
"""
|
|
def __init__(self, database, phoenix, titanic, phoenix_fh, phoenix_mh):
|
|
self.database = database
|
|
self.phoenix = phoenix
|
|
self.titanic = titanic
|
|
self.phoenix_fh = phoenix_fh
|
|
if self.phoenix.get_father_handle() == self.phoenix_fh:
|
|
self.titanic_fh = self.titanic.get_father_handle()
|
|
self.father_swapped = False
|
|
else:
|
|
assert self.phoenix_fh == self.titanic.get_father_handle()
|
|
self.titanic_fh = self.phoenix.get_father_handle()
|
|
self.father_swapped = True
|
|
self.phoenix_mh = phoenix_mh
|
|
if self.phoenix.get_mother_handle() == self.phoenix_mh:
|
|
self.titanic_mh = self.titanic.get_mother_handle()
|
|
self.mother_swapped = False
|
|
else:
|
|
assert self.phoenix_mh == self.titanic.get_mother_handle()
|
|
self.titanic_mh = self.phoenix.get_mother_handle()
|
|
self.mother_swapped = True
|
|
|
|
def merge_person(self, phoenix_person, titanic_person, parent, trans):
|
|
"""
|
|
Merge two persons even if they are None; no families are merged!
|
|
"""
|
|
new_handle = self.phoenix.get_handle()
|
|
old_handle = self.titanic.get_handle()
|
|
|
|
if parent == 'father':
|
|
swapped = self.father_swapped
|
|
adjust_family_parent_handle = 'set_father_handle'
|
|
elif parent == 'mother':
|
|
swapped = self.mother_swapped
|
|
adjust_family_parent_handle = 'set_mother_handle'
|
|
else:
|
|
raise ValueError(_("A parent should be a father or mother."))
|
|
|
|
if phoenix_person is None:
|
|
if titanic_person is not None:
|
|
raise MergeError(_("""When merging people where one person """
|
|
"""doesn't exist, that "person" must be the person that """
|
|
"""will be deleted from the database."""))
|
|
return
|
|
elif titanic_person is None:
|
|
if swapped:
|
|
if [childref for childref in self.phoenix.get_child_ref_list()
|
|
if childref.get_reference_handle() ==
|
|
phoenix_person.get_handle()]:
|
|
raise MergeError(_("A parent and child cannot be merged. "
|
|
"To merge these people, you must first break the "
|
|
"relationship between them."))
|
|
phoenix_person.add_family_handle(new_handle)
|
|
getattr(self.phoenix, adjust_family_parent_handle)(
|
|
phoenix_person.get_handle())
|
|
self.database.commit_family(self.phoenix, trans)
|
|
else:
|
|
if [childref for childref in self.titanic.get_child_ref_list()
|
|
if childref.get_reference_handle() ==
|
|
phoenix_person.get_handle()]:
|
|
raise MergeError(_("A parent and child cannot be merged. "
|
|
"To merge these people, you must first break the "
|
|
"relationship between them."))
|
|
phoenix_person.add_family_handle(old_handle)
|
|
getattr(self.titanic, adjust_family_parent_handle)(
|
|
phoenix_person.get_handle())
|
|
self.database.commit_family(self.titanic, trans)
|
|
self.database.commit_person(phoenix_person, trans)
|
|
else:
|
|
query = MergePersonQuery(self.database, phoenix_person, titanic_person)
|
|
query.execute(family_merger=False, trans=trans)
|
|
|
|
def execute(self):
|
|
"""
|
|
Merges two families into a single family.
|
|
"""
|
|
new_handle = self.phoenix.get_handle()
|
|
old_handle = self.titanic.get_handle()
|
|
|
|
with self.database.transaction_begin(_('Merge Family')) as trans:
|
|
|
|
phoenix_father = self.database.get_person_from_handle(self.phoenix_fh)
|
|
titanic_father = self.database.get_person_from_handle(self.titanic_fh)
|
|
self.merge_person(phoenix_father, titanic_father, 'father', trans)
|
|
|
|
phoenix_mother = self.database.get_person_from_handle(self.phoenix_mh)
|
|
titanic_mother = self.database.get_person_from_handle(self.titanic_mh)
|
|
self.phoenix = self.database.get_family_from_handle(new_handle)
|
|
self.titanic = self.database.get_family_from_handle(old_handle)
|
|
self.merge_person(phoenix_mother, titanic_mother, 'mother', trans)
|
|
|
|
phoenix_father = self.database.get_person_from_handle(self.phoenix_fh)
|
|
phoenix_mother = self.database.get_person_from_handle(self.phoenix_mh)
|
|
self.phoenix = self.database.get_family_from_handle(new_handle)
|
|
self.titanic = self.database.get_family_from_handle(old_handle)
|
|
self.phoenix.merge(self.titanic)
|
|
for childref in self.titanic.get_child_ref_list():
|
|
child = self.database.get_person_from_handle(
|
|
childref.get_reference_handle())
|
|
if new_handle in child.parent_family_list:
|
|
child.remove_handle_references('Family', [old_handle])
|
|
else:
|
|
child.replace_handle_reference('Family', old_handle, new_handle)
|
|
self.database.commit_person(child, trans)
|
|
if phoenix_father:
|
|
phoenix_father.remove_family_handle(old_handle)
|
|
self.database.commit_person(phoenix_father, trans)
|
|
if phoenix_mother:
|
|
phoenix_mother.remove_family_handle(old_handle)
|
|
self.database.commit_person(phoenix_mother, trans)
|
|
self.database.remove_family(old_handle, trans)
|
|
self.database.commit_family(self.phoenix, trans)
|