From 411367eb55747f7385502bd9477510000e60a502 Mon Sep 17 00:00:00 2001 From: Thiago Franco de Moraes Date: Mon, 15 May 2017 14:57:40 -0300 Subject: [PATCH] Checking surface index exists in actors_dict before removing it --- invesalius/data/surface.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/invesalius/data/surface.py b/invesalius/data/surface.py index cdf87a8..b08be6f 100644 --- a/invesalius/data/surface.py +++ b/invesalius/data/surface.py @@ -180,14 +180,15 @@ class SurfaceManager(): if selected_items: for index in selected_items: proj.RemoveSurface(index) - actor = old_dict[index] - for i in old_dict: - if i < index: - new_dict[i] = old_dict[i] - if i > index: - new_dict[i-1] = old_dict[i] - old_dict = new_dict - Publisher.sendMessage('Remove surface actor from viewer', actor) + if index in old_dict: + actor = old_dict[index] + for i in old_dict: + if i < index: + new_dict[i] = old_dict[i] + if i > index: + new_dict[i-1] = old_dict[i] + old_dict = new_dict + Publisher.sendMessage('Remove surface actor from viewer', actor) self.actors_dict = new_dict if self.last_surface_index in selected_items: -- libgit2 0.21.2