Do not show categories which only hold disabled packages

3 files changed, 16 insertions, 7 deletions

diff --git a/portato/db/database.py b/portato/db/database.py
index c36c5dc..bc78b01 100644
--- a/portato/db/database.py
+++ b/portato/db/database.py
@@ -59,13 +59,15 @@ class Database (object):
         """
         raise NotImplentedError
 
-    def get_cat (self, cat = None, byName = True):
+    def get_cat (self, cat = None, byName = True, showDisabled = False):
         """Returns the packages in the category.
         
         @param cat: category to return the packages from; if None it defaults to C{ALL}
         @type cat: string
         @param byName: selects whether to return the list sorted by name or by installation
         @type byName: boolean
+        @param showDisabled: should disabled packages be returned
+        @type showDisabled: boolean
         @return: an iterator over the packages
         @rtype: L{PkgData}<iterator>
         """
diff --git a/portato/db/dict.py b/portato/db/dict.py
index fd859dd..d7e2649 100644
--- a/portato/db/dict.py
+++ b/portato/db/dict.py
@@ -62,17 +62,20 @@ class DictDatabase (Database):
             self._db[key].sort(key = self.__sort_key)
 
     @lock
-    def get_cat (self, cat = None, byName = True):
+    def get_cat (self, cat = None, byName = True, showDisabled = False):
         if not cat:
             cat = self.ALL
 
         def get_pkgs():
             if byName:
                 for pkg in self._db[cat]:
-                    yield pkg
+                    if showDisabled or not pkg.disabled:
+                        yield pkg
             else:
                 ninst = []
                 for pkg in self._db[cat]:
+                    if not showDisabled and pkg.disabled: continue
+
                     if pkg.inst:
                         yield pkg
                     else:
diff --git a/portato/db/sql.py b/portato/db/sql.py
index b406981..cac5c97 100644
--- a/portato/db/sql.py
+++ b/portato/db/sql.py
@@ -169,15 +169,19 @@ class SQLDatabase (Database):
         connection.commit()
 
     @con
-    def get_cat (self, category = None, byName = True, connection = None):
+    def get_cat (self, category = None, byName = True, showDisabled = False, connection = None):
         sort = "ORDER BY name"
         if not byName:
             sort = "ORDER BY inst DESC, name"
 
+        disabled = "1=1"
+        if not showDisabled:
+            disabled = "disabled = 0"
+
         if not category or category == self.ALL:
-            c = connection.execute("SELECT cat, name, inst, disabled FROM packages WHERE 1=1 %s %s" % (self.restrict, sort))
+            c = connection.execute("SELECT cat, name, inst, disabled FROM packages WHERE %s %s %s" % (disabled, self.restrict, sort))
         else:
-            c = connection.execute("SELECT cat, name, inst, disabled FROM packages WHERE cat = ? %s %s" % (self.restrict ,sort), (category,))
+            c = connection.execute("SELECT cat, name, inst, disabled FROM packages WHERE cat = ? AND %s %s %s" % (disabled, self.restrict ,sort), (category,))
         
         for pkg in c:
             yield PkgData(pkg["cat"], pkg["name"], pkg["inst"], pkg["disabled"])
@@ -191,7 +195,7 @@ class SQLDatabase (Database):
         else:
             where = "1 = 1"
 
-        c = connection.execute("SELECT cat FROM packages WHERE %s %s GROUP BY cat" % (where, self.restrict))
+        c = connection.execute("SELECT cat FROM packages WHERE disabled = 0 AND %s %s GROUP BY cat" % (where, self.restrict))
 
         l = c.fetchall()
         c.close()