busc

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
"""
Script que recibe como argumentos la ruta a una carpeta y una extensiòn
y devuelve el nombre de los archivos hallados (sin recursividad). Si no
se pasan los argumentos, debe señalarse el error y terminar el programa.
Si la ruta no existe el script debe indicar el error y terminar el programa
Ej:
>>> python buscar_archivos.py "C:\\Users\\Usuario" ".jpeg"
>>> python buscar_archivos.py "C:\\Users\\Usuario" "jpeg"

"""
import sys
import os

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

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

# verifico que exista la ruta
if not os.path.exists(ruta):
    print(f"Error: no existe la ruta {ruta}")
    sys.exit()

# de existir un punto en la extension lo borro
if extension.startswith("."):
    extension = extension.replace(".", "")

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

# si no encontré ningun archivo, lo imprimo
if not hallados:
    print(f"No se hallaron archivos con extensión '{extension}' en la ruta '{ruta}'")