from Lectura import Lectura_De_Archivo
from GraficarTabla import GraficarTablas
from Clases import Matriz, Lista
from GraficarMatriz import GraficaMatriz
from GraficarLista import GraficarListas
import os
def menu():
    while True:
        print("-------------------------------------------------------")
        print("----------------------Proyecto 2-----------------------")
        print("Lenguajes Formales y de Programación(796), Sección A-")
        print("Jonathan Alexander Alvarado Fernández, carné: 201903004")
        print("-------------------------------------------------------")
        print("1. Cargar Archivo")
        print("2. Generar Gráfica")
        print("3. Salir")
        entrada = input("Ingrese una opción: ")
        if entrada == "1":
            Ruta = input("Ingrese la ruta del archivo: ")
            if os.path.isfile(Ruta) and Ruta[-4:] == ".lfp":
                Lista, nododefecto, encabezado = Lectura_De_Archivo(Ruta)
            else:
                print("El documento no existe o no es .lfp")
            input()
        if entrada == "2":
            if encabezado != []:
                GraficarTablas(Lista, nododefecto, encabezado)
                print("Gráfica generada con exito")
            else:
                if type(Lista[0]) is Matriz:
                    GraficaMatriz(Lista, nododefecto)
                    print("Gráfica generada con exito")
                elif type(Lista[0]) is Lista:
                    GraficarListas(Lista, nododefecto)
                    print("Gráfica generada con exito")
                pass
            input()
        if entrada == "3":
            print("Gracias, vuelva pronto :)")
            break
if __name__ == '__main__':
    menu()

