buscar_archivos

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Programa que saluda
"""
Script que recibe como argumentos una ruta a una carpeta y una extension
para buscar archivos dentro de ella y mostrarlos en pantalla. Si no se pasan
los 2 argumentos, debe mostrarse error y cerrar al programa. En el segundo
argumento se puede pasar un punto (.exe, .xlsx) o no (exe, xlsx). Si la ruta
no existe, el programa debe mostrarlo y terminar
Ej:
>>>python buscar_archivos.py C:/Users/Usuario .xlsx

NOTA: el programa no busca en forma recursiva
"""
import sys
import os

# verificar argumentos
if len(sys.argv) != 3:
    print("""Error: el programa debe usarse como
            buscar_archivos.py <ruta> <extension>
            donde:
            <ruta> es el directorio de búsqueda
            <extension> es el tipo de archivo (jpeg, exe, etc)
            La extensión puede usarse con o sin punto (exe o .exe)
        """)
    sys.exit()

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

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

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

# busco e imprimo la lista de archivos
total = 0
for elemento in os.listdir(ruta):
    if "." in elemento:
        nombre, terminacion = elemento.split(".")
        if extension == terminacion:
            print(elemento)
            total = total + 1
if not total:
    print(f"No existen archivos con la extension {extension} en el directorio {ruta}")