Comment obtenir l'extension de fichier en Python

Nous pouvons utiliser la fonction splitext() du module os de Python pour obtenir l'extension du fichier. Cette fonction divise le chemin du fichier en un tuple ayant deux valeurs - la racine et l'extension.

Obtenir l'extension de fichier en Python

Voici un programme simple pour obtenir l'extension de fichier en Python.

import os

unpacking the tuple
file_name, file_extension = os.path.splitext("/Users/pankaj/abc.txt")

print(file_name)
print(file_extension)

print(os.path.splitext("/Users/pankaj/.bashrc"))
print(os.path.splitext("/Users/pankaj/a.b/image.png"))

Sortie:

• Dans le premier exemple, nous décompressons directement les valeurs du tuple dans les deux variables.
• Notez que le fichier .bashrc n'a pas d'extension. Le point est ajouté au nom du fichier pour en faire un fichier caché.
• Dans le troisième exemple, il y a un point dans le nom du répertoire.

Obtenir l'extension de fichier en utilisant le module Pathlib

Nous pouvons également utiliser le module pathlib pour obtenir l'extension de fichier. Ce module a été introduit dans la version 3.4 de Python.

>>> import pathlib
>>> pathlib.Path("/Users/pankaj/abc.txt").suffix
'.txt'
>>> pathlib.Path("/Users/pankaj/.bashrc").suffix
''
>>> pathlib.Path("/Users/pankaj/.bashrc")
PosixPath('/Users/pankaj/.bashrc')
>>> pathlib.Path("/Users/pankaj/a.b/abc.jpg").suffix
'.jpg'
>>> 

Conclusion

Il est toujours préférable d'utiliser les méthodes standard pour obtenir l'extension de fichier. Si vous utilisez déjà le module os, utilisez alors la méthode splitext(). Pour une approche orientée objet, utilisez le module pathlib.