buscar_archivos.py

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

# Crer un script que reciba como argumentos una ruta y un tipo de archivo
# y devuelva una lista con todos ellos y la cantidad total. El programa
# debe indicar si faltanm argumentos, si la ruta no existe, etc. No vamos
# a validar extensiones: se considera válida cualquier extensión
# LA BUSQUEDA DEBE SER RECURSIVA
# Ejemplo: de uso
# >>> python buscar_archivos.py C:/Users/Usuario .jpeg
# >>> python buscar_archivos.py <ruta> <extension>

import os
import sys
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 parámetros
ruta, extension = sys.argv[1:]

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

# busco los archivos dentro de la ruta
# os.walk() devuelve 3 valores en cada iteración:
# el nombre de la carpeta actual
# una lista de carpetas dentro de la carpeta actual
# 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 hallados:
    print(f"Total de archivos: {hallados}")
else:
    print(f"No se hallaron archivos con extensión {extension} en la ruta '{ruta}'")