gramps/gramps2/src/EditSource.py
Don Allingham ee78b4394f Database saving
svn: r2927
2004-02-28 20:15:48 +00:00

402 lines
15 KiB
Python

#
# 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, 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.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()
f = self.db.find_person_from_id(f_id)
m = self.db.find_person_from_id(m_id)
if f and m:
name = _("%(father)s and %(mother)s") % {
"father" : GrampsCfg.nameof(f),
"mother" : GrampsCfg.nameof(m)}
elif f:
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
self.db.commit_source(self.source)
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)
def query_response(self):
self.db.remove_source_id(self.source.get_id())
for key in self.db.get_person_keys():
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:
self.delete_source(v)
for v in p.get_attribute_list():
self.delete_source(v)
for v in p.get_alternate_names() + [p.get_primary_name()]:
self.delete_source(v)
for v in p.get_address_list():
self.delete_source(v)
for p_id in self.db.get_family_keys():
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:
self.delete_source(v)
for v in p.get_attribute_list():
self.delete_source(v)
for p_id in self.db.get_object_keys():
p = self.db.find_object_from_id(p_id)
self.delete_source(p)
for key in self.db.get_place_id_keys():
p = self.db.find_place_from_id(key)
self.delete_source(self.db.find_place_from_id(key))
self.update(0)