Gestion des catégories

1 files changed, 12 insertions, 5 deletions

diff --git a/data_textes.py b/data_textes.py
index 85baa54..dc8626a 100644
--- a/data_textes.py
+++ b/data_textes.py
@@ -74,12 +74,19 @@ def extraire_textes_par_cat(table):
             
     return (categ,tableqr)
 
-def extraire_qr(table):
-    """ on enlève juste la catégorie en début de ligne"""
-    table2 = [ ligne[1:] for ligne in table]
+def extraire_qr(table, listecateg):
+    """ 
+    table est la table des remarques. On sélectionne uniquement
+    les remarques des catégories de la liste et
+    on enlève juste la catégorie en début de ligne
+    Si listecateg est vide on met tout le monde.
+    """
+    table2 = []
+    for ligne in table:
+        if ligne[0] in listecateg or listecateg == []:
+            table2.append(ligne[1:])
     return table2
 
-tablecomplete = lire_fichier_csv_simple(fichier_bingo)
-tabletextes = extraire_qr(tablecomplete)
+
 
 #liste_cat, liste_textes = extraire_textes_par_cat(lire_fichier_csv_simple(fichier_bingo))
\ No newline at end of file