gramps/src/plugins/ChangeTypes.py

225 lines
7.6 KiB
Python
Raw Normal View History

2002-10-20 19:55:16 +05:30
#
# Gramps - a GTK+/GNOME based genealogy program
#
# Copyright (C) 2000-2006 Donald N. Allingham
2002-10-20 19:55:16 +05:30
#
# 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$
2002-10-20 19:55:16 +05:30
"Database Processing/Rename personal event types"
#------------------------------------------------------------------------
#
# standard python modules
#
#------------------------------------------------------------------------
2002-10-20 19:55:16 +05:30
import os
from gettext import gettext as _
2002-10-20 19:55:16 +05:30
#------------------------------------------------------------------------
#
# GNOME/GTK modules
#
#------------------------------------------------------------------------
2002-10-20 19:55:16 +05:30
import gtk
import gtk.glade
#------------------------------------------------------------------------
#
# GRAMPS modules
#
#------------------------------------------------------------------------
2002-10-20 19:55:16 +05:30
import const
import Utils
2002-11-04 09:35:47 +05:30
from QuestionDialog import OkDialog
2003-03-17 10:51:40 +05:30
import AutoComp
from PluginUtils import Tool, register_tool
2002-10-20 19:55:16 +05:30
#-------------------------------------------------------------------------
#
#
#
#-------------------------------------------------------------------------
2005-12-06 12:08:09 +05:30
class ChangeTypes(Tool.Tool):
def __init__(self,db,person,options_class,name,callback=None,parent=None):
Tool.Tool.__init__(self,db,person,options_class,name)
2002-10-20 19:55:16 +05:30
2005-12-06 12:08:09 +05:30
if parent:
self.init_gui(parent)
else:
self.run_tool(cli=True)
def init_gui(self,parent):
# Draw dialog and make it handle everything
self.parent = parent
if self.parent.child_windows.has_key(self.__class__):
self.parent.child_windows[self.__class__].present(None)
return
self.win_key = self.__class__
2005-12-06 12:08:09 +05:30
2002-10-20 19:55:16 +05:30
base = os.path.dirname(__file__)
glade_file = "%s/%s" % (base,"changetype.glade")
2003-08-17 07:44:33 +05:30
self.glade = gtk.glade.XML(glade_file,"top","gramps")
2005-12-06 12:08:09 +05:30
self.auto1 = self.glade.get_widget("original")
self.auto2 = self.glade.get_widget("new")
2005-12-06 12:08:09 +05:30
AutoComp.fill_combo(self.auto1,const.personalEvents)
AutoComp.fill_combo(self.auto2,const.personalEvents)
# Need to display localized event names
self.auto1.child.set_text(const.display_event(
self.options.handler.options_dict['fromtype']))
self.auto2.child.set_text(const.display_event(
self.options.handler.options_dict['totype']))
2003-03-17 10:51:40 +05:30
self.title = _('Change Event Types')
self.window = self.glade.get_widget('top')
self.window.set_icon(self.parent.topWindow.get_icon())
Utils.set_titles(self.window,
2003-03-17 10:51:40 +05:30
self.glade.get_widget('title'),
self.title)
2002-10-20 19:55:16 +05:30
self.glade.signal_autoconnect({
"on_close_clicked" : self.close,
"on_delete_event" : self.on_delete_event,
"on_apply_clicked" : self.on_apply_clicked,
2002-10-20 19:55:16 +05:30
})
2005-12-06 12:08:09 +05:30
self.add_itself_to_menu()
self.window.show()
2005-12-06 12:08:09 +05:30
def run_tool(self,cli=False):
# Run tool and return results
# These are English names, no conversion needed
2005-12-06 12:08:09 +05:30
fromtype = self.options.handler.options_dict['fromtype']
totype = self.options.handler.options_dict['totype']
modified = 0
self.trans = self.db.transaction_begin()
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_handle in person.get_event_list():
if not event_handle:
continue
event = self.db.get_event_from_handle(event_handle)
if event.get_name() == fromtype:
event.set_name(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'))
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_delete_event(self,obj,b):
self.remove_itself_from_menu()
def close(self,obj):
self.remove_itself_from_menu()
self.window.destroy()
def add_itself_to_menu(self):
self.parent.child_windows[self.win_key] = self
self.parent_menu_item = gtk.MenuItem(self.title)
self.parent_menu_item.connect("activate",self.present)
self.parent_menu_item.show()
self.parent.winsmenu.append(self.parent_menu_item)
def remove_itself_from_menu(self):
del self.parent.child_windows[self.win_key]
self.parent_menu_item.destroy()
def present(self,obj):
self.window.present()
2002-10-20 19:55:16 +05:30
def on_apply_clicked(self,obj):
# Need to store English names for later comparison
self.options.handler.options_dict['fromtype'] = const.save_event(
unicode(self.auto1.child.get_text()))
self.options.handler.options_dict['totype'] = const.save_event(
unicode(self.auto2.child.get_text()))
2005-12-06 12:08:09 +05:30
modified,msg = self.run_tool(cli=False)
OkDialog(_('Change types'),msg,self.parent.topWindow)
2005-12-06 12:08:09 +05:30
# Save options
self.options.handler.save_options()
self.close(None)
2002-10-20 19:55:16 +05:30
#------------------------------------------------------------------------
#
#
#
#------------------------------------------------------------------------
2005-12-06 12:08:09 +05:30
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"),
}
2002-10-20 19:55:16 +05:30
2005-12-06 12:08:09 +05:30
#------------------------------------------------------------------------
#
#
#
#------------------------------------------------------------------------
2002-10-20 19:55:16 +05:30
register_tool(
2005-12-06 12:08:09 +05:30
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.")
2002-10-20 19:55:16 +05:30
)