#------------------------------------------------------------------------- # # 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 Config import GrampsWidgets import Errors from PluginUtils import ReportUtils _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) 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.connect_to_db(dbstate.db) self.redrawing = False self.child = None def connect_to_db(self, db): db.connect('person-update', self.person_update) db.connect('person-rebuild', self.person_rebuild) db.connect('family-update', self.family_update) db.connect('family-add', self.family_add) db.connect('family-delete', self.family_delete) db.connect('family-rebuild', self.family_rebuild) def person_update(self, handle_list): if self.dbstate.active: while not self.change_person(self.dbstate.active.handle): pass def person_rebuild(self): if self.dbstate.active: while not self.change_person(self.dbstate.active.handle): pass def family_update(self, handle_list): if self.dbstate.active: while not self.change_person(self.dbstate.active.handle): pass def family_add(self, handle_list): if self.dbstate.active: while not self.change_person(self.dbstate.active.handle): pass def family_delete(self, handle_list): if self.dbstate.active: while not self.change_person(self.dbstate.active.handle): pass def family_rebuild(self): if self.dbstate.active: while not self.change_person(self.dbstate.active.handle): pass 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): self.connect_to_db(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 False self.redrawing = True old_child = self.child self.attach = AttachList() 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 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(9) 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 return True def write_title(self, person): # name and edit button name = NameDisplay.displayer.display(person) fmt = '%s %s' text = fmt % (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(12) if is_parent: call_fcn = self.add_parent_family del_fcn = self.delete_parent_family else: call_fcn = self.add_family del_fcn = self.delete_family add = GrampsWidgets.IconButton(call_fcn, None, gtk.STOCK_ADD) hbox.pack_start(add, False) if is_parent: add = GrampsWidgets.IconButton(self.select_family, None, gtk.STOCK_INDEX) 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(del_fcn, 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) if not family: return 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) if handle: 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) else: link_label = gtk.Label(_('Unknown')) link_label.set_alignment(0, 0.5) link_label.show() self.attach.attach(link_label, _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_ref_list() label = _("Children") if child_list: for child_ref in child_list: self.write_child(label, child_ref.ref) 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 select_family(self, obj, event, handle): if event.type == gtk.gdk.BUTTON_PRESS and event.button == 1: from SelectFamily import SelectFamily dialog = SelectFamily(self.dbstate.db, _('Select Family')) family = dialog.run() if family: person = self.dbstate.db.get_person_from_handle(self.dbstate.active.handle) GrampsDb.add_child_to_family(family, child, (RelLib.ChildRef.CHILD_BIRTH,'') (RelLib.ChildRef.CHILD_BIRTH,'')) 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: import GrampsDb GrampsDb.remove_parent_from_family(self.dbstate.db, self.dbstate.active.handle, handle) def delete_parent_family(self, obj, event, handle): if event.type == gtk.gdk.BUTTON_PRESS and event.button == 1: import GrampsDb GrampsDb.remove_child_from_family(self.dbstate.db, self.dbstate.active.handle, handle) def change_to(self, obj, handle): self.dbstate.change_active_handle(handle)