Added remove_plugin method for DockMixin

Author: Chilipp

psyplot_gui/common.py

@@ -133,6 +133,27 @@ def create_view_action(self, main, docktype='pane'):
                 main.dataframe_menu.addAction(action)
         return self._view_action
 
+    def remove_plugin(self):
+        """Remove this plugin and close it"""
+        mainwindow = self.dock.parent() if self.dock else self.parent()
+        key = next((key for key, w in mainwindow.plugins.items()
+                    if w is self), None)
+        if mainwindow.centralWidget() is self:
+            mainwindow.set_central_widget(
+                mainwindow.__class__.central_widget_key)
+        if self._view_action is not None:
+            mainwindow.panes_menu.removeAction(self._view_action)
+            mainwindow.dataframe_menu.removeAction(self._view_action)
+        if self._set_central_action is not None:
+            mainwindow.central_widgets_menu.removeAction(
+                self._set_central_action)
+        if key is not None:
+            del mainwindow.plugins[key]
+        if self.dock is not None:
+            mainwindow.removeDockWidget(self.dock)
+            self.dock.close()
+        self.close()
+
 
 class LoadFromConsoleButton(QToolButton):
     """A toolbutton to load an object from the console"""

tests/test_main.py

@@ -32,6 +32,10 @@ def test_central_widget(self):
         self.window.set_central_widget(self.window.figures_tree)
         self.assertIs(self.window.centralWidget(), self.window.figures_tree)
 
+    def test_remove_plugin(self):
+        self.window.plugins['psyplot_gui_test.plugin:W1:w1'].remove_plugin()
+        self.assertNotIn('psyplot_gui_test.plugin:W1:w1', self.window.plugins)
+
 
 if __name__ == "__main__":
     unittest.main()