diff --git a/main.py b/main.py index 68e51b5..f46329d 100644 --- a/main.py +++ b/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: