gramps/gramps2/src/SelectEvent.py

#
# Gramps - a GTK+/GNOME based genealogy program
#
# Copyright (C) 2003-2005  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$

#-------------------------------------------------------------------------
#
# internationalization
#
#-------------------------------------------------------------------------
from gettext import gettext as _

#-------------------------------------------------------------------------
#
# GTK/Gnome modules
#
#-------------------------------------------------------------------------

import gtk
import gtk.glade

#-------------------------------------------------------------------------
#
# gramps modules
#
#-------------------------------------------------------------------------
import const
import Utils
import ListModel
import RelLib
import DateHandler

#-------------------------------------------------------------------------
#
# SelectEvent
#
#-------------------------------------------------------------------------
class SelectEvent:

    def __init__(self,db,title,parent_window=None):

        self.db = db
        self.glade = gtk.glade.XML(const.gladeFile,"select_person","gramps")
        self.top = self.glade.get_widget('select_person')
        title_label = self.glade.get_widget('title')
        self.elist =  self.glade.get_widget('plist')

        Utils.set_titles(self.top,title_label,title)

        titles = [(_('Description'),4,250), (_('ID'),1,75),
                  (_('Type'),2,75), (_('Date'),3,150), ('',4,0) ] 
        self.ncols = len(titles)      

        self.model = ListModel.ListModel(self.elist,titles)

        self.redraw()
        self.top.show()

        if parent_window:
            self.top.set_transient_for(parent_window)

    def redraw(self):
        self.model.clear()
        self.model.new_model()

        for handle in self.db.get_event_handles():
            event = self.db.get_event_from_handle(handle)
            desc = event.get_description()
            etype = event.get_type()
            if etype[0] == RelLib.Event.CUSTOM \
                   or not Utils.personal_events.has_key(etype[0]):
                name = etype[1]
            else:
                name = Utils.personal_events[etype[0]]
            the_id = event.get_gramps_id()
            place_handle = event.get_place_handle()
            if place_handle:
                pname = self.db.get_place_from_handle(place_handle).get_title()
            else:
                pname = u''
            date = DateHandler.get_date(event)
            cause = event.get_cause()
            self.model.add([desc,the_id,name,date],handle)

        self.model.connect_model()

    def run(self):
        val = self.top.run()

        if val == gtk.RESPONSE_OK:
            store,node = self.model.get_selected()
            if node:
                data = self.model.get_data(node,range(self.ncols))
                handle = data[4]
                return_value = self.db.get_event_from_handle(handle)
            else:
                return_value = None
            self.top.destroy()
	    return return_value
        else:
            self.top.destroy()
            return None