oswalk

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
"""
Script que recibe como argumentos una ruta a una carpeta y una extensión
para buscar archivos dentro de ella y mostrar una lista. Si no se pasan
los argumentos, el script debe indicar el error y finalizar. Si la ruta
no existe, el script debe señalarlo y terminar

Ej:
>>> python buscar_archivos.py C:\\Users\\Usuario xlsx
>>> muestra.xlsx
>>> sample.xlsx
>>>....

"""

import sys
import os

hallados = 0

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

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

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

# busco en las carpetas dentro de la ruta
# os.walk() itera en las carpetas (necesita 3 variables iteradoras)
# 1. el nombre de la carpeta actual
# 2. una lista de carpetas dentro de la carpeta actual
# 3. una lista de archivos dentro de la carpeta actual

for carpeta_actual, lista_carpetas, lista_archivos in os.walk(ruta):
    for archivo in lista_archivos:
        if archivo.endswith(extension):
            print(os.path.join(carpeta_actual, archivo))
            hallados += 1

if not hallados:
    print(f"No se hallaron archivos con extension {extension} en {ruta}")
else:
    print(f"Total: {hallados}")