From 0610d2f9f027f8659e7149f2dfbb316ff30e337d Mon Sep 17 00:00:00 2001
From: wakonig_k <klaus.wakonig@psi.ch>
Date: Thu, 27 Jun 2024 20:05:20 +0200
Subject: [PATCH] fix: fixed qwidget inheritance for ring progress bar

---
 bec_widgets/utils/generate_designer_plugin.py   | 17 +++++++++++++----
 .../ring_progress_bar/ring_progress_bar.py      |  2 +-
 2 files changed, 14 insertions(+), 5 deletions(-)

diff --git a/bec_widgets/utils/generate_designer_plugin.py b/bec_widgets/utils/generate_designer_plugin.py
index 31a46f8f..84d30bdc 100644
--- a/bec_widgets/utils/generate_designer_plugin.py
+++ b/bec_widgets/utils/generate_designer_plugin.py
@@ -1,10 +1,10 @@
 import inspect
 import os
 import re
-from unittest import mock
 
 from qtpy.QtCore import QObject
-from qtpy.QtWidgets import QWidget
+
+EXCLUDED_PLUGINS = ["BECConnector", "BECDockArea", "BECDock"]
 
 
 class DesignerPluginInfo:
@@ -45,7 +45,13 @@ class DesignerPluginInfo:
 
 class DesignerPluginGenerator:
     def __init__(self, widget: type):
+        self._excluded = False
         self.widget = widget
+        if widget.__name__ in EXCLUDED_PLUGINS:
+
+            self._excluded = True
+            return
+
         self.info = DesignerPluginInfo(widget)
 
         self.templates = {}
@@ -54,6 +60,9 @@ class DesignerPluginGenerator:
         )
 
     def run(self):
+        if self._excluded:
+            print(f"Plugin {self.widget.__name__} is excluded from generation.")
+            return
         self._check_class_validity()
         self._load_templates()
         self._write_templates()
@@ -132,7 +141,7 @@ class DesignerPluginGenerator:
 
 if __name__ == "__main__":
     # from bec_widgets.widgets.bec_queue.bec_queue import BECQueue
-    from bec_widgets.widgets.ring_progress_bar.ring_progress_bar import RingProgressBar
+    from bec_widgets.widgets.dock import BECDockArea
 
-    generator = DesignerPluginGenerator(RingProgressBar)
+    generator = DesignerPluginGenerator(BECDockArea)
     generator.run()
diff --git a/bec_widgets/widgets/ring_progress_bar/ring_progress_bar.py b/bec_widgets/widgets/ring_progress_bar/ring_progress_bar.py
index 548944a9..3d50f72a 100644
--- a/bec_widgets/widgets/ring_progress_bar/ring_progress_bar.py
+++ b/bec_widgets/widgets/ring_progress_bar/ring_progress_bar.py
@@ -104,7 +104,7 @@ class RingProgressBar(BECConnector, QWidget):
                 config = RingProgressBarConfig(**config, widget_class=self.__class__.__name__)
             self.config = config
         super().__init__(client=client, config=config, gui_id=gui_id)
-        QWidget.__init__(self, parent=None)
+        QWidget.__init__(self, parent=parent)
 
         self.get_bec_shortcuts()
         self.entry_validator = EntryValidator(self.dev)