# # Gramps - a GTK+/GNOME based genealogy program # # Copyright (C) 2000-2004 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$ #------------------------------------------------------------------------- # # GTK/Gnome modules # #------------------------------------------------------------------------- import gtk.glade import gnome #------------------------------------------------------------------------- # # gramps modules # #------------------------------------------------------------------------- import const import Utils import GrampsCfg import ImageSelect import ListModel from gettext import gettext as _ #------------------------------------------------------------------------- # # Constants # #------------------------------------------------------------------------- class EditSource: def __init__(self,source,db,parent,parent_window=None,func=None): self.source = source self.db = db self.parent = parent if source: if self.parent.child_windows.has_key(source.get_id()): self.parent.child_windows[source.get_id()].present(None) return else: self.win_key = source.get_id() else: self.win_key = self self.child_windows = {} self.callback = func self.path = db.get_save_path() self.not_loaded = 1 self.ref_not_loaded = 1 self.lists_changed = 0 self.gallery_ok = 0 self.top_window = gtk.glade.XML(const.gladeFile,"sourceEditor","gramps") self.top = self.top_window.get_widget("sourceEditor") Utils.set_titles(self.top,self.top_window.get_widget('title'), _('Source Editor')) plwidget = self.top_window.get_widget("iconlist") self.gallery = ImageSelect.Gallery(source, db.commit_place, self.path, plwidget, db, self, self.top) self.author = self.top_window.get_widget("author") self.pubinfo = self.top_window.get_widget("pubinfo") self.abbrev = self.top_window.get_widget("abbrev") self.note = self.top_window.get_widget("source_note") self.notes_buffer = self.note.get_buffer() self.gallery_label = self.top_window.get_widget("gallerySourceEditor") self.refs_label = self.top_window.get_widget("refsSourceEditor") self.notes_label = self.top_window.get_widget("notesSourceEditor") self.flowed = self.top_window.get_widget("source_flowed") self.preform = self.top_window.get_widget("source_preform") self.refinfo = self.top_window.get_widget("refinfo") self.title = self.top_window.get_widget("source_title") self.title.set_text(source.get_title()) self.author.set_text(source.get_author()) self.pubinfo.set_text(source.get_publication_info()) self.abbrev.set_text(source.get_abbreviation()) if source.get_note(): self.notes_buffer.set_text(source.get_note()) Utils.bold_label(self.notes_label) if source.get_note_format() == 1: self.preform.set_active(1) else: self.flowed.set_active(1) if self.source.get_media_list(): Utils.bold_label(self.gallery_label) self.top_window.signal_autoconnect({ "on_switch_page" : self.on_switch_page, "on_addphoto_clicked" : self.gallery.on_add_media_clicked, "on_selectphoto_clicked" : self.gallery.on_select_media_clicked, "on_deletephoto_clicked" : self.gallery.on_delete_media_clicked, "on_editphoto_clicked" : self.gallery.on_edit_media_clicked, "on_edit_properties_clicked": self.gallery.popup_change_description, "on_sourceEditor_help_clicked" : self.on_help_clicked, "on_sourceEditor_ok_clicked" : self.on_source_apply_clicked, "on_sourceEditor_cancel_clicked" : self.close, "on_sourceEditor_delete_event" : self.on_delete_event, }) if self.source.get_id() == "": self.top_window.get_widget("edit_photo").set_sensitive(0) self.top_window.get_widget("delete_photo").set_sensitive(0) if parent_window: self.top.set_transient_for(parent_window) self.display_references() if parent_window: self.top.set_transient_for(parent_window) self.add_itself_to_menu() self.top.show() def on_delete_event(self,obj,b): self.close_child_windows() self.remove_itself_from_menu() def on_help_clicked(self,obj): """Display the relevant portion of GRAMPS manual""" gnome.help_display('gramps-manual','gramps-edit-complete') def close(self,obj): self.gallery.close(self.gallery_ok) self.close_child_windows() self.remove_itself_from_menu() self.top.destroy() def close_child_windows(self): for child_window in self.child_windows.values(): child_window.close(None) self.child_windows = {} def add_itself_to_menu(self): self.parent.child_windows[self.win_key] = self if not self.source: label = _("New Source") else: label = self.source.get_title() if not label.strip(): label = _("New Source") label = "%s: %s" % (_('Source'),label) self.parent_menu_item = gtk.MenuItem(label) self.parent_menu_item.set_submenu(gtk.Menu()) self.parent_menu_item.show() self.parent.winsmenu.append(self.parent_menu_item) self.winsmenu = self.parent_menu_item.get_submenu() self.menu_item = gtk.MenuItem(_('Source Editor')) self.menu_item.connect("activate",self.present) self.menu_item.show() self.winsmenu.append(self.menu_item) def remove_itself_from_menu(self): del self.parent.child_windows[self.win_key] self.menu_item.destroy() self.winsmenu.destroy() self.parent_menu_item.destroy() def present(self,obj): self.top.present() def display_references(self): p_event_list = [] p_attr_list = [] p_addr_list = [] p_name_list = [] m_list = [] f_event_list = [] f_attr_list = [] p_list = [] for key in self.db.get_place_id_keys(): p = self.db.get_place_id(key) name = p.get_title() for sref in p.get_source_references(): if sref.get_base_id() == self.source.get_id(): p_list.append(name) for key in self.db.get_person_keys(): p = self.db.get_person(key) name = GrampsCfg.nameof(p) for event_id in p.get_event_list() + [p.get_birth_id(), p.get_death_id()]: if event_id: event = self.db.find_event_from_id(event_id) for sref in event.get_source_references(): if sref.get_base_id() == self.source.get_id(): p_event_list.append((name,event.get_name())) for v in p.get_attribute_list(): for sref in v.get_source_references(): if sref.get_base_id() == self.source.get_id(): p_attr_list.append((name,v.get_type())) for v in p.get_alternate_names() + [p.get_primary_name()]: for sref in v.get_source_references(): if sref.get_base_id() == self.source.get_id(): p_name_list.append((name,v.get_name())) for v in p.get_address_list(): for sref in v.get_source_references(): if sref.get_base_id() == self.source.get_id(): p_addr_list.append((name,v.get_street())) for object_id in self.db.get_object_keys(): object = self.db.try_to_find_object_from_id(object_id) name = object.get_description() for sref in object.get_source_references(): if sref.get_base_id() == self.source.get_id(): m_list.append(name) for family_id in self.db.get_family_keys(): family = self.db.find_family_from_id(family_id) f_id = family.get_father_id() m_id = family.get_mother_id() if f_id: f = self.db.try_to_find_person_from_id(f_id) if m_id: m = self.db.try_to_find_person_from_id(m_id) if f_id and m_id: name = _("%(father)s and %(mother)s") % { "father" : GrampsCfg.nameof(f), "mother" : GrampsCfg.nameof(m)} elif f_id: name = GrampsCfg.nameof(f) else: name = GrampsCfg.nameof(m) for v_id in p.get_event_list(): v = self.db.find_event_from_id(v_id) if not v: continue for sref in v.get_source_references(): if sref.get_base_id() == self.source.get_id(): f_event_list.append((name,v.get_name())) for v in p.get_attribute_list(): for sref in v.get_source_references(): if sref.get_base_id() == self.source.get_id(): f_attr_list.append((name,v.get_type())) slist = self.top_window.get_widget('slist') titles = [(_('Source Type'),0,150),(_('Object'),1,150),(_('Value'),2,150)] self.model = ListModel.ListModel(slist,titles) any = 0 if len(p_event_list) > 0: any = 1 for p in p_event_list: self.model.add([_("Individual Events"),p[0], const.display_pevent(p[1])]) if len(p_attr_list) > 0: any = 1 for p in p_attr_list: self.model.add([_("Individual Attributes"),p[0], const.display_pattr(p[1])]) if len(p_name_list) > 0: any = 1 for p in p_name_list: self.model.add([_("Individual Names"),p[0],p[1]]) if len(f_event_list) > 0: any = 1 for p in f_event_list: self.model.add([_("Family Events"),p[0], const.display_fevent(p[1])]) if len(f_attr_list) > 0: any = 1 for p in f_event_list: self.model.add([_("Family Attributes"),p[0], const.display_fattr(p[1])]) if len(m_list) > 0: any = 1 for p in m_list: self.model.add([_("Media Objects"),p,'']) if len(p_list) > 0: any = 1 for p in p_list: self.model.add([_("Places"),p,'']) if any: Utils.bold_label(self.refs_label) else: Utils.unbold_label(self.refs_label) self.ref_not_loaded = 0 def on_source_apply_clicked(self,obj): title = unicode(self.title.get_text()) author = unicode(self.author.get_text()) pubinfo = unicode(self.pubinfo.get_text()) abbrev = unicode(self.abbrev.get_text()) note = unicode(self.notes_buffer.get_text(self.notes_buffer.get_start_iter(), self.notes_buffer.get_end_iter(),gtk.FALSE)) format = self.preform.get_active() if author != self.source.get_author(): self.source.set_author(author) if title != self.source.get_title(): self.source.set_title(title) if pubinfo != self.source.get_publication_info(): self.source.set_publication_info(pubinfo) if abbrev != self.source.get_abbreviation(): self.source.set_abbreviation(abbrev) if note != self.source.get_note(): self.source.set_note(note) if format != self.source.get_note_format(): self.source.set_note_format(format) self.gallery_ok = 1 trans = self.db.start_transaction() self.db.commit_source(self.source,trans) self.db.add_transaction(trans,_("Edit Source (%s)") % title) if self.callback: self.callback(self.source) self.close(obj) def on_switch_page(self,obj,a,page): if page == 2 and self.not_loaded: self.not_loaded = 0 self.gallery.load_images() elif page == 3 and self.ref_not_loaded: self.ref_not_loaded = 0 self.display_references() text = unicode(self.notes_buffer.get_text(self.notes_buffer.get_start_iter(), self.notes_buffer.get_end_iter(),gtk.FALSE)) if text: Utils.bold_label(self.notes_label) else: Utils.unbold_label(self.notes_label) class DelSrcQuery: def __init__(self,source,db,update): self.source = source self.db = db self.update = update def delete_source(self,object): m = 0 l = [] for sref in object.get_source_references(): if sref.get_base_id() != self.source.get_id(): l.append(sref) else: m = 1 if m: object.set_source_reference_list(l) return m def query_response(self): trans = self.db.start_transaction() for key in self.db.get_person_keys(): commit = 0 p = self.db.get_person(key) for v_id in p.get_event_list() + [p.get_birth_id(), p.get_death_id()]: v = self.db.find_event_from_id(v_id) if v: commit += self.delete_source(v) for v in p.get_attribute_list(): commit += self.delete_source(v) for v in p.get_alternate_names() + [p.get_primary_name()]: commit += self.delete_source(v) for v in p.get_address_list(): commit += self.delete_source(v) if commit > 0: self.db.commit_person(p,trans) for p_id in self.db.get_family_keys(): commit = 0 p = self.db.find_family_from_id(p_id) for v_id in p.get_event_list(): v = self.db.find_event_from_id(v_id) if v: commit += self.delete_source(v) for v in p.get_attribute_list(): commit += self.delete_source(v) if commit > 0: self.db.commit_family(p,trans) for p_id in self.db.get_object_keys(): p = self.db.try_to_find_object_from_id(p_id,trans) if self.delete_source(p): self.db.commit_media_object(p,trans) for key in self.db.get_place_id_keys(): p = self.db.try_to_find_place_from_id(key) if self.delete_source(self.db.try_to_find_place_from_id(key)): self.db.commit_place(p,trans) self.db.remove_source_id(self.source.get_id(),trans) self.db.add_transaction(trans,_("Delete Source (%s)") % self.source.get_title()) self.update()