from pieces import Pawn, Knight, Bishop, Rook, Queen, King, Empty

class Board:
    def __init__(self):
        self.turn = 1 # White starts
        self.board = []
        self.board.append([Rook('b', 1, 8), Knight('b', 2, 8), Bishop('b', 3, 8),
          Queen('b', 4, 8), King('b', 5, 8), Bishop('b', 6, 8), Knight('b', 7, 8),
          Rook('b', 8, 8)])
        self.board.append([Pawn('b', 1, 7), Pawn('b', 2, 7), Pawn('b', 3, 7),
          Pawn('b', 4, 7), Pawn('b', 5, 7), Pawn('b', 6, 7), Pawn('b', 7, 7),
          Pawn('b', 8, 7)])
        for i in range(5):
            self.board.append([Empty(1, 6 - i), Empty(2, 6 - i), Empty(3, 6 - i),
              Empty(4, 6 - i), Empty(5, 6 - i), Empty(6, 6 - i), Empty(7, 6 - i),
              Empty(8, 6 - i)])
        self.board.append([Pawn('w', 1, 2), Pawn('w', 2, 2), Pawn('w', 3, 2),
          Pawn('w', 4, 2), Pawn('w', 5, 2), Pawn('w', 6, 2), Pawn('w', 7, 2),
          Pawn('w', 8, 2)])
        self.board.append([Rook('w', 1, 1), Knight('w', 2, 1), Bishop('w', 3, 1),
          Queen('w', 4, 1), King('w', 5, 1), Bishop('w', 6, 1), Knight('w', 7, 1),
          Rook('w', 8, 1)])

    def __str__(self):
        return_str = ""
        for i in range(len(self.board)):
            for j in range(len(self.board[i])):
                return_str += str(self.board[i][j]) + '\t'
            return_str += '\n'
        return return_str