utilisation fichier csv et ajout groupe de chiffres

main.py

@@ -1,19 +1,45 @@
-from os import walk
+import re
 import sys
 from pathlib import Path
+from os import walk
 
 def listeDossier(path = None):
+    """Ecris tous les noms des fichiers d'un dossier dans un fichier csv, en séparant les groupes de chiffres"""
+    # récupération path
     if path == None:
         path = Path().absolute()
     else:
         path = Path(path)
 
+    # récupération des fichiers
     _, _, filenames = next(walk(path))
 
-    with open(path.parent.absolute() / "resultat.txt", 'w') as f:
-        f.write('\n'.join(filenames))
+    # création du fichier résultat s'il n'existe pas
+    resultatPath = path.parent.absolute() / "resultat.csv"
+    if not isFileExists(resultatPath):
+        open(resultatPath, "x")
     
+    # ajout des groupes de chiffres dans les colomnes du csv
+    for i in range(0, len(filenames)):
+        ID = re.findall(r'\d+', filenames[i])
+        for j in range(0, len(ID)):
+            filenames[i] = f"{filenames[i]};{ID[j]}"
+
+    # ajout des résultat au fichier
+    with open(resultatPath, 'a') as f:
+        f.write('\n'.join(filenames))
+        f.write('\n')
+
     return "Done!"
 
+def isFileExists(path):
+    """Vérifie qu'un fichier existe"""
+    try:
+        open(path, "r")
+    except FileNotFoundError:
+        return False
+    else:
+        return True
+
 if __name__ == "__main__":
     print(listeDossier(sys.argv[1] if len(sys.argv) > 1 else None))