gramps/src/DisplayTabs/_NameEmbedList.py
2009-07-24 06:14:21 +00:00

210 lines
6.9 KiB
Python

#
# Gramps - a GTK+/GNOME based genealogy program
#
# Copyright (C) 2000-2006 Donald N. Allingham
# 2009 Benny Malengier
#
# 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$
#-------------------------------------------------------------------------
#
# Python classes
#
#-------------------------------------------------------------------------
import gtk
#-------------------------------------------------------------------------
#
# Python classes
#
#-------------------------------------------------------------------------
from gettext import gettext as _
#-------------------------------------------------------------------------
#
# GRAMPS classes
#
#-------------------------------------------------------------------------
import gen.lib
import Errors
from DdTargets import DdTargets
from DisplayTabs._NameModel import NameModel
from DisplayTabs._GroupEmbeddedList import GroupEmbeddedList
#-------------------------------------------------------------------------
#
#
#
#-------------------------------------------------------------------------
class NameEmbedList(GroupEmbeddedList):
_HANDLE_COL = 2
_DND_TYPE = DdTargets.NAME
_WORKGROUP = NameModel.ALTINDEX
_MSG = {
'add' : _('Create and add a new name'),
'del' : _('Remove the existing name'),
'edit' : _('Edit the selected name'),
'up' : _('Move the selected name upwards'),
'down' : _('Move the selected name downwards'),
}
#index = column in model. Value =
# (name, sortcol in model, width, markup/text, weigth_col
_column_names = [
(_('Name'), -1, 250, 0, NameModel.COL_FONTWEIGHT[0]),
(_('Type'), NameModel.COL_TYPE[0], 100, 0, -1),
None,
None,
(_('Group As'), NameModel.COL_GROUPAS[0],100, 0, -1),
(_('Source'), NameModel.COL_HASSOURCE[0],60, 0, -1),
(_('Note Preview'), NameModel.COL_NOTEPREVIEW[0], 250, 0, -1),
]
def __init__(self, dbstate, uistate, track, data, person, callback):
self.data = data
self.person = person
self.callback = callback
GroupEmbeddedList.__init__(self, dbstate, uistate, track, _('_Names'),
NameModel, move_buttons=True)
self.tree.expand_all()
def get_data(self):
return ([self.person.get_primary_name()],
self.data)
def groups(self):
"""
Return the (group key, group name)s in the order as given by get_data()
"""
return ((None, NameModel.DEFNAME), (None, NameModel.ALTNAME))
def column_order(self):
"""
The columns to show as a tuple of tuples containing
tuples (show/noshow, model column)
"""
return ((1, 0), (1, 1), (1, 4), (1, 5), (1, 6))
def get_popup_menu_items(self):
if self._tmpgroup == self._WORKGROUP:
return [
(True, True, gtk.STOCK_ADD, self.add_button_clicked),
(False,True, gtk.STOCK_EDIT, self.edit_button_clicked),
(True, True, gtk.STOCK_REMOVE, self.del_button_clicked),
(True, False, _('Set as default name'), self.name_button_clicked),
]
else:
return [
(True, True, gtk.STOCK_ADD, self.add_button_clicked),
(False,True, gtk.STOCK_EDIT, self.edit_button_clicked),
]
def name_button_clicked(self, obj):
name = self.get_selected()
if name and name[1]:
self.set_default_name(name[1])
def set_default_name(self, name):
pname = self.person.get_primary_name()
self.person.set_primary_name(name)
remove = []
for altname in self.data:
if altname.is_equal(name):
remove.append(altname)
for altname in remove:
self.data.remove(altname)
#only non empty name should move to alternative names
if not name.is_equal(gen.lib.Name()):
self.data.append(pname)
self.rebuild()
self.callback()
def update_defname(self):
"""
callback from person editor if change to the preferred name happens
"""
self.model.update_defname(self.person.get_primary_name())
self.tree.expand_all()
def add_button_clicked(self, obj):
name = gen.lib.Name()
try:
from Editors import EditName
EditName(self.dbstate, self.uistate, self.track,
name, self.add_callback)
except Errors.WindowActiveError:
pass
def add_callback(self, name):
self.get_data()[self._WORKGROUP].append(name)
self.rebuild()
def edit_button_clicked(self, obj):
name = self.get_selected()
if name and name[1] is not None:
try:
from Editors import EditName
if name[0] == NameModel.ALTINDEX:
EditName(self.dbstate, self.uistate, self.track,
name[1], self.edit_callback)
elif name[0] == NameModel.DEFINDEX:
EditName(self.dbstate, self.uistate, self.track,
name[1], self.editdef_callback)
except Errors.WindowActiveError:
pass
def edit_callback(self, name):
self.rebuild()
def editdef_callback(self, name):
"""
callback after default name has changed
"""
self.rebuild()
self.callback()
def dropnotworkgroup(self, row, obj):
"""
Drop of obj on row that is not WORKGROUP
"""
if row[0] == NameModel.DEFINDEX:
#drop on default name
self.set_default_name(obj)
def move_away_work(self, row_from, row_to, obj):
"""
move from the workgroup to a not workgroup
we allow to change the default name like this
"""
if row_from[0] == self._WORKGROUP and row_to[0] == NameModel.DEFINDEX:
self.set_default_name(obj)
def post_rebuild(self, prebuildpath):
"""
Allow post rebuild specific handling.
@param prebuildpath: path selected before rebuild, None if none
@type prebuildpath: tree path
"""
self.tree.expand_all()
if not prebuildpath is None:
self.selection.select_path(prebuildpath)