Better event handling.

[naja.git] / naja / widgets / robot.py

diff --git a/naja/widgets/robot.py b/naja/widgets/robot.py
index 60c3f406ed9b8f08460b8c968e5b56d42e8eb907..75082d061395512a96bf4e5bc0a6013363027c6c 100644 (file)
--- a/naja/widgets/robot.py
+++ b/naja/widgets/robot.py
@@ -3,7 +3,7 @@
 import pygame.locals as pgl
 
 from naja.constants import PLAYER_SIZE, BIT_SIZE, TILE_SIZE, BITS, ACT, KEYS
-from naja.events import InvalidateTheWorld, PlayerMoved
+from naja.events import PlayerMoved, finish_event
 from naja.resources import resources
 from naja.resources.mutators import EIGHT_BIT
 from naja.widgets.base import Widget
@@ -45,29 +45,20 @@ class RobotWidget(Widget):
             if ev.key in KEYS.UP:
                 if self.state.player.move(BITS.NORTH):
                     self.state.gameboard.change_mode()
-                    PlayerMoved.post()
-                    InvalidateTheWorld.post()
-                    return True
+                    return finish_event(events=[PlayerMoved])
             if ev.key in KEYS.DOWN:
                 if self.state.player.move(BITS.SOUTH):
                     self.state.gameboard.change_mode()
-                    PlayerMoved.post()
-                    InvalidateTheWorld.post()
-                    return True
+                    return finish_event(events=[PlayerMoved])
             if ev.key in KEYS.LEFT:
                 if self.state.player.move(BITS.WEST):
                     self.state.gameboard.change_mode()
-                    PlayerMoved.post()
-                    InvalidateTheWorld.post()
-                    return True
+                    return finish_event(events=[PlayerMoved])
             if ev.key in KEYS.RIGHT:
                 if self.state.player.move(BITS.EAST):
                     self.state.gameboard.change_mode()
-                    PlayerMoved.post()
-                    InvalidateTheWorld.post()
-                    return True
+                    return finish_event(events=[PlayerMoved])
             if ev.key in (pgl.K_SPACE,):
                 self.state.gameboard.change_mode()
-                InvalidateTheWorld.post()
-                return True
+                return finish_event()
         return super(RobotWidget, self).handle_event(ev)