IFTS11_POO_Python/main.py at main · fasmote/IFTS11_POO_Python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
# main.py
from clases.collection import Collection

def mostrar_menu():
    print("\n--- Base de Datos Documental ---")
    print("1. Crear colección")
    print("2. Importar CSV a colección")
    print("3. Consultar documento en colección")
    print("4. Eliminar documento de colección")
    print("5. Listar todos los documentos en colección")
    print("6. Salir")
    return input("Seleccione una opción: ")

def main():
    db = {}  # Diccionario para almacenar las colecciones

    while True:
        opcion = mostrar_menu()

        if opcion == "1":
            nombre_coleccion = input("Ingrese el nombre de la colección: ")
            db[nombre_coleccion] = Collection(nombre_coleccion)
            print(f"Colección '{nombre_coleccion}' creada.")

        elif opcion == "2":
            nombre_coleccion = input("Ingrese el nombre de la colección: ")
            coleccion = db.get(nombre_coleccion)
            if coleccion:
                ruta_csv = input("Ingrese la ruta del archivo CSV: ")
                try:
                    coleccion.import_csv(ruta_csv)
                    print(f"Datos importados a la colección '{nombre_coleccion}'.")
                except FileNotFoundError:
                    print(f"El archivo '{ruta_csv}' no se encontró. Verifique la ruta.")
                except ValueError as e:
                    print(f"Error al importar datos: {e}")
            else:
                print(f"La colección '{nombre_coleccion}' no existe.")


        elif opcion == "3":
            nombre_coleccion = input("Ingrese el nombre de la colección: ")
            doc_id = int(input("Ingrese el ID del documento: "))
            coleccion = db.get(nombre_coleccion)
            if coleccion:
                documento = coleccion.get_document(doc_id)
                if documento:
                    print("Documento encontrado:")
                    print(documento)
                else:
                    print("Documento no encontrado.")
            else:
                print(f"La colección '{nombre_coleccion}' no existe.")

        elif opcion == "4":
            nombre_coleccion = input("Ingrese el nombre de la colección: ")
            doc_id = int(input("Ingrese el ID del documento a eliminar: "))
            coleccion = db.get(nombre_coleccion)
            if coleccion:
                coleccion.delete_document(doc_id)
                print(f"Documento con ID {doc_id} eliminado.")
            else:
                print(f"La colección '{nombre_coleccion}' no existe.")

        elif opcion == "5":
            nombre_coleccion = input("Ingrese el nombre de la colección: ")
            coleccion = db.get(nombre_coleccion)
            if coleccion:
                documentos = coleccion.list_documents()
                if documentos:
                    print("\n--- Lista de Documentos ---")
                    for doc in documentos.values():
                        print(doc)
                        print("-----------")
                else:
                    print("No hay documentos en la colección.")
            else:
                print(f"La colección '{nombre_coleccion}' no existe.")

        elif opcion == "6":
            print("Saliendo del programa.")
            break

        else:
            print("Opción no válida. Intente nuevamente.")

if __name__ == "__main__":
    main()