buscar procesos

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
"""
Script que pide un directorio y una extensión y devuelve una lista
con todos los archivos hallados
>>> python buscar_archivos <ruta> <extension>

"""
import os
import sys

# variable que cuenta los archivos hallados
hallados = 0

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

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

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

# voy a buscar en forma recursiva dentro de ruta
# os.walk() devuelve 3 valores en cada iteración
# 1) el nombre de la carpeta actual
# 2) una lista de carpetas dentro de carpeta actual
# 3) una lista de archivos dentro de 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 hallados:
    print(f"Total de archivos: {hallados}")

else:
    print(f"No se hallaron archivos con extension '{extension}' en '{ruta}'")