Colores para la salida de test

Author: Santserrano

engine/__pycache__/arbol.cpython-311.pyc

@@ -1,7 +1,8 @@
 from engine.nodo import Nodo
+from engine.colors import colors
 
 class Arbol:
-    # Funciones privadas
+    
     def __init__(self, dato):
         self.raiz = Nodo(dato)
 
@@ -45,23 +46,21 @@ def __buscar(self, nodo, busqueda):
         else:
             return self.__buscar(nodo.derecha, busqueda)
 
-    # Funciones públicas
-
     def agregar(self, dato):
         self.__agregar_recursivo(self.raiz, dato)
 
     def inorden(self):
-        print("Imprimiendo árbol inorden: ")
+        print(colors.RED + "Imprimiendo árbol inorden: " + colors.RESET)
         self.__inorden_recursivo(self.raiz)
         print("")
 
     def preorden(self):
-        print("Imprimiendo árbol preorden: ")
+        print(colors.YELLOW +"Imprimiendo árbol preorden: " + colors.RESET)
         self.__preorden_recursivo(self.raiz)
         print("")
 
     def postorden(self):
-        print("Imprimiendo árbol postorden: ")
+        print(colors.CYAN +"Imprimiendo árbol postorden: " + colors.RESET)
         self.__postorden_recursivo(self.raiz)
         print("")
 

engine/__pycache__/colors.cpython-311.pyc

@@ -0,0 +1,9 @@
+class colors:
+    RED = '\033[91m'
+    GREEN = '\033[92m'
+    YELLOW = '\033[93m'
+    BLUE = '\033[94m'
+    MAGENTA = '\033[95m'
+    CYAN = '\033[96m'
+    WHITE = '\033[97m'
+    RESET = '\033[0m'

engine/__pycache__/nodo.cpython-311.pyc

@@ -1,6 +1,5 @@
 class Nodo:
     def __init__(self, dato):
-        # "dato" puede ser de cualquier tipo, incluso un objeto si se sobrescriben los operadores de comparación
         self.dato = dato
         self.izquierda = None
         self.derecha = None

engine/arbol.py

@@ -1,14 +1,15 @@
 from engine.arbol import Arbol
+from engine.colors import colors
 
-cantidad = input("Ingresa la cantidad de elementos a ingresar: ")
+cantidad = input(colors.YELLOW + "Ingresa la cantidad de elementos: " + colors.RESET)
 
 for i in range(int(cantidad)):
     if (i==0):
-        nombre = input("Ingresa el elemento raiz: ")
+        nombre = input(colors.GREEN + "Ingresa el elemento raiz: " + colors.RESET)
         arbol = Arbol(nombre)
     else:
-        print(f'[Elemento {i}]')
-        nombre = input("Ingresa el nombre: ")
+        print(colors.GREEN + f'[Elemento {i}]'+ colors.RESET)
+        nombre = input(colors.RED + "Ingresa el nombre: " + colors.RESET)
         arbol.agregar(nombre)
 
 print('...\n')

engine/colors.py

@@ -1,4 +1,5 @@
 from engine.arbol import Arbol
+from engine.colors import colors
 
 lista_nombres = ['emma', 'olivia', 'isabella', 'sofia', 'emily', 'talulah', 'candela']
 
@@ -10,8 +11,8 @@
 arbol.agregar("talulah")
 arbol.agregar("candela")
 
-print(f'Nombres agregados: {lista_nombres}\n')
-print('...Corriendo test\n')
+print(colors.GREEN + f'Nombres agregados: {lista_nombres}\n' + colors.RESET)
+print(colors.GREEN + '...Corriendo test\n' + colors.RESET)
 arbol.preorden()
 arbol.inorden()
 arbol.postorden()