Tweaks and simplifications
svn: r14001
This commit is contained in:
parent
16808b6c0e
commit
afe85ad0d5
@ -384,32 +384,27 @@ class PrivateProxyDb(ProxyDbBase):
|
||||
# private (like a SourceRef or MediaRef). It only checks if the
|
||||
# referenced object is private.
|
||||
|
||||
objects = {
|
||||
'Person' : self.db.get_person_from_handle,
|
||||
'Family' : self.db.get_family_from_handle,
|
||||
'Event' : self.db.get_event_from_handle,
|
||||
'Source' : self.db.get_source_from_handle,
|
||||
'Place' : self.db.get_place_from_handle,
|
||||
'MediaObject' : self.db.get_object_from_handle,
|
||||
'Note' : self.db.get_note_from_handle,
|
||||
'Repository' : self.db.get_repository_from_handle,
|
||||
}
|
||||
|
||||
handle_itr = self.db.find_backlink_handles(handle, include_classes)
|
||||
for (class_name, handle) in handle_itr:
|
||||
if class_name == 'Person':
|
||||
obj = self.db.get_person_from_handle(handle)
|
||||
elif class_name == 'Family':
|
||||
obj = self.db.get_family_from_handle(handle)
|
||||
elif class_name == 'Event':
|
||||
obj = self.db.get_event_from_handle(handle)
|
||||
elif class_name == 'Source':
|
||||
obj = self.db.get_source_from_handle(handle)
|
||||
elif class_name == 'Place':
|
||||
obj = self.db.get_place_from_handle(handle)
|
||||
elif class_name == 'MediaObject':
|
||||
obj = self.db.get_object_from_handle(handle)
|
||||
elif class_name == 'Note':
|
||||
obj = self.db.get_note_from_handle(handle)
|
||||
elif class_name == 'Repository':
|
||||
obj = self.db.get_repository_from_handle(handle)
|
||||
if class_name in objects:
|
||||
obj = objects[class_name](handle)
|
||||
if obj and not obj.get_privacy():
|
||||
yield (class_name, handle)
|
||||
else:
|
||||
raise NotImplementedError
|
||||
|
||||
if obj and not obj.get_privacy():
|
||||
yield (class_name, handle)
|
||||
raise NotImplementedError
|
||||
return
|
||||
|
||||
|
||||
def copy_media_ref_list(db, original_obj, clean_obj):
|
||||
"""
|
||||
Copies media references from one object to another - excluding private
|
||||
|
@ -145,29 +145,24 @@ class ReferencedProxyDb(ProxyDbBase):
|
||||
effectively become unreferenced, so the action is performed in a loop
|
||||
until there are no more objects to unreference.
|
||||
"""
|
||||
object_types = {
|
||||
'Event': {'unref_list': self.unreferenced_events,
|
||||
'handle_list': self.get_event_handles},
|
||||
'Place': {'unref_list': self.unreferenced_places,
|
||||
'handle_list': self.get_place_handles},
|
||||
'Source': {'unref_list': self.unreferenced_sources,
|
||||
'handle_list': self.get_source_handles},
|
||||
'Repositories': {'unref_list': self.unreferenced_repositories,
|
||||
'handle_list': self.get_repository_handles},
|
||||
'MediaObjects': {'unref_list': self.unreferenced_media_objects,
|
||||
'handle_list': self.get_media_object_handles},
|
||||
'Notes': {'unref_list': self.unreferenced_notes,
|
||||
'handle_list': self.get_note_handles}
|
||||
}
|
||||
|
||||
unrefs = (
|
||||
(self.unreferenced_events, self.get_event_handles),
|
||||
(self.unreferenced_places, self.get_place_handles),
|
||||
(self.unreferenced_sources, self.get_source_handles),
|
||||
(self.unreferenced_repositories,
|
||||
self.get_repository_handles),
|
||||
(self.unreferenced_media_objects,
|
||||
self.get_media_object_handles),
|
||||
(self.unreferenced_notes, self.get_note_handles),
|
||||
)
|
||||
|
||||
last_count = 0
|
||||
while True:
|
||||
current_count = 0
|
||||
for object_type, object_dict in object_types.iteritems():
|
||||
unref_list = object_dict['unref_list']
|
||||
handle_list = object_dict['handle_list']()
|
||||
for (unref_list, handle_list) in unrefs:
|
||||
unref_list.update(
|
||||
handle for handle in handle_list
|
||||
handle for handle in handle_list()
|
||||
if not any(self.find_backlink_handles(handle)))
|
||||
current_count += len(unref_list)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user