buscar_archivos

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

"""
Script que recibe como argumentos a traves de la consola un ruta y una
extensión y devuelve una lista de los archivos encontrados y la
cantidad total.
Uso:
>>> python buscar_archivos.py  <ruta>  <extension>

Ej:
>>> python buscar_archivos.py   C:\\Users\\Usuario   .jpeg
>>> python buscar_archivos.py   C:\\Users\\Usuario\\Desktop   pdf
"""
total = 0
# verifico 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 si existe la ruta
if not os.path.exists(ruta):
    sys.exit(f"No existe el directorio '{ruta}'")

# busco en forma recursiva dentro de la ruta
# usaremos os.walk()
# os.walk() devuelve 3 valores en cada iteracion:
#  1. el nombre de la carpeta actual
#  2. La 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))
            total += 1

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