291 lines
12 KiB
Python
291 lines
12 KiB
Python
#
|
|
# 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 gramps.gen.ggettext import gettext as _
|
|
import logging
|
|
log = logging.getLogger(".")
|
|
LOG = logging.getLogger(".citation")
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# GTK/Gnome modules
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
from gi.repository import Gtk
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# gramps modules
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
from gramps.gen.lib import NoteType, Source
|
|
from gramps.gen.db import DbTxn
|
|
from editprimary import EditPrimary
|
|
|
|
from displaytabs import (NoteTab, GalleryTab, DataEmbedList,
|
|
CitationBackRefList, RepoEmbedList)
|
|
from ..widgets import MonitoredEntry, PrivacyButton
|
|
from ..dialog 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 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(),
|
|
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, True, 0)
|
|
|
|
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)
|