gramps/gramps2/src/Find.py

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

"""interface for opening a find person dialog for gramps
"""

__author__ = 'Don Allingham'

#-------------------------------------------------------------------------
#
# python modules
#
#-------------------------------------------------------------------------
import string

#-------------------------------------------------------------------------
#
# Gnome modules
#
#-------------------------------------------------------------------------
import gtk
from gnome.ui import *

#-------------------------------------------------------------------------
#
# gramps modules
#
#-------------------------------------------------------------------------
import GrampsCfg
import AutoComp
from intl import gettext as _

#-------------------------------------------------------------------------
#
# FindBase
#
#-------------------------------------------------------------------------
class FindBase:
    """Opens find person dialog for gramps"""
    
    def __init__(self,clist,task,name,db):
        """Opens a dialog box instance that allows users to
        search for a person.

        clist - GtkCList containing the people information
        task - function to call to change the active person"""

        self.db = db
        self.clist = clist
        self.nlist = []
        self.task = task
        title = "%s - GRAMPS" % name
        self.top = GnomeDialog(title,STOCK_BUTTON_PREV,
                               STOCK_BUTTON_NEXT,STOCK_BUTTON_CLOSE)
        self.top.set_policy(0,1,0)
        self.top.vbox.set_spacing(5)
        self.top.vbox.pack_start(gtk.GtkLabel(name),0,0,5)
        self.top.vbox.pack_start(gtk.GtkHSeparator(),0,0,0)
        self.entry = gtk.GtkEntry()
        self.top.vbox.pack_start(self.entry,0,0,25)
        self.top.button_connect(0,self.on_prev_clicked)
        self.top.button_connect(1,self.on_next_clicked)
        self.top.button_connect(2,self.on_close_clicked)
        self.top.set_usize(350,175)
        self.top.set_default(1)
        self.top.show_all()
        self.top.editable_enters(self.entry)
        self.entry.grab_focus()

    def get_value(self,id):
        return None
    
    def enable_autocomp(self):
        if GrampsCfg.autocomp:
            self.comp = AutoComp.AutoEntry(self.entry,self.nlist)
        
    def advance(self,func):
        try:
            self.row = self.clist.selection[0]
        except IndexError:
            gtk.gdk_beep()
            return

        text = self.entry.get_text()
        if self.row == None or text == "":
            gtk.gdk_beep()
            return
        orow = self.row
        func()
        while self.row != orow:
            id = self.clist.get_row_data(self.row)
            if id == None:
                func()
                continue
            if string.find(string.upper(self.get_value(id)),string.upper(text)) >= 0:
                self.task(self.row)
                return
            func()
        gtk.gdk_beep()

    def forward(self):
        self.row = self.row + 1
        if self.row == self.clist.rows:
            self.row = 0

    def backward(self):
        self.row = self.row - 1
        if self.row < 0:
            self.row =  self.clist.rows

    def on_close_clicked(self,obj):
        """Destroys the window in response to a close window button press"""
        self.top.destroy()

    def on_next_clicked(self,obj):
        """Advances to the next person that matches the dialog text"""
        self.advance(self.forward)

    def on_prev_clicked(self,obj):
        """Advances to the previous person that matches the dialog text"""
        self.advance(self.backward)

#-------------------------------------------------------------------------
#
# FindPerson
#
#-------------------------------------------------------------------------
class FindPerson(FindBase):
    """Opens a Find Person dialog for GRAMPS"""
    
    def __init__(self,clist,task,db):
        """Opens a dialog box instance that allows users to
        search for a person.

        clist - GtkCList containing the people information
        task - function to call to change the active person"""
        
        FindBase.__init__(self,clist,task,_("Find Person"),db)
        for n in self.db.getPersonKeys():
            val = self.db.getPersonDisplay(n)
            self.nlist.append(val[0])
        self.enable_autocomp()

    def get_value(self,id):
        return self.db.getPersonDisplay(id)[0]
    

#-------------------------------------------------------------------------
#
# FindPlace
#
#-------------------------------------------------------------------------
class FindPlace(FindBase):
    """Opens a Find Place dialog for GRAMPS"""
    
    def __init__(self,clist,task,db):
        """Opens a dialog box instance that allows users to
        search for a place.

        clist - GtkCList containing the people information
        task - function to call to change the active person"""
        
        FindBase.__init__(self,clist,task,_("Find Place"),db)
        for n in self.db.getPlaceKeys():
            self.nlist.append(self.db.getPlaceDisplay(n)[0])
        self.enable_autocomp()
        
    def get_value(self,id):
        return  self.db.getPlaceDisplay(id)[0]

#-------------------------------------------------------------------------
#
# FindSource
#
#-------------------------------------------------------------------------
class FindSource(FindBase):
    """Opens a Find Place dialog for GRAMPS"""
    
    def __init__(self,clist,task,db):
        """Opens a dialog box instance that allows users to
        search for a place.

        clist - GtkCList containing the people information
        task - function to call to change the active person"""
        
        FindBase.__init__(self,clist,task,_("Find Source"),db)
        for n in self.db.getSourceKeys():
            self.nlist.append(n[0])
        self.enable_autocomp()
        
    def get_value(self,id):
        return  self.db.getSourceDisplay(id)[0]

#-------------------------------------------------------------------------
#
# FindMedia
#
#-------------------------------------------------------------------------
class FindMedia(FindBase):
    """Opens a Find Media Object dialog for GRAMPS"""
    
    def __init__(self,clist,task,db):
        """Opens a dialog box instance that allows users to
        search for a place.

        clist - GtkCList containing the people information
        task - function to call to change the active person"""
        
        FindBase.__init__(self,clist,task,_("Find Media Object"),db)
        for n in self.db.getObjectMap().values():
            self.nlist.append(n.getDescription())
        self.enable_autocomp()
        
    def advance(self,func):
        try:
            self.row = self.clist.selection[0]
        except IndexError:
            gtk.gdk_beep()
            return

        text = self.entry.get_text()
        if self.row == None or text == "":
            gtk.gdk_beep()
            return
        orow = self.row
        func()
        while self.row != orow:
            value = self.clist.get_row_data(self.row)
            if value == None:
                func()
                continue
            name = value.getDescription()
            if string.find(string.upper(name),string.upper(text)) >= 0:
                self.task(self.row)
                return
            func()
        gtk.gdk_beep()