buscar archivos

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
"""
Script que solicita una ruta y una extension y devuelve una lista con
todos los archivos hallados.
EJ:
>> python buscar_archivos.py C:/Users jpeg
>> C:/Users/Usuarios/Desktop/foto.jpeg
>>...
>> Total: 1256 archivos hallados
"""
import os
import sys

hallados = 0

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


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

# verifico que la ruta exista:
if not os.path.exists(ruta):
    sys.exit(f"No existe la ruta '{ruta}'")

# busco los directorios dentro de la ruta en forma recursiva
# os.walk() devuelve 3 iteradores en cada iteración
# a) el nombre de la carpeta actual
# b) una lista de subcarpetas dentro de la carpeta actual
# c) 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 '{extension}' en '{ruta}'")

else:
    print(f"Total: {hallados} archivos hallados")