script

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
"""
Script que recibe como argumentos la ruta a una carpeta y una
extensión para buscar archivos dentro de ella (sin recursividad). Luego,
se deben mostrar los archivos hallados. Si no se pasan los dos argumentos
correspondientes, el programa debe indicar el error en la consola y
finalizar. Si la ruta no existe, el programa debe indicarlo en consola
y finalizar
Ejemplo:
    python buscar_archivos.py C:\\User\\Usuario\\Desktop .jpeg
    python buscar_archivos.py C:\\User\\Usuario\\Desktop jpeg

    python buscar_archivos.py <ruta> <extension>
"""
import sys
import os

# lista que almacena los archivos buscados
archivos = []

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

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

# verifico la validez de la ruta
if not os.path.exists(ruta):
    print(f"Error: no existe la ruta '{ruta}'")
    sys.exit()

# necesito la extension sin el punto
if extension.startswith("."):
    extension = extension.replace(".","")

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

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