Cyclone-Team/gramps
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

* src/GrampsDb/_ReadGrdb.py: Add repos, modularize.

Browse Source 
svn: r6704
This commit is contained in:
Alex Roitman 2006-05-18 04:16:54 +00:00
parent ed6e277141
commit 8215cf553d
2 changed files with 105 additions and 98 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ChangeLog
View File

@ -1,4 +1,5 @@
2006-05-17 Alex Roitman <shura@gramps-project.org> 2006-05-17 Alex Roitman <shura@gramps-project.org>
* src/GrampsDb/_ReadGrdb.py: Add repos, modularize.
* src/ViewManager.py: Reorder calls to make work with ArgHandler. * src/ViewManager.py: Reorder calls to make work with ArgHandler.
2006-05-17 Don Allingham <don@gramps-project.org> 2006-05-17 Don Allingham <don@gramps-project.org>

194
src/GrampsDb/_ReadGrdb.py
View File

@ -59,6 +59,7 @@ def importData(database, filename, callback=None,cl=0,use_trans=True):
else: else:
ErrorDialog(_("%s could not be opened") % filename) ErrorDialog(_("%s could not be opened") % filename)
return return
if not other_database.version_supported(): if not other_database.version_supported():
if cl: if cl:
print "Error: %s could not be opened.\n%s Exiting." \ print "Error: %s could not be opened.\n%s Exiting." \
@ -74,115 +75,98 @@ def importData(database, filename, callback=None,cl=0,use_trans=True):
"by this version of GRAMPS.")) "by this version of GRAMPS."))
return return
# Check for duplicate handles. At the moment we simply exit here,
# before modifying any data. In the future we will need to handle
# this better.
handles = sets.Set(database.person_map.keys())
other_handles = sets.Set(other_database.person_map.keys())
if handles.intersection(other_handles):
raise Errors.HandleError('Personal handles in two databases overlap.')
handles = sets.Set(database.family_map.keys()) # Prepare table and method definitions
other_handles = sets.Set(other_database.family_map.keys()) tables = {
if handles.intersection(other_handles): 'Person' : {'table' : database.person_map,
raise Errors.HandleError('Family handles in two databases overlap.') 'id_table' : database.id_trans,
'add_obj' : database.add_person,
'find_next_gramps_id' :database.find_next_person_gramps_id,
'other_get_from_handle':
other_database.get_person_from_handle,
'other_table': other_database.person_map,
},
'Family' : {'table' : database.family_map,
'id_table' : database.fid_trans,
'add_obj' : database.add_family,
'find_next_gramps_id' :database.find_next_family_gramps_id,
'other_get_from_handle':
other_database.get_family_from_handle,
'other_table': other_database.family_map,
},
handles = sets.Set(database.place_map.keys()) 'Event' : {'table' : database.event_map,
other_handles = sets.Set(other_database.place_map.keys()) 'id_table' : database.eid_trans,
if handles.intersection(other_handles): 'add_obj' : database.add_event,
raise Errors.HandleError('Place handles in two databases overlap.') 'find_next_gramps_id' : database.find_next_event_gramps_id,
'other_get_from_handle':
other_database.get_event_from_handle,
'other_table': other_database.event_map,
},
'Source' : {'table' : database.family_map,
'id_table' : database.sid_trans,
'add_obj' : database.add_family,
'find_next_gramps_id': database.find_next_family_gramps_id,
'other_get_from_handle':
other_database.get_family_from_handle,
'other_table': other_database.source_map,
},
'Place' : {'table' : database.source_map,
'id_table' : database.pid_trans,
'add_obj' : database.add_source,
'find_next_gramps_id' :database.find_next_source_gramps_id,
'other_get_from_handle':
other_database.get_source_from_handle,
'other_table': other_database.place_map,
},
'Media' : {'table' : database.media_map,
'id_table' : database.oid_trans,
'add_obj' : database.add_object,
'find_next_gramps_id' : database.find_next_object_gramps_id,
'other_get_from_handle':
other_database.get_object_from_handle,
'other_table': other_database.media_map,
},
'Repository' : {'table' : database.repository_map,
'id_table' : database.rid_trans,
'add_obj' : database.add_repository,
'find_next_gramps_id' :
database.find_next_repository_gramps_id,
'other_get_from_handle':
other_database.get_repository_from_handle,
'other_table': other_database.repository_map,
},
}
handles = sets.Set(database.source_map.keys())
other_handles = sets.Set(other_database.source_map.keys())
if handles.intersection(other_handles):
raise Errors.HandleError('Source handles in two databases overlap.')
handles = sets.Set(database.media_map.keys()) # Check for duplicate handles.
other_handles = sets.Set(other_database.media_map.keys()) for key in tables:
if handles.intersection(other_handles): table_dict = tables[key]
raise Errors.HandleError('Media handles in two databases overlap.') table = table_dict['table']
other_table = table_dict['other_table']
handles = sets.Set(database.event_map.keys()) msg = '%s handles in two databases overlap.' % key
other_handles = sets.Set(other_database.event_map.keys()) check_common_handles(table,other_table,msg)
if handles.intersection(other_handles):
raise Errors.HandleError('Event handles in two databases overlap.')
# Proceed with preparing for import # Proceed with preparing for import
if use_trans: if use_trans:
trans = database.transaction_begin("",batch=True) trans = database.transaction_begin("",batch=True)
else: else:
print "Transaction is None! This is no way to go!"
trans = None trans = None
database.disable_signals() database.disable_signals()
# copy all data from new_database to database, # copy all data from new_database to database,
# rename gramps IDs of first-class objects when conflicts are found # rename gramps IDs of first-class objects when conflicts are found
for key in tables:
# People table table_dict = tables[key]
for person_handle in other_database.person_map.keys(): id_table = table_dict['id_table']
person = other_database.get_person_from_handle(person_handle) add_obj = table_dict['add_obj']
find_next_gramps_id = table_dict['find_next_gramps_id']
# Then we check gramps_id for conflicts and change it if needed other_table = table_dict['other_table']
gramps_id = str(person.get_gramps_id()) other_get_from_handle = table_dict['other_get_from_handle']
if database.id_trans.has_key(gramps_id): import_table(id_table,add_obj,find_next_gramps_id,
gramps_id = database.find_next_person_gramps_id() other_table,other_get_from_handle,trans)
person.set_gramps_id(gramps_id)
database.add_person(person,trans)
# Family table
for family_handle in other_database.family_map.keys():
family = other_database.get_family_from_handle(family_handle)
# Then we check gramps_id for conflicts and change it if needed
gramps_id = str(family.get_gramps_id())
if database.fid_trans.has_key(gramps_id):
gramps_id = database.find_next_family_gramps_id()
family.set_gramps_id(gramps_id)
database.add_family(family,trans)
# Place table
for place_handle in other_database.place_map.keys():
place = other_database.get_place_from_handle(place_handle)
# Then we check gramps_id for conflicts and change it if needed
gramps_id = str(place.get_gramps_id())
if database.pid_trans.has_key(gramps_id):
gramps_id = database.find_next_place_gramps_id()
place.set_gramps_id(gramps_id)
database.add_place(place,trans)
# Source table
for source_handle in other_database.source_map.keys():
source = other_database.get_source_from_handle(source_handle)
# Then we check gramps_id for conflicts and change it if needed
gramps_id = str(source.get_gramps_id())
if database.sid_trans.has_key(gramps_id):
gramps_id = database.find_next_source_gramps_id()
source.set_gramps_id(gramps_id)
database.add_source(source,trans)
# Media table
for media_handle in other_database.media_map.keys():
media = other_database.get_object_from_handle(media_handle)
# Then we check gramps_id for conflicts and change it if needed
gramps_id = str(media.get_gramps_id())
if database.oid_trans.has_key(gramps_id):
gramps_id = database.find_next_object_gramps_id()
media.set_gramps_id(gramps_id)
database.add_object(media,trans)
# Event table
for event_handle in other_database.event_map.keys():
event = other_database.get_event_from_handle(event_handle)
# Then we check gramps_id for conflicts and change it if needed
gramps_id = str(event.get_gramps_id())
if database.eid_trans.has_key(gramps_id):
gramps_id = database.find_next_event_gramps_id()
event.set_gramps_id(gramps_id)
database.add_event(event,trans)
# close the other database and clean things up # close the other database and clean things up
other_database.close() other_database.close()
@ -190,3 +174,25 @@ def importData(database, filename, callback=None,cl=0,use_trans=True):
database.transaction_commit(trans,_("Import database")) database.transaction_commit(trans,_("Import database"))
database.enable_signals() database.enable_signals()
database.request_rebuild() database.request_rebuild()
def check_common_handles(table,other_table,msg):
# Check for duplicate handles. At the moment we simply exit here,
# before modifying any data. In the future we will need to handle
# this better. How?
handles = sets.Set(table.keys())
other_handles = sets.Set(other_table.keys())
if handles.intersection(other_handles):
raise Errors.HandleError(msg)
def import_table(id_table,add_obj,find_next_gramps_id,
other_table,other_get_from_handle,trans):
for handle in other_table.keys():
obj = other_get_from_handle(handle)
# Then we check gramps_id for conflicts and change it if needed
gramps_id = str(obj.gramps_id)
if id_table.has_key(gramps_id):
gramps_id = find_next_gramps_id()
obj.gramps_id = gramps_id
add_obj(obj,trans)