utilisation fichier csv et ajout groupe de chiffres

This commit is contained in:
Mylloon 2021-06-03 21:58:14 +02:00
parent e4c7c2a75b
commit 89e12b5f1f

32
main.py
View file

@ -1,19 +1,45 @@
from os import walk import re
import sys import sys
from pathlib import Path from pathlib import Path
from os import walk
def listeDossier(path = None): 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: if path == None:
path = Path().absolute() path = Path().absolute()
else: else:
path = Path(path) path = Path(path)
# récupération des fichiers
_, _, filenames = next(walk(path)) _, _, filenames = next(walk(path))
with open(path.parent.absolute() / "resultat.txt", 'w') as f: # création du fichier résultat s'il n'existe pas
f.write('\n'.join(filenames)) 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!" 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__": if __name__ == "__main__":
print(listeDossier(sys.argv[1] if len(sys.argv) > 1 else None)) print(listeDossier(sys.argv[1] if len(sys.argv) > 1 else None))