buscar_archivoa

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
"""
Script que busca archivos de un tipo determinado en un directorio en
forma recursiva (python buscar_archivos.py <ruta> <extension>). Indica
también la cantidad total de archivos hallados

Ej:
>>> python buscar_archivos.py C:\\Users\\Usuario\\Desktop pdf
>>> saludo.pdf
>>> notas.pdf
>>> saldos.pdf
>>> .....
>>> Total: 143 archivos
"""

import os
import sys
total = 0


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

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

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

# busco en forma recursiva los archivos con os.walk()
# os.walk itera y devuelve 3 valores en cada iteración
# 1. el nombre de la carpeta actual
# 2. una lista con las subcarpetas dentro de la carpeta actual
# 3. una lista con los archivos dentro de la carpeta actual
for carpeta_actual, lista_subcarpetas, 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 total:
    print(f"\nTotal de archivos hallados: {total}")
else:
    print(f"No se hallaron archivos '{extension}' en {ruta}")