[flatpak] fix: category formatting | sorting categories

bauh/gems/flatpak/worker.py

@@ -1,5 +1,5 @@
-import re
 import traceback
+from io import StringIO
 from threading import Thread
 
 from bauh.api.abstract.cache import MemoryCache
@@ -10,8 +10,6 @@ from bauh.api.http import HttpClient
 from bauh.gems.flatpak.constants import FLATHUB_API_URL, FLATHUB_URL
 from bauh.gems.flatpak.model import FlatpakApplication
 
-RE_SPLIT_UPPER = re.compile(r'[A-Z][a-z]*')
-
 
 class FlatpakAsyncDataLoader(Thread):
 
@@ -25,6 +23,20 @@ class FlatpakAsyncDataLoader(Thread):
         self.logger = context.logger
         self.category_cache = category_cache
 
+    @staticmethod
+    def format_category(category: str) -> str:
+        word = StringIO()
+        last_l = None
+        for idx, l in enumerate(category):
+            if idx != 0 and last_l != ' ' and l.isupper() and idx + 1 < len(category) and category[idx + 1].islower():
+                word.write(' ')
+
+            last_l = l.lower()
+            word.write(last_l)
+
+        word.seek(0)
+        return word.read()
+
     def run(self):
         if self.app:
             self.app.status = PackageStatus.LOADING_DATA
@@ -60,7 +72,7 @@ class FlatpakAsyncDataLoader(Thread):
                             cached = self.category_cache.get(c['name'])
 
                             if not cached:
-                                cached = ' '.join(RE_SPLIT_UPPER.findall(c['name'])).lower() if not c['name'].isupper() else c['name'].lower()
+                                cached = self.format_category(c['name'])
                                 self.category_cache.add_non_existing(c['name'], cached)
 
                             cats.append(cached)

bauh/view/qt/commons.py

@@ -31,7 +31,8 @@ def update_info(pkgv: PackageView, pkgs_info: dict):
         pkgs_info['updates'] += 1
 
     if pkgv.model.categories:
-        pkgs_info['categories'].update([s.lower() for s in pkgv.model.categories])
+        for c in pkgv.model.categories:
+            pkgs_info['categories'].add(c.lower())
 
     pkgs_info['pkgs'].append(pkgv)
     pkgs_info['not_installed'] += 1 if not pkgv.model.installed else 0

bauh/view/qt/window.py

@@ -417,7 +417,8 @@ class ManageWindow(QWidget):
 
         for p in self.pkgs_available:
             if p.model.categories:
-                categories.update(p.model.categories)
+                for c in p.model.categories:
+                    categories.add(c.lower())
 
         if categories:
             self._update_categories(categories, keep_selected=True)
@@ -727,7 +728,7 @@ class ManageWindow(QWidget):
         if categories is None:
             self.ref_combo_categories.setVisible(self.combo_categories.count() > 1)
         else:
-            keeping_selected = keep_selected and categories and self.category_filter in {c.lower() for c in categories}
+            keeping_selected = keep_selected and categories and self.category_filter in categories
 
             if not keeping_selected:
                 self.category_filter = self.any_category_filter
@@ -738,13 +739,15 @@ class ManageWindow(QWidget):
                         self.combo_categories.removeItem(1)
 
                 selected_cat = -1
-                for idx, c in enumerate(categories):
-                    lower_cat = c.lower()
-                    i18n_cat = self.i18n.get(lower_cat)
-                    cat_label = i18n_cat if i18n_cat else c
-                    self.combo_categories.addItem(cat_label.capitalize(), lower_cat)
+                cat_list = list(categories)
+                cat_list.sort()
 
-                    if keeping_selected and lower_cat == self.category_filter:
+                for idx, c in enumerate(cat_list):
+                    i18n_cat = self.i18n.get(c)
+                    cat_label = i18n_cat if i18n_cat else c
+                    self.combo_categories.addItem(cat_label.capitalize(), c)
+
+                    if keeping_selected and c == self.category_filter:
                         selected_cat = idx + 1
 
                 self.combo_categories.blockSignals(True)