make ref update optional
svn: r5780
This commit is contained in:
parent
a01b665403
commit
f7a15cf601
@ -1,3 +1,6 @@
|
|||||||
|
2006-01-17 Richard Taylor <rjt-gramps@thegrindstone.me.uk>
|
||||||
|
* src/GrampsDb/_GrampsBSDDB.py: make ref update optional
|
||||||
|
|
||||||
2006-01-17 Alex Roitman <shura@gramps-project.org>
|
2006-01-17 Alex Roitman <shura@gramps-project.org>
|
||||||
* src/GrampsDb/_GrampsDbBase.py (redo): Properly abort redo action
|
* src/GrampsDb/_GrampsDbBase.py (redo): Properly abort redo action
|
||||||
when no redo is available.
|
when no redo is available.
|
||||||
|
@ -586,59 +586,68 @@ class GrampsBSDDB(GrampsDbBase):
|
|||||||
|
|
||||||
primary_cur.close()
|
primary_cur.close()
|
||||||
|
|
||||||
def _update_reference_map(self, obj, transaction):
|
def _update_reference_map(self, obj, transaction, update=True):
|
||||||
|
"""
|
||||||
|
If update = True old references are cleaned up and only new references are added
|
||||||
|
if update = False then it is assumed that the object is not already referenced.
|
||||||
|
"""
|
||||||
|
|
||||||
# Add references to the reference_map for all primary object referenced
|
# Add references to the reference_map for all primary object referenced
|
||||||
# from the primary object 'obj' or any of its secondary objects.
|
# from the primary object 'obj' or any of its secondary objects.
|
||||||
|
|
||||||
# FIXME: this needs to be properly integrated into the transaction
|
if update:
|
||||||
# framework so that the reference_map changes are part of the
|
# FIXME: this needs to be properly integrated into the transaction
|
||||||
# transaction
|
# framework so that the reference_map changes are part of the
|
||||||
|
# transaction
|
||||||
|
|
||||||
handle = obj.get_handle()
|
handle = obj.get_handle()
|
||||||
|
|
||||||
# First thing to do is get hold of all rows in the reference_map
|
# First thing to do is get hold of all rows in the reference_map
|
||||||
# table that hold a reference from this primary obj. This means finding
|
# table that hold a reference from this primary obj. This means finding
|
||||||
# all the rows that have this handle somewhere in the list of (class_name,handle)
|
# all the rows that have this handle somewhere in the list of (class_name,handle)
|
||||||
# pairs.
|
# pairs.
|
||||||
# The primary_map secondary index allows us to look this up quickly.
|
# The primary_map secondary index allows us to look this up quickly.
|
||||||
|
|
||||||
existing_references = set()
|
existing_references = set()
|
||||||
|
|
||||||
primary_cur = self.get_reference_map_primary_cursor()
|
primary_cur = self.get_reference_map_primary_cursor()
|
||||||
|
|
||||||
try:
|
try:
|
||||||
ret = primary_cur.set(handle)
|
ret = primary_cur.set(handle)
|
||||||
except:
|
except:
|
||||||
ret = None
|
ret = None
|
||||||
|
|
||||||
while (ret is not None):
|
while (ret is not None):
|
||||||
(key,data) = ret
|
(key,data) = ret
|
||||||
|
|
||||||
# data values are of the form:
|
# data values are of the form:
|
||||||
# ((primary_object_class_name, primary_object_handle),
|
# ((primary_object_class_name, primary_object_handle),
|
||||||
# (referenced_object_class_name, referenced_object_handle))
|
# (referenced_object_class_name, referenced_object_handle))
|
||||||
# so we need the second tuple give us a reference that we can
|
# so we need the second tuple give us a reference that we can
|
||||||
# compare with what is returned from get_referenced_handles_recursively
|
# compare with what is returned from get_referenced_handles_recursively
|
||||||
|
|
||||||
# Looks like there is a bug in the set() and next_dup() methods
|
# Looks like there is a bug in the set() and next_dup() methods
|
||||||
# because they do not run the data through cPickle.loads before
|
# because they do not run the data through cPickle.loads before
|
||||||
# returning it, so we have to here.
|
# returning it, so we have to here.
|
||||||
existing_reference = cPickle.loads(data)[1]
|
existing_reference = cPickle.loads(data)[1]
|
||||||
existing_references.add((KEY_TO_CLASS_MAP[existing_reference[0]],
|
existing_references.add((KEY_TO_CLASS_MAP[existing_reference[0]],
|
||||||
existing_reference[1]))
|
existing_reference[1]))
|
||||||
ret = primary_cur.next_dup()
|
ret = primary_cur.next_dup()
|
||||||
|
|
||||||
primary_cur.close()
|
primary_cur.close()
|
||||||
|
|
||||||
# Once we have the list of rows that already have a reference we need to compare
|
# Once we have the list of rows that already have a reference we need to compare
|
||||||
# it with the list of objects that are still references from the primary object.
|
# it with the list of objects that are still references from the primary object.
|
||||||
|
|
||||||
current_references = set(obj.get_referenced_handles_recursively())
|
current_references = set(obj.get_referenced_handles_recursively())
|
||||||
|
|
||||||
no_longer_required_references = existing_references.difference(current_references)
|
no_longer_required_references = existing_references.difference(current_references)
|
||||||
|
|
||||||
|
|
||||||
new_references = current_references.difference(existing_references)
|
new_references = current_references.difference(existing_references)
|
||||||
|
|
||||||
|
else:
|
||||||
|
new_references = set(obj.get_referenced_handles_recursively())
|
||||||
|
|
||||||
# handle addition of new references
|
# handle addition of new references
|
||||||
|
|
||||||
@ -647,21 +656,17 @@ class GrampsBSDDB(GrampsDbBase):
|
|||||||
data = ((CLASS_TO_KEY_MAP[obj.__class__.__name__],handle),
|
data = ((CLASS_TO_KEY_MAP[obj.__class__.__name__],handle),
|
||||||
(CLASS_TO_KEY_MAP[ref_class_name],ref_handle),)
|
(CLASS_TO_KEY_MAP[ref_class_name],ref_handle),)
|
||||||
self._add_reference((handle,ref_handle),data,transaction)
|
self._add_reference((handle,ref_handle),data,transaction)
|
||||||
#self.reference_map.put(
|
|
||||||
# str((handle,ref_handle),),
|
|
||||||
# ((CLASS_TO_KEY_MAP[obj.__class__.__name__],handle),
|
|
||||||
# (CLASS_TO_KEY_MAP[ref_class_name],ref_handle),),
|
|
||||||
# txn=self.txn)
|
|
||||||
|
|
||||||
# handle deletion of old references
|
if update:
|
||||||
if len(no_longer_required_references) > 0:
|
# handle deletion of old references
|
||||||
for (ref_class_name,ref_handle) in no_longer_required_references:
|
if len(no_longer_required_references) > 0:
|
||||||
try:
|
for (ref_class_name,ref_handle) in no_longer_required_references:
|
||||||
self._remove_reference((handle,ref_handle),transaction)
|
try:
|
||||||
#self.reference_map.delete(str((handle,ref_handle),),
|
self._remove_reference((handle,ref_handle),transaction)
|
||||||
# txn=self.txn)
|
#self.reference_map.delete(str((handle,ref_handle),),
|
||||||
except: # ignore missing old reference
|
# txn=self.txn)
|
||||||
pass
|
except: # ignore missing old reference
|
||||||
|
pass
|
||||||
|
|
||||||
def _remove_reference(self,key,transaction):
|
def _remove_reference(self,key,transaction):
|
||||||
"""
|
"""
|
||||||
|
@ -7,7 +7,7 @@ import time
|
|||||||
import traceback
|
import traceback
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
sys.path.append('../src')
|
sys.path.append('../../src')
|
||||||
|
|
||||||
try:
|
try:
|
||||||
set()
|
set()
|
||||||
|
Loading…
Reference in New Issue
Block a user