diff options
| -rw-r--r-- | snakegame/engine.py | 5 |
1 files 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: |