From 0d85f89a27fa075dcfd9933f4a506c8c95b191c6 Mon Sep 17 00:00:00 2001
From: Peter Ward <peteraward@gmail.com>
Date: Mon, 19 Nov 2012 18:57:35 +1100
Subject: enable user-defined bot colours

---
 snakegame/engine.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/snakegame/engine.py b/snakegame/engine.py
index 402dc67..ba6c24c 100644
--- a/snakegame/engine.py
+++ b/snakegame/engine.py
@@ -61,7 +61,7 @@ class Engine(object):
             x, y = self.get_random_position()
             self.board[y][x] = common.APPLE
 
-    def add_bot(self, bot, team=None):
+    def add_bot(self, bot, team=None, colour=None):
         """
         A bot is a callable object, with this method signature:
             def bot_callable(
@@ -76,7 +76,8 @@ class Engine(object):
         letter = self.letters.pop()
 
         name = bot.__name__
-        colour = hash_colour(name)
+        if colour is None:
+            colour = hash_colour(name)
 
         position = self.replace_random(common.EMPTY, letter.upper())
         if position is None:
-- 
cgit v1.2.3