buscar_archivos

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
"""
Script que recibe como argumentos una ruta a una carpeta y una extension
para buscar archivos. Luego debe mostrar los archivos hallados.
Si no se pasan argumentos, el programa debe indicar error y cerrarse
Si la ruta no existe. el programa debe señalarlo y finalizar
Extensiones: jpeg, .jpeg, jpg, .xlsx, xlsx, .......

nota: no busca recursivamente dentro de carpetas

Ej: buscar_archivos.py <ruta> <extension>

"""
import os
import sys

# verifico la cantidad de argumentos
if len(sys.argv) != 3:
    print("Error: se debe ejecutar buscar_archivos.py <ruta> <extension>")
    sys.exit()

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

# verifico si la ruta existe
if not os.path.exists(ruta):
    print("No existe la ruta",ruta)
    sys.exit()

# si existe el punto al inicio de la extension, lo elimino
if extension.startswith("."):
    extension = extension.replace(".","")

# guardo la lista de los archivos buscados
lista = []
for archivo in os.listdir(ruta):
    nombre = []
    nombre = archivo.split(".")
    if len(nombre) < 2:
        continue
    if nombre[1] == extension:
        lista.append(archivo)

# imprimo la lista
if lista:
    print(f"Archivos con extension '{extension}' dentro de '{ruta}':")
    for elemento in lista:
        print(elemento)
else:
    print(f"No se hallaron archivos con extension '{extension}' dentro de '{ruta}'")