gramps/gramps2/src/SelectChild.py
Alex Roitman 7329720f07 * RelLib/_*.py: Make objects serialize all the way down to built-in
python objects, so that Gramps classes are never pickled.
* Date.py, CalSdn.py: Move to RelLib.
* src/dates/Date_*.py: Use new Date.
* src/plugins/*.py: Use new Date.
* src/GrampsDb/_*.py: Use new Date, new unserialize.
* src/*.py: Use new Date.


svn: r5875
2006-02-03 22:03:53 +00:00

357 lines
13 KiB
Python
Raw Blame History

# -*- coding: utf-8 -*-
#
# 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$
#-------------------------------------------------------------------------
#
# internationalization
#
#-------------------------------------------------------------------------
from gettext import gettext as _
import gc
#-------------------------------------------------------------------------
#
# GTK/Gnome modules
#
#-------------------------------------------------------------------------
import gtk.glade
import gobject
#-------------------------------------------------------------------------
#
# gramps modules
#
#-------------------------------------------------------------------------
import RelLib
import const
import Utils
import PeopleModel
import NameDisplay
import AutoComp
import GenericFilter
import GrampsDisplay
from QuestionDialog import ErrorDialog
from WindowUtils import GladeIf
#-------------------------------------------------------------------------
#
# Constants
#
#-------------------------------------------------------------------------
UNKNOWN_REL = (RelLib.Person.CHILD_UNKNOWN,
Utils.child_relations[RelLib.Person.CHILD_UNKNOWN])
#-------------------------------------------------------------------------
#
# SelectChild
#
#-------------------------------------------------------------------------
class SelectChild:
def __init__(self,parent,db,family,person,callback):
self.parent = parent
self.db = db
self.callback = callback
self.person = person
self.family = family
self.renderer = gtk.CellRendererText()
self.xml = gtk.glade.XML(const.gladeFile,"select_child","gramps")
self.gladeif = GladeIf(self.xml)
if person:
self.default_name = person.get_primary_name().get_surname().upper()
else:
self.default_name = ""
self.gladeif.connect('select_child','delete_event', self.on_delete_event)
self.gladeif.connect('button49','clicked', self.close)
self.gladeif.connect('button48','clicked', self.on_save_child_clicked)
self.gladeif.connect('button163','clicked',self.on_child_help_clicked)
self.gladeif.connect('checkbutton1','toggled',self.on_show_toggled)
self.select_child_list = {}
self.top = self.xml.get_widget("select_child")
title_label = self.xml.get_widget('title')
Utils.set_titles(self.top,title_label,_('Add Child to Family'))
self.add_child = self.xml.get_widget("childlist")
self.mrel = self.xml.get_widget("mrel_combo")
self.frel = self.xml.get_widget("frel_combo")
self.frel_selector = AutoComp.StandardCustomSelector(
Utils.child_relations,self.frel,
RelLib.Person.CHILD_CUSTOM,RelLib.Person.CHILD_BIRTH)
self.mrel_selector = AutoComp.StandardCustomSelector(
Utils.child_relations,self.mrel,
RelLib.Person.CHILD_CUSTOM,RelLib.Person.CHILD_BIRTH)
#self.build_list(self.mrel,RelLib.Person.CHILD_BIRTH)
#self.build_list(self.frel,RelLib.Person.CHILD_BIRTH)
if self.family:
father = self.db.get_person_from_handle(self.family.get_father_handle())
mother = self.db.get_person_from_handle(self.family.get_mother_handle())
else:
if self.person.get_gender() == RelLib.Person.MALE:
self.mrel.set_sensitive(False)
else:
self.frel.set_sensitive(False)
self.likely_filter = GenericFilter.GenericFilter()
self.likely_filter.add_rule(LikelyFilter([self.person.handle]))
self.active_filter = self.likely_filter
self.top.window.set_cursor(gtk.gdk.Cursor(gtk.gdk.WATCH))
gobject.idle_add(self.redraw_child_list)
self.add_itself_to_menu()
self.add_columns(self.add_child)
self.top.show()
def build_list(self,opt_menu,sel):
cell = gtk.CellRendererText()
opt_menu.pack_start(cell,True)
opt_menu.add_attribute(cell,'text',0)
store = gtk.ListStore(str)
for val in Utils.child_relations.values():
store.append(row=[val])
opt_menu.set_model(store)
opt_menu.set_active(sel)
def add_columns(self,tree):
column = gtk.TreeViewColumn(_('Name'), self.renderer,text=0)
column.set_resizable(True)
#column.set_clickable(True)
column.set_min_width(225)
tree.append_column(column)
column = gtk.TreeViewColumn(_('ID'), self.renderer,text=1)
column.set_resizable(True)
#column.set_clickable(True)
column.set_min_width(75)
tree.append_column(column)
column = gtk.TreeViewColumn(_('Birth date'), self.renderer,text=3)
#column.set_resizable(True)
column.set_clickable(True)
tree.append_column(column)
def on_delete_event(self,obj,b):
self.gladeif.close()
self.remove_itself_from_menu()
gc.collect()
def close(self,obj):
self.gladeif.close()
self.remove_itself_from_menu()
self.top.destroy()
gc.collect()
def add_itself_to_menu(self):
self.parent.child_windows[self] = self
self.parent_menu_item = gtk.MenuItem(_('Add Child to Family'))
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]
self.parent_menu_item.destroy()
def present(self,obj):
self.top.present()
def on_child_help_clicked(self,obj):
"""Display the relevant portion of GRAMPS manual"""
GrampsDisplay.help('gramps-edit-quick')
def redraw_child_list(self):
self.refmodel = PeopleModel.PeopleModel(self.db,self.active_filter)
self.add_child.set_model(self.refmodel)
self.top.window.set_cursor(None)
def select_function(self,store,path,node,id_list):
id_list.append(self.refmodel.get_value(node,PeopleModel.COLUMN_INT_ID))
def get_selected_ids(self):
mlist = []
self.add_child.get_selection().selected_foreach(self.select_function,mlist)
return mlist
def on_save_child_clicked(self,obj):
idlist = self.get_selected_ids()
if not idlist or not idlist[0]:
return
handle = idlist[0]
select_child = self.db.get_person_from_handle(handle)
if self.person.get_handle() == handle:
ErrorDialog(_("Error selecting a child"),
_("A person cannot be linked as his/her own child"),
self.top)
return
trans = self.db.transaction_begin()
if self.family == None:
self.family = RelLib.Family()
self.db.add_family(self.family,trans)
self.person.add_family_handle(self.family.get_handle())
self.db.commit_person(self.person,trans)
if self.person.get_gender() == RelLib.Person.MALE:
self.family.set_father_handle(self.person.get_handle())
else:
self.family.set_mother_handle(self.person.get_handle())
self.db.commit_family(self.family,trans)
# check that selected child is not already a child in family
if handle in self.family.get_child_handle_list():
ErrorDialog(_("Error selecting a child"),
_("The person is already linked as child"),
self.top)
return
# check that selected child is not already a spouse in family
if handle in (self.family.get_father_handle(),self.family.get_mother_handle()):
ErrorDialog(_("Error selecting a child"),
_("A person cannot be linked as his/her own child"),
self.top)
return
# TODO: Add child ordered by birth day
self.family.add_child_handle(select_child.get_handle())
mrel = self.mrel_selector.get_values()
mother = self.db.get_person_from_handle(self.family.get_mother_handle())
if mother and mother.get_gender() != RelLib.Person.FEMALE:
if mrel[0] == RelLib.Person.CHILD_BIRTH:
mrel = UNKNOWN_REL
frel = self.frel_selector.get_values()
father = self.db.get_person_from_handle(self.family.get_father_handle())
if father and father.get_gender() != RelLib.Person.MALE:
if frel[0] == RelLib.Person.CHILD_BIRTH:
frel = UNKNOWN_REL
select_child.add_parent_family_handle(self.family.get_handle(),
mrel,frel)
self.db.commit_person(select_child,trans)
self.db.commit_family(self.family,trans)
n = NameDisplay.displayer.display(select_child)
self.db.transaction_commit(trans,_("Add Child to Family (%s)") % n)
self.close(obj)
self.callback()
def on_show_toggled(self,obj):
if obj.get_active():
self.active_filter = None
else:
self.active_filter = self.likely_filter
self.top.window.set_cursor(gtk.gdk.Cursor(gtk.gdk.WATCH))
while(gtk.events_pending()):
gtk.main_iteration()
self.redraw_child_list()
def north_american(self,val):
if self.person.get_gender() == RelLib.Person.MALE:
return self.person.get_primary_name().get_surname()
elif self.family:
f = self.family.get_father_handle()
if f:
pname = f.get_primary_name()
return (pname.get_surname_prefix(),pname.get_surname())
return ("","")
def no_name(self,val):
return ("","")
def latin_american(self,val):
if self.family:
father = self.family.get_father_handle()
mother = self.family.get_mother_handle()
if not father or not mother:
return ("","")
fsn = father.get_primary_name().get_surname()
msn = mother.get_primary_name().get_surname()
if not father or not mother:
return ("","")
try:
return ("","%s %s" % (fsn.split()[0],msn.split()[0]))
except:
return ("","")
else:
return ("","")
def icelandic(self,val):
fname = ""
if self.person.get_gender() == RelLib.Person.MALE:
fname = self.person.get_primary_name().get_first_name()
elif self.family:
f = self.family.get_father_handle()
if f:
fname = f.get_primary_name().get_first_name()
if fname:
fname = fname.split()[0]
if val == 0:
return ("","%ssson" % fname)
elif val == 1:
return ("","%sd<EFBFBD>ttir" % fname)
else:
return ("","")
#-------------------------------------------------------------------------
#
# Likely Filters
#
#-------------------------------------------------------------------------
class LikelyFilter(GenericFilter.Rule):
labels = [ 'Person handle' ]
category = _('General filters')
def prepare(self,db):
person = db.get_person_from_handle(self.list[0])
if person.birth_handle:
birth = db.get_event_from_handle(person.birth_handle)
dateobj = RelLib.Date(birth.date)
year = dateobj.get_year()
dateobj.set_year(year+10)
self.lower = dateobj.sortval
dateobj.set_year(year+70)
self.upper = dateobj.sortval
else:
self.lower = None
self.upper = None
def apply(self,db,person):
if not person.birth_handle or (self.upper == None and
self.lower == None):
return True
event = db.get_event_from_handle(person.birth_handle)
return (event.date == None or event.date.sortval == 0 or
self.lower < event.date.sortval < self.upper)