# # 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$ "Database Processing/Rename personal event types" #------------------------------------------------------------------------ # # standard python modules # #------------------------------------------------------------------------ import os from gettext import gettext as _ #------------------------------------------------------------------------ # # GNOME/GTK modules # #------------------------------------------------------------------------ import gtk import gtk.glade #------------------------------------------------------------------------ # # GRAMPS modules # #------------------------------------------------------------------------ import const import Utils import ManagedWindow import AutoComp from RelLib import EventType from QuestionDialog import OkDialog from PluginUtils import Tool, register_tool #------------------------------------------------------------------------- # # # #------------------------------------------------------------------------- class ChangeTypes(Tool.BatchTool, ManagedWindow.ManagedWindow): def __init__(self, dbstate, uistate, options_class, name, callback=None): Tool.BatchTool.__init__(self, dbstate, options_class, name) if self.fail: return if uistate: self.title = _('Change Event Types') ManagedWindow.ManagedWindow.__init__(self,uistate,[], self.__class__) self.init_gui() else: self.run_tool(cli=True) def init_gui(self): # Draw dialog and make it handle everything base = os.path.dirname(__file__) glade_file = "%s/%s" % (base,"changetype.glade") self.glade = gtk.glade.XML(glade_file,"top","gramps") self.auto1 = self.glade.get_widget("original") self.auto2 = self.glade.get_widget("new") event_names = [item[1] for item in EventType._DATAMAP if item[0] > 0 ] AutoComp.fill_combo(self.auto1,event_names) AutoComp.fill_combo(self.auto2,event_names) # Need to display localized event names fromtype = self.options.handler.options_dict['fromtype'] totype = self.options.handler.options_dict['totype'] self.auto1.child.set_text(_(fromtype)) self.auto2.child.set_text(_(totype)) window = self.glade.get_widget('top') self.set_window(window,self.glade.get_widget('title'),self.title) self.glade.signal_autoconnect({ "on_close_clicked" : self.close, "on_apply_clicked" : self.on_apply_clicked, }) self.show() def build_menu_names(self,obj): return (self.title,None) def run_tool(self,cli=False): # Run tool and return results # These are English names, no conversion needed fromtype = self.options.handler.options_dict['fromtype'] totype = self.options.handler.options_dict['totype'] modified = 0 self.trans = self.db.transaction_begin("",batch=True) self.db.disable_signals() if not cli: progress = Utils.ProgressMeter(_('Analyzing events'),'') progress.set_pass('',self.db.get_number_of_people()) for person_handle in self.db.get_person_handles(sort_handles=False): person = self.db.get_person_from_handle(person_handle) for event_ref in person.get_event_ref_list(): if not event_ref.ref: continue event = self.db.get_event_from_handle(event_ref.ref) if str(event.get_type()) == fromtype: event.set_type(totype) modified = modified + 1 self.db.commit_event(event,self.trans) if not cli: progress.step() if not cli: progress.close() self.db.transaction_commit(self.trans,_('Change types')) self.db.enable_signals() self.db.request_rebuild() if modified == 0: msg = _("No event record was modified.") elif modified == 1: msg = _("1 event record was modified.") else: msg = _("%d event records were modified.") % modified if cli: print "Done: ", msg return (bool(modified),msg) def on_apply_clicked(self,obj): # Need to store English names for later comparison self.options.handler.options_dict['fromtype'] = EventType._I2EMAP[ EventType._S2IMAP[unicode(self.auto1.child.get_text())]] self.options.handler.options_dict['totype'] = EventType._I2EMAP[ EventType._S2IMAP[unicode(self.auto2.child.get_text())]] modified,msg = self.run_tool(cli=False) OkDialog(_('Change types'),msg,self.window) # Save options self.options.handler.save_options() self.close() #------------------------------------------------------------------------ # # # #------------------------------------------------------------------------ class ChangeTypesOptions(Tool.ToolOptions): """ Defines options and provides handling interface. """ def __init__(self,name,person_id=None): Tool.ToolOptions.__init__(self,name,person_id) def set_new_options(self): # Options specific for this report self.options_dict = { 'fromtype' : '', 'totype' : '', } self.options_help = { 'fromtype' : ("=str","Type of events to replace", "Event type string"), 'totype' : ("=str","New type replacing the old one", "Event type string"), } #------------------------------------------------------------------------ # # # #------------------------------------------------------------------------ register_tool( name = 'chtype', category = Tool.TOOL_DBPROC, tool_class = ChangeTypes, options_class = ChangeTypesOptions, modes = Tool.MODE_GUI | Tool.MODE_CLI, translated_name = _("Rename personal event types"), status = _("Stable"), author_name = "Donald N. Allingham", author_email = "don@gramps-project.org", description = _("Allows all the events of a certain name " "to be renamed to a new name.") )