Don't forget the last unidad

main.py

@@ -172,13 +172,7 @@ class Sesion:
         for palabra in self.palabras:
             if paso != palabra['paso']:
                 if cur_palabras != False and (quiero_unidad == None or quiero_unidad == unidad_no):
-                    c = [len(cur_palabras[x]) for x in cur_palabras]
-                    print(f"{bcolors.BOLD}{unidad}{bcolors.ENDC}: {paso} ({c[0] + c[6]}/{c[1]}/{c[2]}/{c[3]}/{c[4]}/{bcolors.OKGREEN}{c[5]}{bcolors.ENDC})")
-                    for n in range(5): # 1..4, no 5
-                        if self.hace_palabras(cur_palabras[n], n) == Resultado.ADIOS:
-                            return
-                    n = None
-                    if self.hace_palabras(cur_palabras[n], n) == Resultado.ADIOS:
+                    if self.hace_paso(unidad, paso, cur_palabras) == Resultado.ADIOS:
                         return
 
                 cur_palabras = {0: [], 1: [], 2: [], 3: [], 4: [], 5: [], None: []}
@@ -189,6 +183,20 @@ class Sesion:
 
             cur_palabras[self.status.get(palabra['id'])].append(palabra)
 
+        if cur_palabras != False and (quiero_unidad == None or quiero_unidad == unidad_no):
+            if self.hace_paso(unidad, paso, cur_palabras) == Resultado.ADIOS:
+                return
+
+    def hace_paso(self, unidad, paso, cur_palabras):
+        c = [len(cur_palabras[x]) for x in cur_palabras]
+        print(f"{bcolors.BOLD}{unidad}{bcolors.ENDC}: {paso} ({c[0] + c[6]}/{c[1]}/{c[2]}/{c[3]}/{c[4]}/{bcolors.OKGREEN}{c[5]}{bcolors.ENDC})")
+        for n in range(5): # 1..4, no 5
+            if self.hace_palabras(cur_palabras[n], n) == Resultado.ADIOS:
+                return Resultado.ADIOS
+        n = None
+        if self.hace_palabras(cur_palabras[n], n) == Resultado.ADIOS:
+            return Resultado.ADIOS
+
     def hace_palabras(self, palabras, status):
         random.shuffle(palabras)
         for palabra in palabras: