dirs

#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
Script que recibe como argumentos la ruta a un directorio y una extensión
para buscar todos los archivos. El programa debe manejar los errores
que pudieran surgir (ruta inexistente, falta de argumentos, etc...)
EJ:
>>> python buscar_archivos.py
>>> Error: se debe ejecutar 'buscar_archivos.py <ruta> <extension>'
>>> python buscar_archivos.py C:\\Users py
>>> hola.py
    saludo.py
    ....

"""
import os
import sys

# lista que almacena los archivos hallados
hallados = []


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

# asigno los parámetros
ruta, extension = sys.argv[1:]

# verifico si la ruta existe
if not os.path.exists(ruta):
    sys.exit(f"No se ha encontrado el directorio '{ruta}'")

# busqueda de archivos en forma recursiva a partir de un directorio raiz
for root, directorio, archivos in os.walk(ruta):
    for archivo in archivos:
        if archivo.endswith(extension):
            hallados.append(archivo)

# si no encuentro archivos, lo muestro en pantalla
if hallados:
    for hallado in hallados:
        print(hallado)
else:
    print(f"No se hallaron archivos con extension '{extension}' en el directorio '{ruta}'")