Sort Comps groups by category

1 files changed, 10 insertions, 8 deletions

diff --git a/gui.py b/gui.py
index 9fefecd..e5d667b 100755
--- a/gui.py
+++ b/gui.py
@@ -150,15 +150,17 @@ class PackagesUI:
         item = GenericItem("Comps Groups")
         self.searchItems.append(item)
 
-        self.categories = {}
         for cat in self.yum.comps.get_categories():
-            self.categories[cat.ui_name] = []
-
-        for group in self.yum.comps.groups:
-            pkgs = self._names2pkgs(group.packages)
-            if not group.user_visible or not pkgs:
-                continue
-            item.append(SearchItem(group.name, pkgs))
+            categoryItem = GenericItem(cat.ui_name)
+            item.append(categoryItem)
+            for g in cat.groups:
+                if self.yum.comps.has_group(g):
+                    group = self.yum.comps.return_group(g)
+                    if not group.user_visible:
+                        continue
+                    pkgs = self._names2pkgs(group.packages)
+                    if pkgs:
+                        categoryItem.append(SearchItem(group.ui_name, pkgs))
 
     def _getRepositories(self):
         item = GenericItem("Repositories")