X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Fwidgets%2Fbase.py;h=ee6190b91939bdd35176e11257fc570ac2a78adf;hb=1521f5cdb98df760ed59746901bfbc212c7466a4;hp=133cc532c2ae8fb0886117255decbba65f1e374c;hpb=4e27b4e10d7fd4f66097ece17a8c51b80233a363;p=naja.git

diff --git a/naja/widgets/base.py b/naja/widgets/base.py
index 133cc53..ee6190b 100644
--- a/naja/widgets/base.py
+++ b/naja/widgets/base.py
@@ -1,5 +1,9 @@
+from collections import defaultdict
+
 import pygame
-from naja.events import InvalidateTheWorld
+from pygame import locals as pgl
+
+from naja.events import InvalidateTheWorld, SelectEvent
 
 
 class Widget(object):
@@ -7,6 +11,7 @@ class Widget(object):
         self.pos = pos
         self.size = size or (0, 0)
         self._prepared = False
+        self.callbacks = defaultdict(list)
 
     @property
     def rect(self):
@@ -33,8 +38,18 @@ class Widget(object):
             # True for for this event
             self._prepared = False
             return False
+        if SelectEvent.matches(ev) or ev.type == pgl.MOUSEBUTTONDOWN:
+            return self.callback('click')
         return False
 
+    def add_callback(self, event, callback):
+        self.callbacks[event].append(callback)
+
+    def callback(self, event):
+        for callback in self.callbacks[event]:
+            callback(event)
+        return bool(self.callbacks[event])
+
 
 class Container(object):
     def __init__(self, *widgets):
@@ -50,7 +65,16 @@ class Container(object):
             widget.render(surface)
 
     def handle_event(self, ev):
-        for widget in self.widgets:
-            if widget.handle_event(ev):
-                return True
+        if hasattr(ev, 'pos'):
+            for widget in self.widgets:
+                if isinstance(widget, Container):
+                    if widget.handle_event(ev):
+                        return True
+                elif widget.rect.collidepoint(ev.pos):
+                    if widget.handle_event(ev):
+                        return True
+        else:
+            for widget in self.widgets:
+                if widget.handle_event(ev):
+                    return True
         return False