Функция кнопки странно останавливается при открытии другого окна

Question

from tkinter import *


class MyClass:
    def return_value(self):
        self.value = 5
        self.window2.destroy()

    def my_window(self, parent):
        self.value = 10
        self.window2 = Toplevel(parent)

        Button(self.window2, text="text2", command=self.return_value).pack()

        self.window2.mainloop()
        return self.value

the_class = MyClass()

def button():
    if the_class.my_window(window) == 10:
        print("10")
    else:
        print("5")

window = Tk()
Button(window,text="text",command=button).pack()
window.mainloop()

Если вы запустите это, печать будет вызвана только после того, как все windows будут закрыты, а не когда будет вторичный. Я изучил использование wait_window (), но не увидел, как можно применить его к своему коду

Редактировать: я попытался реализовать update () безуспешно и был бы признателен за пример