Modificacion de get_categories para que filtre correctamente el segundo combobox

1 files changed, 19 insertions, 6 deletions

diff --git a/store.py b/store.py
index fdfc131..1c54f48 100644
--- a/store.py
+++ b/store.py
@@ -74,14 +74,27 @@ class Db:
         c.execute('END TRANSACTION')
         self._close_connection(con)
 
-    def get_categories(self, order='+'):
+    def get_categories(self, order='+',value="Tipo"):
         if order == '+':
             order = "ASC"
         else:
             order = "DESC"
-        query = "select distinct type from notifications order by type "+order
-        return map(lambda x:x['type'],self.run_query(query))
-        
+
+        mappedValue = "type"
+
+        if value == "Tipo":
+            mappedValue = "type"
+        elif value == "Prioridad":
+            mappedValue = "priority"
+        elif value == "Lanzamiento":
+            mappedValue = "launched"
+        elif value == u"Expiración":
+            mappedValue = "expires"
+
+        query = "select distinct " + mappedValue + " from notifications order by type " + order
+
+        return map(lambda x:x[mappedValue],self.run_query(query))
+
     def set_fav(self, id_msg, fav=True):
         """Marca un mensaje como favorito.
         Si fav == False lo desmarca"""
@@ -188,7 +201,7 @@ class Store:
         values = []
         for item in keys:
             values.append(str(msg[item]))
-        self.db.add_message(keys, values)     
+        self.db.add_message(keys, values)
 
     def _save_XML_message(self, msg):
         # For future releases
@@ -200,7 +213,7 @@ class Store:
         map(lambda x: values.append(msg.getAttribute(x)), keys)
         for node in refNode:
             if node.nodeType == 1:
-                #print "clave: %s, valor: %s" % (node.localName, node.firstChild.data) 
+                #print "clave: %s, valor: %s" % (node.localName, node.firstChild.data)
                 keys.append(node.localName)
                 values.append(node.firstChild.wholeText.strip())
         self.db.add_message(keys, values)