gramps/src/GrampsDb/_DbUtils.py

245 lines
8.2 KiB
Python
Raw Normal View History

2006-03-18 07:00:23 +05:30
#
# Gramps - a GTK+/GNOME based genealogy program
#
# Copyright (C) 2004-2006 Donald N. Allingham
2006-03-18 07:00:23 +05:30
#
# 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: __init__.py 6086 2006-03-06 03:54:58Z dallingham $
import RelLib
from BasicUtils import UpdateCallback
2006-03-18 07:00:23 +05:30
def remove_family_relationships(db, family_handle, trans=None):
2006-03-18 08:51:03 +05:30
family = db.get_family_from_handle(family_handle)
2006-03-18 07:00:23 +05:30
if trans == None:
need_commit = True
trans = db.transaction_begin()
else:
need_commit = False
for phandle in [ family.get_father_handle(),
family.get_mother_handle()]:
if phandle:
person = db.get_person_from_handle(phandle)
2006-03-18 08:51:03 +05:30
person.remove_family_handle(family_handle)
db.commit_person(person, trans)
2006-03-18 07:00:23 +05:30
for ref in family.get_child_ref_list():
phandle = ref.ref
2006-03-18 07:00:23 +05:30
person = db.get_person_from_handle(phandle)
2006-03-18 08:51:03 +05:30
person.remove_parent_family_handle(family_handle)
db.commit_person(person, trans)
2006-03-18 07:00:23 +05:30
2006-03-18 08:51:03 +05:30
db.remove_family(family_handle, trans)
2006-03-18 07:00:23 +05:30
if need_commit:
2006-03-18 08:51:03 +05:30
db.transaction_commit(trans, _("Remove Family"))
2006-03-18 07:00:23 +05:30
def remove_parent_from_family(db, person_handle, family_handle, trans=None):
"""
Removes a person as either the father or mother of a family,
deleting the family if it becomes empty.
"""
person = db.get_person_from_handle(person_handle)
family = db.get_family_from_handle(family_handle)
if trans == None:
need_commit = True
trans = db.transaction_begin()
else:
need_commit = False
person.remove_family_handle(family_handle)
if family.get_father_handle() == person_handle:
family.set_father_handle(None)
elif family.get_mother_handle() == person_handle:
family.set_mother_handle(None)
2006-04-20 09:20:57 +05:30
child_list = family.get_child_ref_list()
2006-03-18 07:00:23 +05:30
if (not family.get_father_handle() and not family.get_mother_handle() and
len(child_list) <= 1):
2006-03-18 07:00:23 +05:30
db.remove_family(family_handle, trans)
if child_list:
2006-04-20 09:20:57 +05:30
child = db.get_person_from_handle(child_list[0].ref)
child.remove_parent_family_handle(family_handle)
db.commit_person(child, trans)
2006-03-18 07:00:23 +05:30
msg = _("Remove father from family")
else:
db.commit_family(family, trans)
msg = _("Remove mother from family")
db.commit_person(person, trans)
if need_commit:
db.transaction_commit(trans,msg)
def remove_child_from_family(db, person_handle, family_handle, trans=None):
"""
Removes a person as a child of the family, deleting the family if
it becomes empty.
"""
person = db.get_person_from_handle(person_handle)
family = db.get_family_from_handle(family_handle)
person.remove_parent_family_handle(family_handle)
family.remove_child_handle(person_handle)
if trans == None:
need_commit = True
trans = db.transaction_begin()
else:
need_commit = False
2006-04-20 09:20:57 +05:30
child_list = family.get_child_ref_list()
2006-03-18 07:00:23 +05:30
if (not family.get_father_handle() and not family.get_mother_handle() and
2006-04-12 08:40:28 +05:30
len(child_list) <= 1):
2006-03-18 07:00:23 +05:30
db.remove_family(family_handle, trans)
2006-04-12 08:40:28 +05:30
if child_list:
2006-04-20 09:20:57 +05:30
child = db.get_person_from_handle(child_list[0].ref)
2006-04-12 08:40:28 +05:30
child.remove_parent_family_handle(family_handle)
db.commit_person(child, trans)
2006-03-18 07:00:23 +05:30
else:
db.commit_family(family, trans)
db.commit_person(person, trans)
if need_commit:
db.transaction_commit(trans,_("Remove child from family"))
def add_child_to_family(db, family, child,
2006-04-14 10:06:25 +05:30
mrel=RelLib.ChildRefType(),
frel=RelLib.ChildRefType(),
trans=None):
cref = RelLib.ChildRef()
cref.ref = child.handle
cref.set_father_relation(frel)
cref.set_mother_relation(mrel)
family.add_child_ref(cref)
child.add_parent_family_handle(family.handle)
if trans == None:
need_commit = True
trans = db.transaction_begin()
else:
need_commit = False
db.commit_family(family,trans)
db.commit_person(child,trans)
if need_commit:
db.transaction_commit(trans, _('Add child to family') )
def get_total(db):
person_len = db.get_number_of_people()
family_len = db.get_number_of_families()
event_len = db.get_number_of_events()
source_len = db.get_number_of_sources()
place_len = db.get_number_of_places()
repo_len = db.get_number_of_repositories()
obj_len = db.get_number_of_media_objects()
return person_len + family_len + event_len + \
place_len + source_len + obj_len + repo_len
def db_copy(from_db,to_db,callback):
uc = UpdateCallback(callback)
uc.set_total(get_total(from_db))
tables = {
'Person': {'cursor_func': from_db.get_person_cursor,
'table': to_db.person_map,
'sec_table' : to_db.id_trans },
'Family': {'cursor_func': from_db.get_family_cursor,
'table': to_db.family_map,
'sec_table' : to_db.fid_trans },
'Event': {'cursor_func': from_db.get_event_cursor,
'table': to_db.event_map,
'sec_table' : to_db.eid_trans },
'Place': {'cursor_func': from_db.get_place_cursor,
'table': to_db.place_map,
'sec_table' : to_db.pid_trans },
'Source': {'cursor_func': from_db.get_source_cursor,
'table': to_db.source_map,
'sec_table' : to_db.sid_trans },
'MediaObject': {'cursor_func': from_db.get_media_cursor,
'table': to_db.media_map,
'sec_table' : to_db.oid_trans },
'Repository': {'cursor_func': from_db.get_repository_cursor,
'table': to_db.repository_map,
'sec_table' : to_db.rid_trans },
}
if to_db.__class__.__name__ == 'GrampsBSDDB':
if to_db.UseTXN:
add_data = add_data_txn
else:
add_data = add_data_notxn
update_secondary = update_secondary_empty
else:
add_data = add_data_dict
# For InMem databases, the secondary indices need to be
# created as we copy objects
update_secondary = update_secondary_inmem
# Start batch transaction to use async TXN and other tricks
trans = to_db.transaction_begin("",batch=True)
for table_name in tables.keys():
cursor_func = tables[table_name]['cursor_func']
table = tables[table_name]['table']
sec_table = tables[table_name]['sec_table']
cursor = cursor_func()
item = cursor.first()
while item:
(handle,data) = item
add_data(to_db,table,handle,data)
update_secondary(sec_table,handle,data)
item = cursor.next()
uc.update()
cursor.close()
# Commit batch transaction: does nothing, except undoing the tricks
to_db.transaction_commit(trans,"")
# The metadata is always transactionless,
# and the table is small, so using key iteration is OK here.
for handle in from_db.metadata.keys():
data = from_db.metadata.get(handle)
to_db.metadata[handle] = data
def add_data_txn(db,table,handle,data):
the_txn = db.env.txn_begin()
table.put(handle,data,txn=the_txn)
the_txn.commit()
def add_data_notxn(db,table,handle,data):
table.put(handle,data)
def add_data_dict(db,table,handle,data):
table[handle] = data
def update_secondary_empty(sec_table,handle,data):
pass
def update_secondary_inmem(sec_table,handle,data):
sec_table[str(data[1])] = str(handle)