arch

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

"""
Script que recibe como argumentos la ruta a una carpeta y una extension
para buscar archivos dentro de ella. Luego debe mostrar los archivos.
Si no se pasan los dos argumentos, el sistema debe indicar el error y
finalizar la ejecución del programa. Lo mismo si la ruta no existe
Ej
>>> python buscar_archivos.py C:\\Users .jpeg


"""
import os
import sys

total = 0

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

# asignar los parametros
ruta, extension = sys.argv[1:]

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

# busco los archivos en forma
# os.walk() devuelve 3 valores en cada iteración:
#   * el nombre de la carpeta actual en donde está iterando
#   * una lista de carpetas dentro de la carpeta actual
#   * una lista de archivos de la carpeta actual

for carpeta_actual, lista_carpetas, lista_archivos in os.walk(ruta):
    for archivo in lista_archivos:
        if archivo.endswith(extension):
            total += 1
            print(os.path.join(carpeta_actual,archivo))


if total:
    print(f"total de archivos: {total}")
else:
    print(f"No existen archivos '{extension}' en {ruta}")