archivos

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
"""
Script que pide una ruta y una extensión y devuelve una lista con todos
los archivos de ese tipo hallados:

>>> python buscar_archivos.py <directorio> <extension>
EJ
>> python buscar_archivos.py C:\\Users\\Usuario\\Desktop .jpeg
>>> hola.jpeg
>>> foto.jpeg
........
>>> Total: 2345 archivos
"""
import os
import sys

total = 0

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

# asigno los parámetros
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 los archivos dentro de la ruta
# uso os.walk
# devuelve 3 valores en cada iteración:
# a) el nombre de la carpeta actual
# b) una lista de carpetas 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))
            total += 1

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