# # Gramps - a GTK+/GNOME based genealogy program # # Copyright (C) 2000-2006 Donald N. Allingham # Copyright (C) 2009 Gary Burton # # 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$ #------------------------------------------------------------------------- # # Python modules # #------------------------------------------------------------------------- from gen.ggettext import gettext as _ import logging log = logging.getLogger(".") #------------------------------------------------------------------------- # # GTK/Gnome modules # #------------------------------------------------------------------------- import gtk #------------------------------------------------------------------------- # # gramps modules # #------------------------------------------------------------------------- import gen.lib from editprimary import EditPrimary from displaytabs import (NoteTab, GalleryTab, DataEmbedList, SourceBackRefList, RepoEmbedList) from gui.widgets import MonitoredEntry, PrivacyButton from QuestionDialog import ErrorDialog from glade import Glade #------------------------------------------------------------------------- # # EditSource class # #------------------------------------------------------------------------- class EditSource(EditPrimary): def __init__(self, dbstate, uistate, track, source): EditPrimary.__init__(self, dbstate, uistate, track, source, dbstate.db.get_source_from_handle, dbstate.db.get_source_from_gramps_id) def empty_object(self): return gen.lib.Source() def get_menu_title(self): title = self.obj.get_title() if title: title = _('Source') + ": " + title else: title = _('New Source') return title def _local_init(self): self.width_key = 'interface.source-width' self.height_key = 'interface.source-height' assert(self.obj) self.glade = Glade() self.set_window(self.glade.toplevel, None, self.get_menu_title()) def _connect_signals(self): self.define_ok_button(self.glade.get_object('ok'),self.save) self.define_cancel_button(self.glade.get_object('cancel')) self.define_help_button(self.glade.get_object('help')) def _connect_db_signals(self): """ Connect any signals that need to be connected. Called by the init routine of the base class (_EditPrimary). """ self._add_db_signal('source-rebuild', self._do_close) self._add_db_signal('source-delete', self.check_for_close) def _setup_fields(self): self.author = MonitoredEntry(self.glade.get_object("author"), self.obj.set_author, self.obj.get_author, self.db.readonly) self.pubinfo = MonitoredEntry(self.glade.get_object("pubinfo"), self.obj.set_publication_info, self.obj.get_publication_info, self.db.readonly) self.gid = MonitoredEntry(self.glade.get_object("gid"), self.obj.set_gramps_id, self.obj.get_gramps_id, self.db.readonly) self.priv = PrivacyButton(self.glade.get_object("private"), self.obj, self.db.readonly) self.abbrev = MonitoredEntry(self.glade.get_object("abbrev"), self.obj.set_abbreviation, self.obj.get_abbreviation, self.db.readonly) self.title = MonitoredEntry(self.glade.get_object("source_title"), self.obj.set_title, self.obj.get_title, self.db.readonly) def _create_tabbed_pages(self): notebook = gtk.Notebook() self.note_tab = NoteTab(self.dbstate, self.uistate, self.track, self.obj.get_note_list(), self.get_menu_title(), gen.lib.NoteType.SOURCE) self._add_tab(notebook, self.note_tab) self.track_ref_for_deletion("note_tab") self.gallery_tab = GalleryTab(self.dbstate, self.uistate, self.track, self.obj.get_media_list()) self._add_tab(notebook, self.gallery_tab) self.track_ref_for_deletion("gallery_tab") self.data_tab = DataEmbedList(self.dbstate, self.uistate, self.track, self.obj) self._add_tab(notebook, self.data_tab) self.track_ref_for_deletion("data_tab") self.repo_tab = RepoEmbedList(self.dbstate, self.uistate, self.track, self.obj.get_reporef_list()) self._add_tab(notebook, self.repo_tab) self.track_ref_for_deletion("repo_tab") self.backref_list = SourceBackRefList(self.dbstate, self.uistate, self.track, self.db.find_backlink_handles(self.obj.handle)) self.backref_tab = self._add_tab(notebook, self.backref_list) self.track_ref_for_deletion("backref_tab") self.track_ref_for_deletion("backref_list") self._setup_notebook_tabs(notebook) notebook.show_all() self.glade.get_object('vbox').pack_start(notebook, True) def build_menu_names(self, source): return (_('Edit Source'), self.get_menu_title()) def save(self, *obj): self.ok_button.set_sensitive(False) if self.object_is_empty(): ErrorDialog(_("Cannot save source"), _("No data exists for this source. Please " "enter data or cancel the edit.")) self.ok_button.set_sensitive(True) return (uses_dupe_id, id) = self._uses_duplicate_id() if uses_dupe_id: prim_object = self.get_from_gramps_id(id) name = prim_object.get_title() msg1 = _("Cannot save source. ID already exists.") msg2 = _("You have attempted to use the existing Gramps ID with " "value %(id)s. This value is already used by '" "%(prim_object)s'. Please enter a different ID or leave " "blank to get the next available ID value.") % { 'id' : id, 'prim_object' : name } ErrorDialog(msg1, msg2) self.ok_button.set_sensitive(True) return trans = self.db.transaction_begin() if not self.obj.get_handle(): self.db.add_source(self.obj, trans) msg = _("Add Source (%s)") % self.obj.get_title() else: if not self.obj.get_gramps_id(): self.obj.set_gramps_id(self.db.find_next_source_gramps_id()) self.db.commit_source(self.obj, trans) msg = _("Edit Source (%s)") % self.obj.get_title() self.db.transaction_commit(trans, msg) self.close() def _cleanup_on_exit(self): self.backref_tab.close() class DeleteSrcQuery(object): def __init__(self, dbstate, uistate, source, the_lists): self.source = source self.db = dbstate.db self.uistate = uistate self.the_lists = the_lists def query_response(self): trans = self.db.transaction_begin() self.db.disable_signals() (person_list, family_list, event_list, place_list, source_list, media_list, repo_list) = self.the_lists src_handle_list = [self.source.get_handle()] for handle in person_list: person = self.db.get_person_from_handle(handle) person.remove_source_references(src_handle_list) self.db.commit_person(person, trans) for handle in family_list: family = self.db.get_family_from_handle(handle) family.remove_source_references(src_handle_list) self.db.commit_family(family, trans) for handle in event_list: event = self.db.get_event_from_handle(handle) event.remove_source_references(src_handle_list) self.db.commit_event(event, trans) for handle in place_list: place = self.db.get_place_from_handle(handle) place.remove_source_references(src_handle_list) self.db.commit_place(place, trans) for handle in source_list: source = self.db.get_source_from_handle(handle) source.remove_source_references(src_handle_list) self.db.commit_source(source, trans) for handle in media_list: media = self.db.get_object_from_handle(handle) media.remove_source_references(src_handle_list) self.db.commit_media_object(media, trans) for handle in repo_list: repo = self.db.get_repository_from_handle(handle) repo.remove_source_references(src_handle_list) self.db.commit_repository(repo, trans) self.db.enable_signals() self.db.remove_source(self.source.get_handle(), trans) self.db.transaction_commit( trans, _("Delete Source (%s)") % self.source.get_title())