d58641415d
* various: gen.lib integration. probably touches almost every file * various: gen.utils integration svn: r9101
214 lines
6.9 KiB
Python
214 lines
6.9 KiB
Python
#
|
|
# 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 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 locale
|
|
import ManagedWindow
|
|
import AutoComp
|
|
from gen.lib 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")
|
|
|
|
# Need to display localized event names
|
|
etype = EventType()
|
|
event_names = etype.get_standard_names()
|
|
event_names.sort(locale.strcoll)
|
|
|
|
AutoComp.fill_combo(self.auto1,event_names)
|
|
AutoComp.fill_combo(self.auto2,event_names)
|
|
|
|
etype.set_from_xml_str(self.options.handler.options_dict['fromtype'])
|
|
self.auto1.child.set_text(str(etype))
|
|
|
|
etype.set_from_xml_str(self.options.handler.options_dict['totype'])
|
|
self.auto2.child.set_text(str(etype))
|
|
|
|
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_events())
|
|
|
|
for event_handle in self.db.get_event_handles():
|
|
event = self.db.get_event_from_handle(event_handle)
|
|
if event.get_type().xml_str() == fromtype:
|
|
event.type.set_from_xml_str(totype)
|
|
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
|
|
the_type = EventType()
|
|
|
|
the_type.set(self.auto1.child.get_text())
|
|
self.options.handler.options_dict['fromtype'] = the_type.xml_str()
|
|
|
|
the_type.set(self.auto2.child.get_text())
|
|
self.options.handler.options_dict['totype'] = the_type.xml_str()
|
|
|
|
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 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.")
|
|
)
|