From bc0e50c8f088677bb0806ab7b2d8fdd9a6205f0e Mon Sep 17 00:00:00 2001 From: Don Allingham Date: Tue, 31 Aug 2004 03:51:05 +0000 Subject: [PATCH] * src/AddMedia.py: use new MediaList update routine * src/DisplayModels.py: new update list routine * src/ImageSelect.py: use new MediaList update routine * src/MediaView.py: use new MediaList update routine * src/PlaceView.py: new update list routine * src/SourceView.py: new update list routine svn: r3521 --- gramps2/ChangeLog | 8 ++++ gramps2/src/AddMedia.py | 2 +- gramps2/src/DisplayModels.py | 6 +++ gramps2/src/ImageSelect.py | 4 +- gramps2/src/MediaView.py | 75 +++++++++++++++++++++--------------- gramps2/src/PlaceView.py | 2 +- gramps2/src/SourceView.py | 2 +- 7 files changed, 62 insertions(+), 37 deletions(-) diff --git a/gramps2/ChangeLog b/gramps2/ChangeLog index 976a3e827..9f89b2ae6 100644 --- a/gramps2/ChangeLog +++ b/gramps2/ChangeLog @@ -1,3 +1,11 @@ +2004-08-30 Don Allingham + * src/AddMedia.py: use new MediaList update routine + * src/DisplayModels.py: new update list routine + * src/ImageSelect.py: use new MediaList update routine + * src/MediaView.py: use new MediaList update routine + * src/PlaceView.py: new update list routine + * src/SourceView.py: new update list routine + 2004-08-29 Don Allingham * src/GrampsBSDDB.py: add remove_place function * src/DisplayModels.py: added row deletion and updating diff --git a/gramps2/src/AddMedia.py b/gramps2/src/AddMedia.py index c76edca76..e90b73dec 100644 --- a/gramps2/src/AddMedia.py +++ b/gramps2/src/AddMedia.py @@ -134,7 +134,7 @@ class AddMediaObject: self.db.commit_media_object(mobj,trans) self.db.transaction_commit(trans,_("Add Media Object")) if self.update: - self.update() + self.update(mobj.get_handle()) def on_name_changed(self,*obj): """ diff --git a/gramps2/src/DisplayModels.py b/gramps2/src/DisplayModels.py index a65de18d4..1a476f2b8 100644 --- a/gramps2/src/DisplayModels.py +++ b/gramps2/src/DisplayModels.py @@ -59,6 +59,12 @@ class BaseModel(gtk.GenericTreeModel): del self.datalist[index] self.row_deleted(index) + def add_row_by_handle(self,handle): + self.datalist = self.sort_keys() + index = self.datalist.index(handle) + iter = self.get_iter(index) + self.row_inserted(index,iter) + def delete_row_by_handle(self,handle): index = self.datalist.index(handle) del self.datalist[index] diff --git a/gramps2/src/ImageSelect.py b/gramps2/src/ImageSelect.py index 0c0b659b7..0304716f5 100644 --- a/gramps2/src/ImageSelect.py +++ b/gramps2/src/ImageSelect.py @@ -1108,11 +1108,11 @@ class GlobalMediaProperties: if self.lists_changed: self.obj.set_attribute_list(self.alist) self.obj.set_source_reference_list(self.srcreflist) - if self.update != None: - self.update() trans = self.db.transaction_begin() self.db.commit_media_object(self.obj,trans) self.db.transaction_commit(trans,_("Edit Media Object")) + if self.update != None: + self.update(self.obj) def on_help_clicked(self, obj): """Display the relevant portion of GRAMPS manual""" diff --git a/gramps2/src/MediaView.py b/gramps2/src/MediaView.py index c18f71752..b3243c027 100644 --- a/gramps2/src/MediaView.py +++ b/gramps2/src/MediaView.py @@ -150,8 +150,8 @@ class MediaView: self.build_tree() def build_tree(self): - self.model = gtk.TreeModelSort(DisplayModels.MediaModel(self.parent.db)) - # self.model = DisplayModels.MediaModel(self.parent.db) + #self.model = gtk.TreeModelSort(DisplayModels.MediaModel(self.parent.db)) + self.model = DisplayModels.MediaModel(self.parent.db) self.list.set_model(self.model) self.selection = self.list.get_selection() @@ -258,21 +258,24 @@ class MediaView: name = RelImage.import_media_object(self.obj.get_path(),path,handle) if name: self.obj.set_path(name) - self.load_media() def popup_change_description(self, obj): - ImageSelect.GlobalMediaProperties(self.db,self.obj,self.load_media, - self,self.topWindow) - - def load_media(self): - pass + ImageSelect.GlobalMediaProperties(self.db,self.obj, + self.update_display, + self,self.topWindow) def on_add_clicked(self,obj): """Add a new media object to the media list""" import AddMedia - am = AddMedia.AddMediaObject(self.db,self.build_tree) + am = AddMedia.AddMediaObject(self.db,self.add_object) am.run() + def add_object(self,mobj_handle): + self.model.add_row_by_handle(mobj_handle) + + def update_display(self,mobj): + self.model.update_row_by_handle(mobj.get_handle()) + def on_edit_clicked(self,obj): """Edit the properties of an existing media object in the media list""" @@ -280,8 +283,9 @@ class MediaView: if node: handle = list_store.get_value(node,5) obj = self.db.get_object_from_handle(handle) - ImageSelect.GlobalMediaProperties(self.db,obj,self.load_media, - self,self.topWindow) + ImageSelect.GlobalMediaProperties(self.db,obj, + self.update_display, + self,self.topWindow) def on_delete_clicked(self,obj): store,node = self.selection.get_selected() @@ -291,7 +295,8 @@ class MediaView: handle = store.get_value(node,5) mobj = self.db.get_object_from_handle(handle) if self.is_object_used(mobj): - ans = ImageSelect.DeleteMediaQuery(mobj,self.db,self.build_tree) + ans = ImageSelect.DeleteMediaQuery(mobj,self.db, + self.model.delete_row_by_handle) QuestionDialog(_('Delete Media Object?'), _('This media object is currently being used. ' 'If you delete this object, it will be removed ' @@ -300,33 +305,39 @@ class MediaView: _('_Delete Media Object'), ans.query_response) else: - trans = self.db.transaction_begin() - self.db.remove_object(mobj.get_handle(),trans) - self.db.transaction_commit(trans,_("Remove Media Object")) - self.build_tree() + self.delete_object(mobj) + + def delete_object(self,media_obj): + trans = self.parent.db.transaction_begin() + mobj_handle = media_obj.get_handle() + self.parent.db.remove_object(mobj_handle,trans) + title_msg = _("Delete Media Object?") + self.parent.db.transaction_commit(trans,title_msg) + self.model.delete_row_by_handle(mobj_handle) def is_object_used(self,mobj): - for family_handle in self.db.get_family_handles(): + mhandle = mobj.get_handle() + for family_handle in self.db.get_family_handles(sort_handles=False): p = self.db.get_family_from_handle(family_handle) for o in p.get_media_list(): - if o.get_reference_handle() == mobj.get_handle(): - return 1 + if o.get_reference_handle() == mhandle: + return True for key in self.db.get_person_handles(sort_handles=False): p = self.db.get_person_from_handle(key) for o in p.get_media_list(): - if o.get_reference_handle() == mobj.get_handle(): - return 1 - for key in self.db.get_source_handles(): + if o.get_reference_handle() == mhandle: + return True + for key in self.db.get_source_handles(sort_handles=False): p = self.db.get_source_from_handle(key) for o in p.get_media_list(): - if o.get_reference_handle() == mobj.get_handle(): - return 1 - for key in self.db.get_place_handles(): - p = self.db.get_place_handle(key) + if o.get_reference_handle() == mhandle: + return True + for key in self.db.get_place_handles(sort_handles=False): + p = self.db.get_place_from_handle(key) for o in p.get_media_list(): - if o.get_reference_handle() == mobj.get_handle(): - return 1 - return 0 + if o.get_reference_handle() == mhandle: + return True + return False def on_drag_begin(self,obj,context): store,node = self.selection.get_selected() @@ -363,7 +374,6 @@ class MediaView: photo.set_description(description) trans = self.db.transaction_begin() self.db.add_object(photo,trans) - self.load_media() if GrampsCfg.get_media_reference() == 0: name = RelImage.import_media_object(name, self.db.get_save_path(), @@ -375,8 +385,9 @@ class MediaView: self.db.transaction_commit(trans,_("Add Media Object")) if GrampsCfg.get_media_global(): - ImageSelect.GlobalMediaProperties(self.db,photo,self.load_media, - self,self.topWindow) + ImageSelect.GlobalMediaProperties(self.db,photo, + self.update_display, + self,self.topWindow) elif protocol != "": import urllib u = urllib.URLopener() diff --git a/gramps2/src/PlaceView.py b/gramps2/src/PlaceView.py index 8db35c8b7..b650df99d 100644 --- a/gramps2/src/PlaceView.py +++ b/gramps2/src/PlaceView.py @@ -173,7 +173,7 @@ class PlaceView: EditPlace.EditPlace(self.parent,RelLib.Place(),self.new_place_after_edit) def new_place_after_edit(self,place): - self.build_tree() + self.model.add_row_by_handle(place.get_handle()) def update_display(self,place): self.model.update_row_by_handle(place.get_handle()) diff --git a/gramps2/src/SourceView.py b/gramps2/src/SourceView.py index cba23bf37..eaedbdca7 100644 --- a/gramps2/src/SourceView.py +++ b/gramps2/src/SourceView.py @@ -249,7 +249,7 @@ class SourceView: self.topWindow, self.update_display) def new_after_edit(self,source): - self.build_tree() + self.model.add_row_by_handle(source.get_handle()) def update_display(self,source): self.model.update_row_by_handle(source.get_handle())