Color roaches according to their abilities.

[koperkapel.git] / koperkapel / roaches.py

diff --git a/koperkapel/roaches.py b/koperkapel/roaches.py
index 904b8c7dfc9bdf927018cb200372080ce97ca635..20220c4c6da52165651910e691ad354d62c83dde 100644 (file)
--- a/koperkapel/roaches.py
+++ b/koperkapel/roaches.py
@@ -10,6 +10,7 @@ ROACH_COLORS = {
     "blue": (0, 0, 255, 255),
     "green": (0, 255, 0, 255),
     "purple": (255, 0, 255, 255),
+    "brown": (170, 68, 0, 255),
 }
 
 
@@ -37,8 +38,14 @@ class RoachFactory:
         self.suffix = suffix
         self.frames = 4
 
-    def roach_color(self, roach_data):
-        return random.choice(list(ROACH_COLORS.values()))
+    def roach_color(self, roach):
+        if roach.smart:
+            return ROACH_COLORS["blue"]
+        elif roach.fast:
+            return ROACH_COLORS["green"]
+        elif roach.strong:
+            return ROACH_COLORS["purple"]
+        return ROACH_COLORS["brown"]
 
     def assemble_frame(self, i, color, roach_data):
         roach = images.load("roach%s/roach_%d" % (self.suffix, i + 1))