lib.py: tune the code + enabling item will enable its ancestors

Signed-off-by: Jan Pokorný <jpokorny@redhat.com>

1 files changed, 16 insertions, 10 deletions

diff --git a/lib.py b/lib.py
index b4fd12a..56f6fbf 100644
--- a/lib.py
+++ b/lib.py
@@ -312,28 +312,34 @@ def xdot_graph(*args, **kwargs):
             label = widget.get_label()
             bl = self._kwargs.setdefault('blacklist', [])
             change = False
-            if widget.active and globals()[label] in bl:
-                bl.remove(globals()[label])
+            cls = globals()[label]
+            if widget.active and cls in bl:
+                bl.remove(cls)
+                bs = bases(cls)
+                bs.remove(cls)
+                for item in filter(lambda x: isinstance(x, gtk.CheckMenuItem)
+                                   and globals()[x.get_label()] in bs,
+                                   widget.get_parent().get_children()):
+                    if not item.get_active():
+                        item.set_active(True)
                 change = True
-            elif not widget.active and label not in bl:
-                bl.append(globals()[label])
+            elif not widget.active and cls not in bl:
+                bl.append(cls)
                 change = True
             if change:
                 self.set_dotcode(
                     gen_graph(*self._args, **self._kwargs).to_string())
 
         def on_clean_all(self, widget):
-            for item in widget.get_parent().get_children():
-                if not isinstance(item, gtk.CheckMenuItem):
-                    continue
+            for item in filter(lambda x: isinstance(x, gtk.CheckMenuItem),
+                               widget.get_parent().get_children()):
                 item.set_active(False)
             item.set_active(True)
             item.activate()
 
         def on_set_all(self, widget):
-            for item in widget.get_parent().get_children():
-                if not isinstance(item, gtk.CheckMenuItem):
-                    continue
+            for item in filter(lambda x: isinstance(x, gtk.CheckMenuItem),
+                               widget.get_parent().get_children()):
                 item.set_active(True)
             item.set_active(False)
             item.activate()