X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=tabakrolletjie%2Fwidgets.py;h=3d12a30840b20cf49c40390a6be5707b8801342b;hb=3dd6b473efa26eff5c02e45a0b405c46c8a52806;hp=8ca16e31510c061a261a5cc39200dd01798f30bf;hpb=e39d0d676be451f3776413051947f6530144c0b7;p=tabakrolletjie.git

diff --git a/tabakrolletjie/widgets.py b/tabakrolletjie/widgets.py
index 8ca16e3..3d12a30 100644
--- a/tabakrolletjie/widgets.py
+++ b/tabakrolletjie/widgets.py
@@ -66,12 +66,18 @@ 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)
         super(ImageButton, self).__init__(self._img.get_size(), name,
                                           pos, padding)
 
+    def update_image(self, *imgparts, **kwargs):
+        transform = kwargs.pop("transform", NullTransform())
+        self._img = loader.load_image(*imgparts, transform=transform)
+        
     def render(self, surface):
         surface.blit(self._img, self._pos, None)