X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=tabakrolletjie%2Fwidgets.py;h=2678e72c46cce18b390686ee079dda184ee1e56b;hb=ed21b64839c0b4f7f3309756e30ff34b2ff97571;hp=b1917e5abaf013f0bddef7c1391a99f15a40674a;hpb=5a95f8b901e16c7db4bf22d6e55955c95ee7e866;p=tabakrolletjie.git

diff --git a/tabakrolletjie/widgets.py b/tabakrolletjie/widgets.py
index b1917e5..2678e72 100644
--- a/tabakrolletjie/widgets.py
+++ b/tabakrolletjie/widgets.py
@@ -66,7 +66,9 @@ class ImageButton(Button):
 
     def __init__(self, *imgparts, **kwargs):
         transform = kwargs.pop("transform", NullTransform())
-        self._img = loader.load_image(*imgparts, transform=transform)
+        self._img = kwargs.pop("image", None)
+        if self._img is None:
+            self._img = loader.load_image(*imgparts, transform=transform)
         name = kwargs.get('name')
         pos = kwargs.get('pos')
         padding = kwargs.get('padding', 0)
@@ -74,8 +76,8 @@ class ImageButton(Button):
                                           pos, padding)
 
     def update_image(self, *imgparts, **kwargs):
-        transform=kwargs.pop("transform", NullTransform())
+        transform = kwargs.pop("transform", NullTransform())
         self._img = loader.load_image(*imgparts, transform=transform)
-        
+
     def render(self, surface):
         surface.blit(self._img, self._pos, None)