busqueda archivos

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
Script que busca archivos en forma recursiva denttro de un directorio
dada una ruta y una extension
Ej de uso:
buscar.py <ruta> <extension>
"""
import os
import sys

hallados = 0
# verifico la cantidad de argumentos
if len(sys.argv) != 3:
    sys.exit("Error: ejecute 'buscar.py <ruta> <extension>'")

# asigno los parametros
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}'")

# busqueda recursiva dentro de la ruta
# os.walk() devuelve 3 iteradores en cada iteración
# 1. carpeta actual
# 2. lista de subcarpetas dentro de la carpeta actual
# 3. 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 la ruta '{ruta}'")
else:
    print(f"\nTotal: {hallados} archivos")