prevent robot from playing wrong color after illegal move

1 files changed, 12 insertions, 1 deletions

diff --git a/chess.py b/chess.py
index 75cbf91..831d7ce 100644
--- a/chess.py
+++ b/chess.py
@@ -373,7 +373,7 @@ class Gnuchess():
                 return
             elif self._activity.playing_white and \
                  len(self.move_list) % 2 == 1:
-                self._wait_your_turn()
+                self._wait_your_turn() 
                 return
 
         # Only play your color
@@ -493,6 +493,17 @@ class Gnuchess():
                 self._flash_tile([self._xy_to_file_and_rank(
                             self.black[4].get_xy())])
 
+        # Check to see if it is the robot's turn
+        if self._activity.playing_robot and \
+           self._activity.playing_white and \
+           len(self.move_list) % 2 == 0:
+            _logger.debug("not the robot's turn")
+            return True
+        if self._activity.playing_robot and \
+           not self._activity.playing_white and \
+           len(self.move_list) % 2 == 1:
+            _logger.debug("not the robot's turn")
+            return True
         if self._activity.playing_robot and not self.checkmate:
             self._activity.set_thinking_cursor()
             self._activity.status.set_label(_('Thinking...'))