Factor out flash light into utils.

[naja.git] / naja / utils.py

diff --git a/naja/utils.py b/naja/utils.py
index b15302a1bb01effb9e2a991b843654ea91c8aee0..7ad4f582c2188a9c6fdcc914fa535eddd70fdc3f 100644 (file)
--- a/naja/utils.py
+++ b/naja/utils.py
@@ -27,8 +27,28 @@ def move_glyph(move_name):
     }.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