Add command line support for the unicode console GUI

Author: lyudmil-mitev

chesslib/gui_console.py

@@ -0,0 +1,60 @@
+# -*- encoding: utf-8 -*-
+import board
+import os
+
+UNICODE_PIECES = {
+  'r': u'♜', 'n': u'♞', 'b': u'♝', 'q': u'♛',
+  'k': u'♚', 'p': u'♟', 'R': u'♖', 'N': u'♘',
+  'B': u'♗', 'Q': u'♕', 'K': u'♔', 'P': u'♙',
+  None: ' '
+}
+
+class BoardGuiConsole(object):
+    '''
+        Print a text-mode chessboard using the unicode chess pieces
+    '''
+    error = ''
+
+    def __init__(self, chessboard):
+        self.board = chessboard
+
+    def move(self):
+        os.system("clear")
+        self.unicode_representation()
+        print "\n", self.error
+        print "State a move in chess notation (e.g. A2A3). Type \"exit\" to leave:\n", ">>>",
+        self.error = ''
+        coord = raw_input()
+        if coord == "exit":
+            print "Bye."
+            exit(0)
+        try:
+            if len(coord) != 4: raise board.InvalidCoord
+            self.board.move(coord[0:2], coord[2:4])
+            os.system("clear")
+        except board.ChessError as error:
+            self.error = "Error: %s" % error.__class__.__name__
+            
+        self.move()
+
+    def unicode_representation(self):
+        print "\n", ("%s's turn\n" % self.board.player_turn.capitalize()).center(28)
+        for number in self.board.axis_x[::-1]:
+            print " " + str(number) + " ",
+            for letter in self.board.axis_y:
+                piece = self.board[letter+str(number)]
+                if piece is not None:
+                    print UNICODE_PIECES[piece.abbriviation] + ' ',
+                else: print '  ',
+            print "\n"
+        print "    " + "  ".join(self.board.axis_y)
+
+
+def display(board):
+    try:
+        gui = BoardGuiConsole(board)
+        gui.move()
+    except (KeyboardInterrupt, EOFError):
+        os.system("clear")
+        exit(0)
+

game.py

@@ -1,27 +1,32 @@
 #!/usr/bin/env python
 
 from chesslib import board
-from chesslib import boardgui
+
 import os
+import sys
+
+# Load a save if it exists
 
 if os.path.exists("state.fen"):
     with open("state.fen") as save:
-        b = board.Board(save.read())
+        game = board.Board(save.read())
 else:
-    b = board.Board()
-boardgui.display(b)
+    game = board.Board()
 
-## Text Mode
+# Choose display method
+if len(sys.argv) > 1:
+    if sys.argv[1] in ('--console', '-c'):
+        from chesslib.gui_console import display
+        display(game)
+        exit(0)
+    elif sys.argv[1] in ('--help', '-h'):
+        print '''Usage: game.py [OPTION]\n\n\tPlay a game of chess\n\n\tOptions:\n\t -c, --console\tplay in console mode\n\n'''
+        exit(0)
 
-#os.system("clear")
-#board.init()
-#board.unicode_representation()
-
-def move(coord):
-    board.move(coord[0:2], coord[2:4])
-    os.system("clear")
-    board.unicode_representation()
+try:
+    from chesslib.gui_tkinter import display
+except ImportError:
+    from chesslib.gui_console import display
+else:
+    display(game)
 
-def render():
-    os.system("clear")
-    board.unicode_representation()