Comment puis-je vérifier l'extension d'un fichier ?

Question

Je travaille sur un certain programme où je dois faire différentes choses en fonction de l'extension du fichier. Pourrais-je simplement utiliser ceci ?

if m == *.mp3
   ...
elif m == *.flac
   ...

Answer 1

En supposant que m est une chaîne de caractères, vous pouvez utiliser endswith :

if m.endswith('.mp3'):
...
elif m.endswith('.flac'):
...

Pour ne pas tenir compte de la casse, et pour éliminer une chaîne else-if potentiellement importante :

m.lower().endswith(('.png', '.jpg', '.jpeg'))

Answer 2

os.path fournit de nombreuses fonctions pour manipuler les chemins/noms de fichiers. ( docs )

os.path.splitext prend un chemin et sépare l'extension de fichier de sa fin.

import os

filepaths = ["/folder/soundfile.mp3", "folder1/folder/soundfile.flac"]

for fp in filepaths:
    # Split the extension from the path and normalise it to lowercase.
    ext = os.path.splitext(fp)[-1].lower()

    # Now we can simply use == to check for equality, no need for wildcards.
    if ext == ".mp3":
        print fp, "is an mp3!"
    elif ext == ".flac":
        print fp, "is a flac file!"
    else:
        print fp, "is an unknown file format."

Donne :

/folder/soundfile.mp3 is an mp3!
folder1/folder/soundfile.flac is a flac file!

Answer 3

Utilisez pathlib À partir de Python3.4.

from pathlib import Path
Path('my_file.mp3').suffix == '.mp3'

Answer 4

Regardez le module fnmatch. Il fera ce que vous essayez de faire.

import fnmatch
import os

for file in os.listdir('.'):
    if fnmatch.fnmatch(file, '*.txt'):
        print file

Answer 5

Ou peut-être :

from glob import glob
...
for files in glob('path/*.mp3'): 
  do something
for files in glob('path/*.flac'): 
  do something else