Factor out flash light into utils.

[naja.git] / naja / utils.py

diff --git a/naja/utils.py b/naja/utils.py
index d69cd4ad73f42dfc71a28912a9e751deb3de3129..7ad4f582c2188a9c6fdcc914fa535eddd70fdc3f 100644 (file)
--- a/naja/utils.py
+++ b/naja/utils.py
@@ -25,3 +25,30 @@ def move_glyph(move_name):
         'BISHOP': u'\u265d',
         'KNIGHT': u'\u265e',
     }.get(move_name, move_name.lower())
+
+
+def parse_bits(bit_list):
+    # Convert names to numbers if applicable.
+    return frozenset(BITS.get(bit, bit) for bit in bit_list)
+
+
+def warp_to_game_state(game_state):
+    from naja.events import LoadGameEvent, SceneChangeEvent
+    from naja.scenes.game import GameScene
+    LoadGameEvent.post(game_state)
+    SceneChangeEvent.post(GameScene)
+
+
+class Flashlight(object):
+    def __init__(self, rate):
+        self.rate = rate
+        self.ticks = 0
+        self.on = False
+
+    def tick(self):
+        self.ticks += 1
+        if self.ticks >= self.rate:
+            self.on = not self.on
+            self.ticks = 0
+            return True
+        return False