A veces se requiere conocer la extensión de un archivo en el lenguaje de programación Python. No hay necesidad de implementar nuestras propias funciones (por ejemplo una que nos diga con qué subcadena termina una cadena) ni trabajar con cadenas.

Python ya provee una manera de conocer la extensión de un archivo en el módulo os.path. Hoy veremos cómo usar dicho módulo para extraer la extensión (pdf, mp3, etcétera) de un archivo.

Extraer extensión de archivo usando Python

Como lo dije hace un momento, lo que nos permite saber la extensión de un archivo está en os.path así que importamos al módulo:

import os.path

La función que nos ayuda se llama splitext, que quiere decir “split extension“, o sea, separar la extensión. Este método recibe una cadena indicando la ruta del archivo, ya sea relativa o absoluta; y devuelve una tupla (dos variables).

El primer valor de la tupla es el nombre del archivo, y el segundo es la extensión. Por lo que la sintaxis es:

nombre_archivo, extension = os.path.splitext("nombre_o_ruta_de_archivo.extension")

En caso de que el archivo no tenga extensión, se devuelve una cadena vacía.

Ejemplo de uso de os.path.splitext para obtener extensión

Aquí algunos ejemplos con la shell de Python:

Como vemos, no es necesario mandarle una ruta absoluta o relativa; con una simple cadena funciona. Eso sí, en caso de mandarle una ruta compleja también sabrá separar el nombre y la extensión.

Ejemplos de uso de splitext

Finalmente aquí dejo algo de código que prueba con distintas cadenas para que nos demos una idea de cómo trabaja con los nombres que no tienen extensión:

"""
	Extraer u obtener la extensión
	de un archivo en Python usando
	os.path
	
	@author parzibyte
"""
import os.path

archivos = [
    "Free Bird.mp3", "Maggie.png", "/home/parzibyte/main.go",
    "C:\\users\\usuario\\Desktop\\Servidor.java", "Hola", ".env", "",
    "hola.php", "////////", "."
]

for archivo in archivos:
    nombre, extension = os.path.splitext(archivo)
    print("El archivo '{}' se llama '{}' y tiene la extensión '{}'".format(
        archivo, nombre, extension))

También puedes descargar el ejemplo, Instalar Python y luego ejecutar python main.py para ejecutarlo. Te invito a conocer más sobre Python aquí.