busqueda

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
"""
Script que recibe como argumento la ruta a una carpeta y una extension
para buscar archivos dentro de ella (sin recursividad). Luego debe
mostrar los archivos hallados. Si la ruta no existe, el script lo debe
indicar y finalizar. Si la cantidad de argumentos no es la correcta, el
script lo debe señalar y finalizar el programa
EJ:
> python buscar_archivos C:/Users .py
> inicio.py
> otro.py

> python buscar_archivos .py
> Error: se debe ejecutar "python buscar_archivos.py <ruta> <extension>"

"""

import os
import sys

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

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

# verifico si existe la ruta
if not os.path.exists(ruta):
    print(f"Eror: no existe la ruta '{ruta}'")
    sys.exit()

# si ingresó extension con punto lo elimino
if extension.startswith("."):
    extension = extension.replace(".","")

# imprimo los archivos buscados
hallados = False
for elemento in os.listdir(ruta):
    lista = elemento.split(".")
    if len(lista) == 1:
        continue
    if lista[-1] == extension:
        print(elemento)
        hallados = True

if not hallados:
    print(f"No se encontraron archivos con extension '{extension}' en la ruta '{ruta}'")