# # Gramps - a GTK+/GNOME based genealogy program # # Copyright (C) 2000-2006 Donald N. Allingham # Copyright (C) 2009 Gary Burton # Copyright (C) 2011 Tim G L Lyons # # 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(".") LOG = logging.getLogger(".citation") #------------------------------------------------------------------------- # # GTK/Gnome modules # #------------------------------------------------------------------------- import gtk #------------------------------------------------------------------------- # # gramps modules # #------------------------------------------------------------------------- import gen.lib from gen.db import DbTxn from editprimary import EditPrimary from displaytabs import (NoteTab, GalleryTab, DataEmbedList, CitationBackRefList, 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 = CitationBackRefList(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 with DbTxn('', self.db) as trans: 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() trans.set_description(msg) self.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): with DbTxn(_("Delete Source (%s)") % self.source.get_title(), self.db) as trans: self.db.disable_signals() # we can have: # object(CitationBase) -> Citation(source_handle) -> Source # We first have to remove the CitationBase references to the # Citation. Then we remove the Citations. (We don't need to # remove the source_handle references to the Source, because we are # removing the whole Citation). Then we can remove the Source (citation_list, citation_referents_list) = self.the_lists # citation_list is a tuple of lists. Only the first, for Citations, # exists. citation_list = citation_list[0] # (1) delete the references to the citation for (citation_handle, refs) in citation_referents_list: LOG.debug('delete citation %s references %s' % (citation_handle, refs)) (person_list, family_list, event_list, place_list, source_list, media_list, repo_list) = refs ctn_handle_list = [citation_handle] for handle in person_list: person = self.db.get_person_from_handle(handle) person.remove_citation_references(ctn_handle_list) self.db.commit_person(person, trans) for handle in family_list: family = self.db.get_family_from_handle(handle) family.remove_citation_references(ctn_handle_list) self.db.commit_family(family, trans) for handle in event_list: event = self.db.get_event_from_handle(handle) event.remove_citation_references(ctn_handle_list) self.db.commit_event(event, trans) for handle in place_list: place = self.db.get_place_from_handle(handle) place.remove_citation_references(ctn_handle_list) self.db.commit_place(place, trans) for handle in source_list: source = self.db.get_source_from_handle(handle) source.remove_citation_references(ctn_handle_list) self.db.commit_source(source, trans) for handle in media_list: media = self.db.get_object_from_handle(handle) media.remove_citation_references(ctn_handle_list) self.db.commit_media_object(media, trans) for handle in repo_list: repo = self.db.get_repository_from_handle(handle) repo.remove_citation_references(ctn_handle_list) self.db.commit_repository(repo, trans) # (2) delete the actual citations LOG.debug('remove the actual citations %s' % citation_list) for citation_handle in citation_list: LOG.debug("remove_citation %s" % citation_handle) self.db.remove_citation(citation_handle, trans) # (3) delete the source self.db.enable_signals() self.db.remove_source(self.source.get_handle(), trans)