From 9c331c13e031c66861bb18a68757a3a582c438b1 Mon Sep 17 00:00:00 2001
From: wakonig_k <klaus.wakonig@psi.ch>
Date: Tue, 12 Nov 2024 15:29:48 +0100
Subject: [PATCH] refactor(gui_tools): minor cleanup

---
 .../plugins/omny/gui_tools.py                 | 42 ++++---------------
 1 file changed, 9 insertions(+), 33 deletions(-)

diff --git a/csaxs_bec/bec_ipython_client/plugins/omny/gui_tools.py b/csaxs_bec/bec_ipython_client/plugins/omny/gui_tools.py
index 328b8b8..2b3ea89 100644
--- a/csaxs_bec/bec_ipython_client/plugins/omny/gui_tools.py
+++ b/csaxs_bec/bec_ipython_client/plugins/omny/gui_tools.py
@@ -1,19 +1,17 @@
-import time
-import numpy as np
-import sys
-import termios
-import tty
+import builtins
 import fcntl
 import os
-import builtins
+import sys
+import termios
+import time
+import tty
 
+import numpy as np
+from bec_widgets.cli.client import BECDockArea
 from rich import box
 from rich.console import Console
 from rich.table import Table
 
-from bec_widgets.cli.client import BECDockArea
-
-
 # from csaxs_bec.bec_ipython_client.plugins.cSAXS import epics_get, epics_put, fshopen, fshclose
 
 if builtins.__dict__.get("bec") is not None:
@@ -78,30 +76,8 @@ class OMNYGuiTools:
                 pass
 
     def omnygui_remove_all_docks(self):
-        try:
-            self.gui.remove_dock(name="omnycam200")
-        except:
-            pass
-        try:
-            self.gui.remove_dock(name="omnycam201")
-        except:
-            pass
-        try:
-            self.gui.remove_dock(name="omnycam202")
-        except:
-            pass
-        try:
-            self.gui.remove_dock(name="omnycam203")
-        except:
-            pass
-        try:
-            self.gui.remove_dock(name="progress")
-        except:
-            pass
-        try:
-            self.gui.remove_dock(name="progress_text")
-        except:
-            pass
+        self.gui.clear_all()
+
         self.fig200 = None
         self.fig201 = None
         self.fig202 = None