buscar_arch

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Programa que saluda
"""
Script que recibe como argumentos una ruta a un directorio y una
extensión y devuelve un listado con todos los archivos encontrados
Ejemplo de uso:
>>> python buscar_archivos.py C:\\Users .docx
"""
import sys
import os

hallados = 0

if sys.argv[1] == "--help":
    sys.exit("Ayuda: bla bla bla")

# 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}'")

# ahora busco los archivos
# os.walk() busca con 3 iteradores
# 1) el nombre de la carpeta actual
# 2) la lista de carpetas dentro de la carpeta actual
# 3) la 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):
            hallados += 1
            print(os.path.join(carpeta_actual, archivo))

if hallados:
    print(f"Total de archivos: {hallados}")
else:
    print(f"No se hallaron archivos '{extension}' en '{ruta}'")