First commit.

Author: Blender3D

.gitignore

@@ -0,0 +1 @@
+*.pyc

board.py

@@ -0,0 +1,105 @@
+import re
+from pieces import Rook, Pawn, King, Knight, Bishop, Queen
+from point import Point
+
+class Board(object):
+  def __init__(self, state=None):
+    self.width = 8
+    self.height = 8
+    
+    self.board = []
+    
+    self.pieces = []
+    self.white_pieces = []
+    self.black_pieces = []
+    
+    self.whites_turn = True
+    
+    if not state:
+      self.loadState('''
+        r n b q k b n r
+        p p p p p p p p
+        . . . . . . . .
+        . . . . . . . .
+        . . . . . . . .
+        . . . . . . . .
+        P P P P P P P P
+        R N B Q K B N R
+      ''')
+    else:
+      self.loadState(state)
+  
+  def copy(self):
+    return Board(str(self))
+  
+  def __getitem__(self, key):
+    if type(key) == int:
+      return self.board[key]
+    elif type(key) == tuple:
+      return self.board[key[1]][key[0]]
+    elif type(key) == Point:
+      return self.board[key.y][key.x]
+    else:
+      raise ValueError('Invalid board coordinate')
+  
+  
+  def __setitem__(self, key, value):
+    if type(key) == int:
+      self.board[key] = value
+    elif type(key) == tuple:
+      self.board[key[1]][key[0]] = value
+    elif type(key) == Point:
+      self.board[key.y][key.x] = value
+    else:
+      raise ValueError('Invalid board coordinate')
+  
+  
+  def loadState(self, state):
+    state = re.split('\n+', re.sub(r'[^prnbqkPRNBQK\.\n]', '', state.strip()))
+    self.state = state
+    
+    board = []
+    
+    for y, row in enumerate(state):
+      new_row = []
+      
+      for x, square in enumerate(row):
+        if square == '.':
+          new_row += [None]
+        else:
+          for piece in [Rook, Pawn, King, Knight, Bishop, Queen]:
+            if piece.character == square.lower():
+              obj = piece(self, not square.islower(), Point(x, y))
+              
+              new_row += [obj]
+              self.pieces += [obj]
+              
+              if not square.islower():
+                self.white_pieces += [obj]
+              else:
+                self.black_pieces += [obj]
+              
+              break
+      
+      board += [new_row]
+    
+    self.board = board
+  
+  
+  def __str__(self):
+    result = []
+    
+    for row in self.board:
+      temp_row = []
+      
+      for cell in row:
+        if cell == None:
+          temp_row += ['.']
+        elif type(cell) == str:
+          temp_row += [cell]
+        else:
+          temp_row += [cell.letter()]
+      
+      result += [' '.join(temp_row)]
+    
+    return '\n'.join(result)

direction.py

@@ -0,0 +1,31 @@
+class Direction(object):
+  def __init__(self, x=None, y=None):
+    self.x = x
+    self.y = y
+  
+  def copy(self):
+    return Direction(self.x, self.y)
+  
+  def flip_y(self):
+    return Direction(self.x, -self.y)
+  
+  def flip_x(self):
+    return Direction(-self.x, self.y)
+  
+  def __add__(self, other):
+    return Direction(self.x + other.x, self.y + other.y)
+  
+  def __sub__(self, other):
+    return Direction(self.x - other.x, self.y - other.y)
+  
+  def __eq__(self, other):
+    return self.x == other.x and self.y == other.y
+  
+  def __ne__(self, other):
+    return self.x != other.x or self.y != other.y
+  
+  def __str__(self):
+    return 'Direction <{x}, {y}>'.format(
+      x=self.x,
+      y=self.y
+    )

main.py

@@ -0,0 +1,21 @@
+import re, sys, random
+
+from board import Board
+
+from colorama import init, Style, Fore
+init()
+
+
+if __name__ == '__main__':
+  board = Board('''
+    r n b q k b n r
+    p p p p p p p p
+    . . . . . . . .
+    . . . . . . . .
+    . . . . . . . .
+    . . . . . . . .
+    P P P P P P P P
+    R N B Q K B N R
+  ''')
+  
+  print random.choice(board.pieces).draw_moves()

piece.py

@@ -0,0 +1,134 @@
+from point import Point
+
+from colorama import init, Style, Fore
+init()
+
+class Piece(object):
+  def __init__(self, board=None, color=True, position=Point()):
+    self.color = color
+    self.color_int = -1 if self.color else 1
+    
+    self.has_moved = False
+    self.is_pinned = False
+    self.board = board
+    
+    self.position = position
+  
+  def moves(self):
+    return self.get_moves(self.move_directions, self.length)
+  
+  def attacks(self):
+    return self.get_attacks(self.move_directions, self.length)
+  
+  def __str__(self):
+    return '{color} {piece} at ({x}, {y})'.format(
+      color='white' if self.color else 'black',
+      piece=self.name,
+      x=self.position.x,
+      y=self.position.y
+    )
+  
+  def get_moves(self, move_directions, length):
+    moves = []
+    
+    for move in move_directions:
+      position = self.position.copy()
+      total_length = 0
+      
+      while total_length < length:
+        position += move
+        total_length += 1
+        
+        if position.in_bounds():
+          square = self.board[position]
+          
+          if square:
+            break
+          else:
+            moves.append(position)
+        else:
+          break
+    
+    return moves
+  
+  def get_attacks(self, move_directions, length):
+    attacks = []
+    
+    for move in move_directions:
+      position = self.position.copy()
+      total_length = 0
+      
+      while total_length < length:
+        position += move
+        total_length += 1
+        
+        if position.in_bounds():
+          square = self.board[position]
+          
+          if square:
+            if square.color != self.color:
+              attacks.append(position)
+            
+            break
+        else:
+          break
+    
+    return attacks
+  
+  def draw_moves(self):
+    temp_board = self.board.copy()
+    
+    temp_board[self.position] = Fore.GREEN + self.letter() + Style.RESET_ALL
+    
+    for move in self.moves():
+      temp_board[move] = Fore.YELLOW + '*' + Style.RESET_ALL
+    
+    for attack in self.attacks():
+      temp_board[attack] = Fore.RED + temp_board[attack].letter() + Style.RESET_ALL
+    
+    return str(temp_board)
+  
+  def letter(self):
+    character = self.character.upper() if self.color else self.character
+    
+    if self.color:
+      character = Style.BRIGHT + character + Style.RESET_ALL
+    
+    return character
+  
+  def can_move(self, x, y):
+    if self.x == x and self.y == y:
+      return False
+    
+    if not self.is_valid((x, y)):
+      return False
+    
+    target = self.board[x, y]
+    
+    if target and target.color == self.color:
+      return False
+    
+    if (x, y) in self.attacks() and self.board[x, y]:
+      return True
+    elif (x, y) in self.moves() and not self.board[x, y]:
+      return True
+    else:
+      return False
+  
+  def move(self, x, y):
+    if self.can_move(x, y):
+      self.board[x, y] = self
+      self.board[self.x, self.y] = None
+      
+      self.x = x
+      self.y = y
+      
+      return True
+    else:
+      return False
+  
+  def __eq__(self, other):
+    return self.letter().lower() == other.letter().lower() if other else False
+  
+  def __ne__(self, other):
+    return self.letter().lower() != other.letter().lower() if other else False