#------------------------------------------------------------------------- # # Python modules # #------------------------------------------------------------------------- from gettext import gettext as _ import gc import re import cgi #------------------------------------------------------------------------- # # GTK/Gnome modules # #------------------------------------------------------------------------- import gobject import gtk import pango #------------------------------------------------------------------------- # # Gramps Modules # #------------------------------------------------------------------------- import RelLib import PageView import NameDisplay import Utils import DateHandler import ImgManip import ReportUtils import Config import GrampsWidgets import Errors _GenderCode = { RelLib.Person.MALE : u'\u2642', RelLib.Person.FEMALE : u'\u2640', RelLib.Person.UNKNOWN : u'\u2650', } _NAME_START = 0 _LABEL_START = 1 _LABEL_STOP = 2 _DATA_START = _LABEL_STOP _DATA_STOP = _DATA_START+1 _BTN_START = _DATA_STOP _BTN_STOP = _BTN_START+2 _PLABEL_START = 2 _PLABEL_STOP = _PLABEL_START+1 _PDATA_START = _PLABEL_STOP _PDATA_STOP = _PDATA_START+2 _PDTLS_START = _PLABEL_STOP _PDTLS_STOP = _PDTLS_START+2 _CLABEL_START = _PLABEL_START+1 _CLABEL_STOP = _CLABEL_START+1 _CDATA_START = _CLABEL_STOP _CDATA_STOP = _CDATA_START+1 _CDTLS_START = _CDATA_START _CDTLS_STOP = _CDTLS_START+1 _ALABEL_START = 1 _ALABEL_STOP = _ALABEL_START+1 _ADATA_START = _ALABEL_STOP _ADATA_STOP = _ADATA_START+3 _SDATA_START = 3 _SDATA_STOP = 5 class AttachList: def __init__(self): self.list = [] self.max_x = 0 self.max_y = 0 def attach(self,widget,x0,x1,y0,y1,xoptions=gtk.EXPAND|gtk.FILL, yoptions=gtk.EXPAND|gtk.FILL): assert(x1>x0) self.list.append((widget,x0,x1,y0,y1,xoptions,yoptions)) self.max_x = max(self.max_x,x1) self.max_y = max(self.max_y,y1) class FamilyView(PageView.PersonNavView): def __init__(self,dbstate,uistate): PageView.PersonNavView.__init__(self,'Relationship View',dbstate,uistate) dbstate.connect('database-changed',self.change_db) dbstate.connect('active-changed',self.change_person) dbstate.db.connect('family-update',self.redraw) dbstate.db.connect('family-add',self.redraw) dbstate.db.connect('family-delete',self.redraw) dbstate.db.connect('person-update',self.redraw) dbstate.db.connect('person-add',self.redraw) dbstate.db.connect('person-delete',self.redraw) self.show_siblings = Config.get_family_siblings() if self.show_siblings == None: self.show_siblings = True self.show_details = Config.get_family_details() if self.show_details == None: self.show_details = True self.redrawing = False self.child = None def get_stock(self): """ Returns the name of the stock icon to use for the display. This assumes that this icon has already been registered with GNOME as a stock icon. """ return 'gramps-family' def build_widget(self): self.scroll = gtk.ScrolledWindow() self.scroll.set_policy(gtk.POLICY_AUTOMATIC,gtk.POLICY_AUTOMATIC) self.scroll.show() self.vbox = gtk.VBox() self.vbox.show() self.child = None self.scroll.add_with_viewport(self.vbox) return self.scroll def ui_definition(self): """ Specifies the UIManager XML code that defines the menus and buttons associated with the interface. """ return ''' ''' def define_actions(self): PageView.PersonNavView.define_actions(self) self.add_toggle_action('Details', None, _('Show details'), None, None, self.details_toggle, self.show_details) self.add_toggle_action('Siblings', None, _('Show siblings'), None, None, self.siblings_toggle, self.show_siblings) def siblings_toggle(self,obj): self.show_siblings = obj.get_active() self.change_person(self.dbstate.active.handle) Config.save_family_siblings(self.show_siblings) def details_toggle(self,obj): self.show_details = obj.get_active() self.change_person(self.dbstate.active.handle) Config.save_family_details(self.show_details) def change_db(self,db): if self.child: self.vbox.remove(self.child) self.child = None self.dbstate.db.connect('family-update',self.redraw) self.dbstate.db.connect('family-add',self.redraw) self.dbstate.db.connect('family-delete',self.redraw) self.dbstate.db.connect('person-update',self.redraw) self.dbstate.db.connect('person-add',self.redraw) self.dbstate.db.connect('person-delete',self.redraw) def get_name(self,handle,use_gender=False): if handle: p = self.dbstate.db.get_person_from_handle(handle) name = NameDisplay.displayer.display(p) if use_gender: gender = _GenderCode[p.gender] else: gender = "" return (name,gender) else: return (_(u"Unknown"),"") def redraw(self,*obj): if self.dbstate.active: self.change_person(self.dbstate.active.handle) def change_person(self,obj): if self.redrawing: return self.redrawing = True old_child = self.child self.attach = AttachList() print "PERSON",obj person = self.dbstate.db.get_person_from_handle(obj) if not person: self.redrawing = False return self.row = 5 family_handle_list = person.get_parent_family_handle_list() if family_handle_list: for (family_handle,mrel,frel) in family_handle_list: if family_handle: self.write_parents(family_handle) else: self.write_label("%s:" % _('Parents'),None,True) self.row += 1 family_handle_list = person.get_family_handle_list() if family_handle_list: for family_handle in family_handle_list: if family_handle: self.write_family(family_handle) else: self.write_label("%s:" % _('Family'),None,False) self.row += 1 self.row = 1 self.write_title(person) # Here it is necessary to beat GTK into submission. For some # bizzare reason, if you have an empty column that is spanned, # you lose the appropriate FILL handling. So, we need to see if # column 3 is unused (usually if there is no siblings or children. # If so, we need to subtract one index of each x coord > 3. found = False for d in self.attach.list: if d[1] == 4 or d[2] == 4: found = True if found: cols = self.attach.max_x else: cols = self.attach.max_x-1 self.child = gtk.Table(self.attach.max_y,cols) self.child.set_border_width(12) self.child.set_col_spacings(12) self.child.set_row_spacings(6) for d in self.attach.list: x0 = d[1] x1 = d[2] if not found: if x0 > 4: x0 -= 1 if x1 > 4: x1 -= 1 self.child.attach(d[0],x0,x1,d[3],d[4],d[5],d[6]) self.child.show_all() if old_child: self.vbox.remove(old_child) self.vbox.pack_start(self.child,False) self.redrawing = False def write_title(self,person): # name and edit button name = NameDisplay.displayer.display(person) text = '%s %s' % (cgi.escape(name), _GenderCode[person.gender]) label = GrampsWidgets.MarkupLabel(text) button = GrampsWidgets.IconButton(self.edit_button_press,person.handle) hbox = GrampsWidgets.LinkBox(label,button) # GRAMPS ID self.row = 1 self.write_person_data("%s:" % _('ID'),person.gramps_id) # birth/death events birth_ref = person.get_birth_ref() if birth_ref: birth = self.dbstate.db.get_event_from_handle(birth_ref.ref) else: birth = None self.write_person_event("%s:" % _('Birth'),birth) death_ref = person.get_death_ref() if death_ref: death = self.dbstate.db.get_event_from_handle(death_ref.ref) else: death = None self.write_person_event("%s:" % _('Death'),death) # separator end = self.attach.max_x sep = gtk.HSeparator() sep.show() self.attach.attach(hbox,_NAME_START,end,0,1,gtk.FILL|gtk.EXPAND) # image image_list = person.get_media_list() if image_list: mobj = self.dbstate.db.get_object_from_handle(image_list[0].ref) if mobj.get_mime_type()[0:5] == "image": pixbuf = ImgManip.get_thumbnail_image(mobj.get_path()) image = gtk.Image() image.set_from_pixbuf(pixbuf) image.show() self.attach.attach(image,end,end+1,0,4,xoptions=gtk.SHRINK|gtk.FILL) self.attach.attach(sep,0,self.attach.max_x,4,5) def write_person_event(self, ename, event): if event: dobj = event.get_date_object() phandle = event.get_place_handle() if phandle: pname = self.place_name(phandle) else: pname = None date_str = DateHandler.displayer.display(dobj) value = { 'date' : DateHandler.displayer.display(dobj), 'place' : pname, } else: pname = None dobj = None if dobj: if pname: self.write_person_data(ename, _('%(date)s in %(place)s') % value) else: self.write_person_data(ename,'%(date)s' % value) elif pname: self.write_person_data(ename,pname) else: self.write_person_data(ename,'') def write_person_data(self,title,data): self.attach.attach(GrampsWidgets.BasicLabel(title),_ALABEL_START, _ALABEL_STOP,self.row,self.row+1, xoptions=gtk.FILL|gtk.SHRINK) self.attach.attach(GrampsWidgets.BasicLabel(data),_ADATA_START,_ADATA_STOP, self.row,self.row+1) self.row += 1 def write_label(self,title,family,is_parent): msg = "%s" % cgi.escape(title) self.attach.attach(GrampsWidgets.MarkupLabel(msg),_LABEL_START,_LABEL_STOP, self.row,self.row+1,gtk.SHRINK|gtk.FILL) if family: value = family.gramps_id else: value = "" self.attach.attach(GrampsWidgets.BasicLabel(value),_DATA_START,_DATA_STOP, self.row,self.row+1,gtk.SHRINK|gtk.FILL) hbox = gtk.HBox() hbox.set_spacing(6) if is_parent: call_fcn = self.add_parent_family else: call_fcn = self.add_family add = GrampsWidgets.IconButton(call_fcn,None,gtk.STOCK_ADD) hbox.pack_start(add,False) if family: edit = GrampsWidgets.IconButton(self.edit_family,family.handle, gtk.STOCK_EDIT) hbox.pack_start(edit,False) delete = GrampsWidgets.IconButton(self.delete_family,family.handle, gtk.STOCK_REMOVE) hbox.pack_start(delete,False) self.attach.attach(hbox,_BTN_START,_BTN_STOP,self.row,self.row+1) self.row += 1 ###################################################################### def write_parents(self,family_handle): family = self.dbstate.db.get_family_from_handle(family_handle) self.write_label("%s:" % _('Parents'),family,True), self.write_person(_('Father'),family.get_father_handle()) if self.show_details: value = self.info_string(family.get_father_handle()) if value: self.attach.attach(GrampsWidgets.BasicLabel(value),_PDTLS_START, _PDTLS_STOP,self.row, self.row+1) self.row += 1 self.write_person(_('Mother'),family.get_mother_handle()) if self.show_details: value = self.info_string(family.get_mother_handle()) if value: self.attach.attach(GrampsWidgets.BasicLabel(value),_PDTLS_START, _PDTLS_STOP, self.row, self.row+1) self.row += 1 if self.show_siblings: active = self.dbstate.active.handle child_list = [handle for handle in family.get_child_handle_list()\ if handle != active] label = _("Siblings") if child_list: for child in child_list: self.write_child(label,child) label = u"" self.row += 1 def write_person(self,title,handle): if title: format = '%s: ' else: format = "%s" label = GrampsWidgets.MarkupLabel(format % cgi.escape(title)) self.attach.attach(label,_PLABEL_START,_PLABEL_STOP,self.row,self.row+1, xoptions=gtk.FILL|gtk.SHRINK) link_label = GrampsWidgets.LinkLabel(self.get_name(handle,True), self.button_press,handle) button = GrampsWidgets.IconButton(self.edit_button_press,handle) self.attach.attach(GrampsWidgets.LinkBox(link_label,button),_PDATA_START, _PDATA_STOP,self.row,self.row+1) self.row += 1 def write_child(self,title,handle): if title: format = '%s: ' else: format = "%s" label = GrampsWidgets.MarkupLabel(format % cgi.escape(title)) self.attach.attach(label,_CLABEL_START,_CLABEL_STOP,self.row,self.row+1, xoptions=gtk.FILL|gtk.SHRINK) link_label = GrampsWidgets.LinkLabel(self.get_name(handle,True), self.button_press,handle) button = GrampsWidgets.IconButton(self.edit_button_press,handle) self.attach.attach(GrampsWidgets.LinkBox(link_label,button),_CDATA_START, _CDATA_STOP,self.row,self.row+1, xoptions=gtk.EXPAND|gtk.FILL) self.row += 1 if self.show_details: value = self.info_string(handle) if value: self.attach.attach(GrampsWidgets.BasicLabel(value),_CDTLS_START, _CDTLS_STOP,self.row, self.row+1) self.row += 1 def write_data(self,title,start_col=_SDATA_START,stop_col=_SDATA_STOP): self.attach.attach(GrampsWidgets.BasicLabel(title),start_col,stop_col, self.row, self.row+1, xoptions=gtk.EXPAND|gtk.FILL) self.row += 1 def info_string(self,handle): child = self.dbstate.db.get_person_from_handle(handle) if not child: return None birth_ref = child.get_birth_ref() death_ref = child.get_death_ref() value = None if birth_ref or death_ref: info = ReportUtils.get_birth_death_strings(self.dbstate.db,child) bdate = info[0] ddate = info[4] if bdate and ddate: value = _("b. %s, d. %s") % (bdate,ddate) elif bdate: value = _("b. %s") % (bdate) elif ddate: value = _("d. %s") % (ddate) return value def button_press(self,obj,event,handle): if event.type == gtk.gdk.BUTTON_PRESS and event.button == 1: self.dbstate.change_active_handle(handle) def write_relationship(self,family): rtype = family.get_relationship() if type(rtype) == tuple: if rtype[0] == RelLib.Family.CUSTOM: rel_text = rtype[1] else: rel_text = Utils.family_relations[rtype[0]] else: rel_text = Utils.family_relations[rtype] self.write_data(_('Relationship type: %s') % rel_text) def place_name(self,handle): p = self.dbstate.db.get_place_from_handle(handle) return p.get_title() def write_marriage(self,family): value = False for event_ref in family.get_event_ref_list(): handle = event_ref.ref event = self.dbstate.db.get_event_from_handle(handle) etype = event.get_type() if etype[0] == RelLib.Event.MARRIAGE: self.write_event_ref(_('Marriage'),event) value = True return value def write_event_ref(self, ename, event,start_col=_SDATA_START,stop_col=_SDATA_STOP): if event: dobj = event.get_date_object() phandle = event.get_place_handle() if phandle: pname = self.place_name(phandle) else: pname = None date_str = DateHandler.displayer.display(dobj) value = { 'date' : DateHandler.displayer.display(dobj), 'place' : pname, 'event_type' : ename, } else: pname = None dobj = None value = { 'event_type' : ename, } if dobj: if pname: self.write_data(_('%(event_type)s: %(date)s in %(place)s') % value,start_col,stop_col) else: self.write_data(_('%(event_type)s: %(date)s') % value, start_col, stop_col) elif pname: self.write_data(_('%(event_type)s: %(place)s') % value, start_col,stop_col) else: self.write_data(_('%(event_type)s:') % value, start_col, stop_col) def write_family(self,family_handle): family = self.dbstate.db.get_family_from_handle(family_handle) father_handle = family.get_father_handle() mother_handle = family.get_mother_handle() if self.dbstate.active.handle == father_handle: handle = mother_handle else: handle = father_handle self.write_label("%s:" % _('Family'),family,False) if handle: self.write_person(_('Spouse'),handle) value = self.info_string(handle) if value: self.attach.attach(GrampsWidgets.BasicLabel(value),_PDTLS_START, _PDTLS_STOP,self.row, self.row+1) self.row += 1 if not self.write_marriage(family): self.write_relationship(family) child_list = family.get_child_handle_list() label = _("Children") if child_list: for child in child_list: self.write_child(label,child) label = u"" self.row += 1 def edit_button_press(self, obj, event, handle): if event.type == gtk.gdk.BUTTON_PRESS and event.button == 1: from Editors import EditPerson person = self.dbstate.db.get_person_from_handle(handle) try: EditPerson(self.dbstate, self.uistate, [], person) except Errors.WindowActiveError: pass def edit_person(self,obj,handle): from Editors import EditPerson person = self.dbstate.db.get_person_from_handle(handle) try: EditPerson(self.dbstate, self.uistate, [], person) except Errors.WindowActiveError: pass def edit_family(self,obj,event,handle): if event.type == gtk.gdk.BUTTON_PRESS and event.button == 1: from Editors import EditFamily family = self.dbstate.db.get_family_from_handle(handle) try: EditFamily(self.dbstate,self.uistate,[],family) except Errors.WindowActiveError: pass def add_family(self,obj,event,handle): if event.type == gtk.gdk.BUTTON_PRESS and event.button == 1: from Editors import EditFamily family = RelLib.Family() person = self.dbstate.active if person.gender == RelLib.Person.MALE: family.set_father_handle(person.handle) else: family.set_mother_handle(person.handle) try: EditFamily(self.dbstate,self.uistate,[],family) except Errors.WindowActiveError: pass def add_parent_family(self,obj,event,handle): if event.type == gtk.gdk.BUTTON_PRESS and event.button == 1: from Editors import EditFamily family = RelLib.Family() person = self.dbstate.active family.add_child_handle(person.handle) try: EditFamily(self.dbstate,self.uistate,[],family) except Errors.WindowActiveError: pass def delete_family(self,obj,event,handle): if event.type == gtk.gdk.BUTTON_PRESS and event.button == 1: print "Delete Family",handle def change_to(self,obj,handle): self.dbstate.change_active_handle(handle)