diff options
Diffstat (limited to 'store.py')
-rw-r--r-- | store.py | 25 |
1 files changed, 19 insertions, 6 deletions
@@ -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) |