# # Gramps - a GTK+/GNOME based genealogy program # # Copyright (C) 2000-2006 Donald N. Allingham # # 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$ """ Change id IDs of all the elements in the database to conform to the scheme specified in the database's prefix ids """ #------------------------------------------------------------------------ # # standard python modules # #------------------------------------------------------------------------ import re from gettext import gettext as _ #------------------------------------------------------------------------ # # GRAMPS modules # #------------------------------------------------------------------------ import Utils import RelLib from PluginUtils import Tool, register_tool from QuestionDialog import WarningDialog _findint = re.compile('^[^\d]*(\d+)[^\d]*') #------------------------------------------------------------------------- # # Actual tool # #------------------------------------------------------------------------- class ReorderIds(Tool.Tool): def __init__(self,db,person,options_class,name,callback=None,parent=None): Tool.Tool.__init__(self,db,person,options_class,name) self.parent = parent if parent: self.progress = Utils.ProgressMeter(_('Reordering GRAMPS IDs'),'') else: print "Reordering GRAMPS IDs..." self.trans = db.transaction_begin() if parent: self.progress.set_pass(_('Reordering People IDs'), db.get_number_of_people()) self.reorder(RelLib.Person, db.get_person_from_gramps_id, db.get_person_from_handle, db.find_next_person_gramps_id, db.get_person_cursor, db.commit_person, db.iprefix) if parent: self.progress.set_pass(_('Reordering Family IDs'), db.get_number_of_families()) self.reorder(RelLib.Family, db.get_family_from_gramps_id, db.get_family_from_handle, db.find_next_family_gramps_id, db.get_family_cursor, db.commit_family, db.fprefix) if parent: self.progress.set_pass(_('Reordering Media Object IDs'), db.get_number_of_media_objects()) self.reorder(RelLib.MediaObject, db.get_object_from_gramps_id, db.get_object_from_handle, db.find_next_object_gramps_id, db.get_media_cursor, db.commit_media_object, db.oprefix) if parent: self.progress.set_pass(_('Reordering Source IDs'), db.get_number_of_sources()) self.reorder(RelLib.Source, db.get_source_from_gramps_id, db.get_source_from_handle, db.find_next_source_gramps_id, db.get_source_cursor, db.commit_source, db.sprefix) if parent: self.progress.set_pass(_('Reordering Place IDs'), db.get_number_of_places()) self.reorder(RelLib.Place, db.get_place_from_gramps_id, db.get_place_from_handle, db.find_next_place_gramps_id, db.get_place_cursor, db.commit_place, db.pprefix) if parent: self.progress.close() else: print "Done." db.transaction_commit(self.trans,_("Reorder GRAMPS IDs")) def reorder(self, class_type, find_from_id, find_from_handle, find_next_id, get_cursor, commit, prefix): dups = [] newids = {} key_list = [] # search all ids in the map cursor = get_cursor() data = cursor.first() while data: if self.parent: self.progress.step() (handle,sdata) = data obj = class_type() obj.unserialize(sdata) gramps_id = obj.get_gramps_id() # attempt to extract integer, if we can't, treat it as a # duplicate match = _findint.match(gramps_id) if match: # get the integer, build the new handle. Make sure it # hasn't already been chosen. If it has, put this # in the duplicate handle list try: index = match.groups()[0] newgramps_id = prefix % int(index) if newgramps_id == gramps_id: newids[newgramps_id] = gramps_id elif find_from_id(newgramps_id) != None: dups.append(obj.get_handle()) else: obj.set_gramps_id(newgramps_id) commit(obj,self.trans) newids[newgramps_id] = gramps_id except: dups.append(handle) else: dups.append(handle) data = cursor.next() cursor.close() # go through the duplicates, looking for the first availble # handle that matches the new scheme. if self.parent: self.progress.set_pass(_('Finding and assigning unused IDs'), len(dups)) for handle in dups: if self.parent: self.progress.step() obj = find_from_handle(handle) obj.set_gramps_id(find_next_id()) commit(obj,self.trans) #------------------------------------------------------------------------ # # # #------------------------------------------------------------------------ class ReorderIdsOptions(Tool.ToolOptions): """ Defines options and provides handling interface. """ def __init__(self,name,person_id=None): Tool.ToolOptions.__init__(self,name,person_id) #------------------------------------------------------------------------- # # # #------------------------------------------------------------------------- register_tool( name = 'reorder_ids', category = Tool.TOOL_DBPROC, tool_class = ReorderIds, options_class = ReorderIdsOptions, modes = Tool.MODE_GUI | Tool.MODE_CLI, translated_name = _("Reorder GRAMPS IDs"), status=(_("Stable")), author_name = "Donald N. Allingham", author_email = "don@gramps-project.org", description=_("Reorders the gramps IDs " "according to gramps' default rules.") )