Minor fixes to avoid crashes and instantiation errors.

freedomofpress · Oct 23, 2018 · e35fa81 · e35fa81
1 parent 9f78c31
commit e35fa81
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/securedrop_client/gui/main.py b/securedrop_client/gui/main.py
@@ -70,6 +70,7 @@ def setup(self, controller):
         """
         self.controller = controller  # Reference the Client logic instance.
         self.tool_bar.setup(self, controller)
+        self.main_view.setup(controller)
 
     def autosize_window(self):
         """
@@ -143,7 +144,8 @@ def on_source_changed(self):
         """
         source_item = self.main_view.source_list.currentItem()
         source_widget = self.main_view.source_list.itemWidget(source_item)
-        self.show_conversation_for(source_widget.source)
+        if source_widget:
+            self.show_conversation_for(source_widget.source)
 
     def show_conversation_for(self, source):
         """