buscar

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
"""
Script que pide una ruta y una extension y devuelve una lista con todos
los archivos hallados, ademàs muestra el total

EJ:
>> python buscar_archivos.py C:/Users/Usuario .jpeg

"""
import os
import sys

hallados = 0

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

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

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

# busco en forma recursiva dentro de "ruta"
# os.walk() devuelve 3 valores en cada iteración:
# a) el nombre del directorio actual
# b) una lista con los subdirectorios dentro del directorio
# c) Una lista con los archivos dentro del directorio

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

if not hallados:
    print(f"No se hallaron archivos con extensión {extension} en '{ruta}'")
else:
    print(f"\nTotal: {hallados}")