Rechercher un fichier en Python

La recherche de fichiers en Python peut être un défi lorsque l'emplacement du fichier varie selon les différents systèmes. Heureusement, Python fournit le module os.walk qui vous permet de parcourir les répertoires et de localiser les fichiers sans effort.

os.walk pour la recherche de fichiers

os.walk crée un générateur objet qui parcourt un répertoire et ses sous-répertoires, produisant trois tuples pour chaque itération :

• root (str) : chemin absolu vers le répertoire actuel directory
• dirs (list) : Liste des sous-répertoires dans le répertoire actuel
• files (list) : Liste des fichiers dans le répertoire actuel

Recherche un fichier unique

Pour trouver la première occurrence d'un fichier spécifique, parcourez simplement les tuples générés et vérifiez si le nom du fichier est dans la liste des fichiers :

import os

def find(name, path):
    for root, dirs, files in os.walk(path):
        if name in files:
            return os.path.join(root, name)

Recherche de toutes les occurrences d'un fichier

Pour rechercher toutes les occurrences d'un fichier, créez une liste pour stocker les correspondances :

def find_all(name, path):
    result = []
    for root, dirs, files in os.walk(path):
        if name in files:
            result.append(os.path.join(root, name))
    return result

Modèles de fichiers correspondants

Pour rechercher des fichiers qui correspondent à un modèle, utilisez le module fnmatch :

import os, fnmatch
def find(pattern, path):
    result = []
    for root, dirs, files in os.walk(path):
        for name in files:
            if fnmatch.fnmatch(name, pattern):
                result.append(os.path.join(root, name))
    return result

find('*.txt', '/path/to/dir')