# # Gramps - a GTK+/GNOME based genealogy program # # Copyright (C) 2000-2006 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$ #------------------------------------------------------------------------- # # Python modules # #------------------------------------------------------------------------- from gettext import gettext as _ try: set() except: print "set import failed" from sets import Set as set #------------------------------------------------------------------------- # # GTK/Gnome modules # #------------------------------------------------------------------------- import gtk import gtk.glade #------------------------------------------------------------------------- # # gramps modules # #------------------------------------------------------------------------- import const import Utils import AutoComp import RelLib from DateHandler import displayer as _dd import DateEdit import DisplayState from QuestionDialog import WarningDialog, ErrorDialog from WindowUtils import GladeIf from DisplayTabs import * from GrampsWidgets import * #------------------------------------------------------------------------- # # Constants # #------------------------------------------------------------------------- total_events = dict(Utils.personal_events) for event_type in Utils.family_events.keys(): if not total_events.has_key(event_type): total_events[event_type] = Utils.family_events[event_type] #------------------------------------------------------------------------- # # helper function # #------------------------------------------------------------------------- def get_place(field,pmap,db): text = unicode(field.get_text().strip()) if text: if pmap.has_key(text): return db.get_place_from_handle(pmap[text]) else: place = RelLib.Place() place.set_title(text) trans = db.transaction_begin() db.add_place(place,trans) db.transaction_commit(trans,_("Add Place")) return place else: return None #------------------------------------------------------------------------- # # EditEventRef class # #------------------------------------------------------------------------- class EditEventRef(DisplayState.ManagedWindow): def __init__(self, state, uistate, track, event, event_ref, referent, update): self.db = state.db self.state = state self.uistate = uistate self.referent = referent self.event_ref = event_ref self.event = event DisplayState.ManagedWindow.__init__(self, uistate, track, event_ref) if self.already_exist: return self.update = update self.pmap = {} for key in self.db.get_place_handles(): title = self.db.get_place_from_handle(key).get_title() self.pmap[title] = key self.title = _('Event Reference Editor') self.top = gtk.glade.XML(const.gladeFile, "event_eref_edit","gramps") self.window = self.top.get_widget('event_eref_edit') self.ref_note_field = self.top.get_widget('eer_ref_note') self.role_combo = self.top.get_widget('eer_role_combo') self.date_field = self.top.get_widget("eer_date") self.place_field = self.top.get_widget("eer_place") self.cause_field = self.top.get_widget("eer_cause") self.ev_note_field = self.top.get_widget("eer_ev_note") self.type_combo = self.top.get_widget("eer_type_combo") self.general_label = self.top.get_widget("eer_general_tab") self.ok = self.top.get_widget('ok') self.expander = self.top.get_widget("eer_expander") self.warning = self.top.get_widget("eer_warning") self.notebook = self.top.get_widget('notebook') if self.referent.__class__.__name__ == 'Person': default_type = RelLib.Event.BIRTH default_role = RelLib.EventRef.PRIMARY ev_dict = Utils.personal_events role_dict = Utils.event_roles elif self.referent.__class__.__name__ == 'Family': default_type = RelLib.Event.MARRIAGE default_role = RelLib.EventRef.FAMILY ev_dict = Utils.family_events role_dict = Utils.family_event_roles if self.event: self.event_added = False self.date = RelLib.Date(self.event.get_date_object()) if self.event_ref: if self.event_ref.get_role()[0] == default_role: self.expander.set_expanded(True) self.warning.hide() else: self.expander.set_expanded(False) self.warning.show_all() else: self.event = RelLib.Event() self.event.set_type((default_type,ev_dict[default_type])) self.event.set_handle(self.db.create_id()) self.event.set_gramps_id(self.db.find_next_event_gramps_id()) self.event_added = True self.date = RelLib.Date(None) self.expander.set_expanded(True) self.warning.hide() if not self.event_ref: self.event_ref = RelLib.EventRef() self.event_ref.set_role((default_role,role_dict[default_role])) self.event_ref.set_reference_handle(self.event.get_handle()) self.cause_monitor = MonitoredEntry(self.cause_field,self.event.set_cause, self.event.get_cause, False) self.ref_privacy = PrivacyButton(self.top.get_widget('eer_ref_priv'), self.event_ref) self.descr_field = MonitoredEntry(self.top.get_widget("eer_description"), self.event.set_description, self.event.get_description, False) self.ev_privacy = PrivacyButton(self.top.get_widget("eer_ev_priv"), self.event) Utils.set_titles(self.window, self.top.get_widget('eer_title'), self.title) self.top.signal_autoconnect({ "on_eer_help_clicked" : self.on_help_clicked, "on_eer_ok_clicked" : self.on_ok_clicked, "on_eer_cancel_clicked" : self.close, "on_eer_delete_event" : self.close, }) self.role_selector = AutoComp.StandardCustomSelector( role_dict,self.role_combo, RelLib.EventRef.CUSTOM,default_role) AutoComp.fill_entry(self.place_field,self.pmap.keys()) self.type_selector = AutoComp.StandardCustomSelector( ev_dict,self.type_combo, RelLib.Event.CUSTOM,default_type) self.date_check = DateEdit.DateEdit(self.date, self.date_field, self.top.get_widget("eer_date_stat"), self.window) # set event_ref values self.role_selector.set_values(self.event_ref.get_role()) self.ref_note_field.get_buffer().set_text(self.event_ref.get_note()) # set event values self.type_selector.set_values(self.event.get_type()) place_handle = self.event.get_place_handle() if not place_handle: place_name = u"" else: place_name = self.db.get_place_from_handle(place_handle).get_title() self.place_field.set_text(place_name) self.date_field.set_text(_dd.display(self.date)) self._create_tabbed_pages() self.show() def _add_page(self,page): self.notebook.insert_page(page) self.notebook.set_tab_label(page,page.get_tab_widget()) return page def _create_tabbed_pages(self): """ Creates the notebook tabs and inserts them into the main window. """ self.srcref_list = self._add_page(SourceEmbedList( self.state,self.uistate, self.track, self.event.source_list)) self.note_tab = self._add_page(NoteTab( self.state, self.uistate, self.track, self.event.get_note_object())) self.gallery_tab = self._add_page(GalleryTab( self.state, self.uistate, self.track, self.event.get_media_list())) def build_menu_names(self,eventref): if self.event: if self.event.get_type()[0] == RelLib.Event.CUSTOM: event_name = self.event.get_type()[1] else: try: event_name = Utils.personal_events[self.event.get_type()[0]] except: event_name = Utils.family_events[self.event.get_type()[0]] submenu_label = _('Event: %s') % event_name else: submenu_label = _('New Event') return (_('Event Reference Editor'),submenu_label) def build_window_key(self,eventref): if self.event: return self.event.get_handle() else: return id(self) def on_help_clicked(self,obj): pass def on_ok_clicked(self,obj): # first, save event if changed etype = self.type_selector.get_values() eplace_obj = get_place(self.place_field,self.pmap,self.db) self.update_event(etype,self.date,eplace_obj) trans = self.db.transaction_begin() self.db.commit_event(self.event,trans) if self.event_added: self.db.transaction_commit(trans,_("Add Event")) else: self.db.transaction_commit(trans,_("Modify Event")) # then, set properties of the event_ref self.event_ref.set_role(self.role_selector.get_values()) self.event_ref.set_privacy(self.ref_privacy.get_active()) self.close(None) if self.update: self.update((self.event_ref,self.event)) def update_event(self,the_type,date,place): if place: if self.event.get_place_handle() != place.get_handle(): self.event.set_place_handle(place.get_handle()) else: if self.event.get_place_handle(): self.event.set_place_handle("") if self.event.get_type() != the_type: self.event.set_type(the_type) dobj = self.event.get_date_object() if not dobj.is_equal(date): self.event.set_date_object(date) #------------------------------------------------------------------------- # # Delete Query class # #------------------------------------------------------------------------- class DelEventQuery: def __init__(self,event,db,person_list,family_list): self.event = event self.db = db self.person_list = person_list self.family_list = family_list def query_response(self): trans = self.db.transaction_begin() self.db.disable_signals() ev_handle_list = [self.event.get_handle()] for handle in self.person_list: person = self.db.get_person_from_handle(handle) person.remove_handle_references('Event',ev_handle_list) self.db.commit_person(person,trans) for handle in self.family_list: family = self.db.get_family_from_handle(handle) family.remove_handle_references('Event',ev_handle_list) self.db.commit_family(family,trans) self.db.enable_signals() self.db.remove_event(self.event.get_handle(),trans) self.db.transaction_commit( trans,_("Delete Event (%s)") % self.event.get_gramps_id())