busqueda_archivos

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
"""
Script que recie como argumentos la ruta a una carpeta y una extensión
para buscar archivos dentro de ella (sin recursividad). Luego, se deben
mostrar los nombres de los archivos hallados.
Ej 1:
  python buscar_archivos.py C:\\User\\Usuario .jpeg

Ej 2:
  python buscar_archivos.py C:\\User\\Usuario jpeg

Si no se pasan los dos argumentos, el script debe indicar el error en consola
y salir. (Idem si la ruta no existe)

"""
import os
import sys

# lista que almacena los archivos buscados
archivos = []

# verificar los argumentos
if len(sys.argv) != 3:
    print("Error: debe ejecutar 'python buscar_archivos.py <ruta> <extension>'")
    sys.exit()

# asigna los argumentos
ruta, extension = sys.argv[1:]

# verifico si existe la ruta
if not os.path.exists(ruta):
    print(f"Error: no se encuentra la ruta {ruta}")
    sys.exit()

# necesito la extension sin el punto (si existe lo borra)
extension = extension.replace(".","")

# busco los archivos y los guardo en la lista
for elemento in os.listdir(ruta):
    nombre = elemento.split(".")
    if len(nombre) == 2 and nombre[1] == extension:
        archivos.append(elemento)

# muestro los resultados
if archivos:
    print(f"Archivos con extensión '{extension}':")
    for archivo in archivos:
        print(archivo)
else:
    print(f"No se hallaron archivos con extensión {extension} en la ruta {ruta}")