Finally fixed that stupid bug of not redrawing on winchange

1 files changed, 8 insertions, 3 deletions

diff --git a/asciifarm/client/display/widget.py b/asciifarm/client/display/widget.py
index 2c90547..3d89a9c 100644
--- a/asciifarm/client/display/widget.py
+++ b/asciifarm/client/display/widget.py
@@ -9,13 +9,15 @@ class Widget:
         self.impl.setWidget(self)
         
         self.win = None
+        self.screen = None
         self.changed = False
     
-    def setWin(self, win):
+    def setWin(self, win, screen):
         self.win = win
+        self.screen = screen
     
     def getWin(self):
-        return self.win
+        return self.win and self.screen and self.screen.getWin(self.win)
     
     def getImpl(self):
         return self.impl
@@ -23,8 +25,11 @@ class Widget:
     def change(self):
         self.changed = True
     
+    def isChanged(self):
+        return self.changed
+    
     def update(self):
-        if not self.changed or not self.win:
+        if not self.getWin():
             return
         self.impl.update()
         self.changed = False