Exécution d'un script Python sur tous les fichiers d'un répertoire

Question

J'ai un script Python qui lit un fichier csv de texte et crée un fichier de liste de lecture. Cependant, je ne peux que faire un à la fois, comme:

python playlist.py foo.csv foolist.txt 

Cependant, j'ai un répertoire de fichiers qui doivent être transformés en une liste de lecture, avec des noms différents, et parfois un nombre différent de fichiers. Jusqu'à présent, j'ai regardé la création d'un fichier txt avec une liste de tous les noms du fichier dans le répertoire, puis la boucle à travers chaque ligne de cela, mais je sais qu'il doit y avoir un moyen plus facile de le faire.

Answer 1

for f in *.csv; do 
    python playlist.py "$f" "${f%.csv}list.txt" 
done 

Est-ce que cela fera l'affaire? Cela mettra foo.csv dans foolist.txt et abc.csv dans abclist.txt. Ou voulez-vous tous dans le même fichier?

Answer 2

si vous avez le nom du répertoire que vous pouvez utiliser os.listdir

os.listdir(dirname) 

si vous voulez sélectionner uniquement un certain type de fichier, par exemple, seul fichier csv, vous pouvez utiliser le module glob.

Answer 3

Il suffit d'utiliser une boucle avec le glob astérisque, en vous assurant que vous citez les choses de façon appropriée pour les espaces dans les noms de fichiers

for file in *.csv; do 
    python playlist.py "$file" >> outputfile.txt; 
done 

Answer 4

Est-il un seul répertoire, ou imbriquées?

Ex.

topfile.csv 
topdir 
    --dir1 
     --file1.csv 
     --file2.txt 
     --dir2 
      --file3.csv 
      --file4.csv 

Pour imbriqué, vous pouvez utiliser os.walk(topdir) pour obtenir tous les fichiers et répertoires récursive dans un répertoire.

Vous pouvez configurer votre script pour accepter ou dirs fichiers:

python playlist.py topfile.csv topdir

import sys 
import os 

def main(): 
    files_toprocess = set() 
    paths = sys.argv[1:] 
    for p in paths: 
     if os.path.isfile(p) and p.endswith('.csv'): 
      files_toprocess.add(p) 
     elif os.path.isdir(p): 
      for root, dirs, files in os.walk(p): 
       files_toprocess.update([os.path.join(root, f) 
             for f in files if f.endswith('.csv')])