bus

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
"""
Script que recibe como argumento la ruta a una carpeta y una extensión
y busca los archivos con esa extensión dentro de ella a través de todos
sus subdirectorios
>>> python buscar_archivos.py <ruta> <extension>
"""

import sys
import os

hallados = 0

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

# asigno los argumentos
ruta, extension = sys.argv[1:]

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

# busco los archivos con la extensión deseada
# os.walk() devuelve 3 iteradores
# 1) El nombre del directorio actual
# 2) Una lista con todas las carpetas del directorio actual
# 3) Una lista con todas los archivos del directorio actual

for dir_actual, lista_dirs, lista_archivos in os.walk(ruta):
	for archivo in lista_archivos:
		if archivo.endswith(extension):
			print(os.path.join(dir_actual, archivo))
			hallados += 1

# Si no encuentra archivos, lo indica
if not hallados:
	print(f"No se hallaron archivos con extensión {extension} en el directorio {ruta}")
else:
	print(f"Total de archivos hallados: {hallados}")