class Gui():
"""Grafical user interface for playing chess"""
font = 'Courier 20'
color_dict = {0: '#b0b0b0', 1: '#ffffff'}
piece_type_dict = {
0: {
0: '\u2654',
1: '\u265a'
},
1: {
0: '\u2655',
1: '\u265b'
},
2: {
0: '\u2656',
1: '\u265c'
},
3: {
0: '\u2657',
1: '\u265d'
},
4: {
0: '\u2658',
1: '\u265e'
},
5: {
0: '\u2659',
1: '\u265f'
}
}
turn_color_dict = {0: 'White', 1: 'Black'}
def __init__(self):
# Init board
self.board = Board()
# Init root
self.root = tk.Tk()
# Create general structure
self.board_frame = tk.Frame(self.root)
self.board_frame.pack()
self.test_frame = tk.Label(self.root, text='Welcome', font=Gui.font)
self.test_frame.pack()
self.user_input = tk.Entry(self.root, font=Gui.font)
self.user_input.configure(state='readonly')
self.user_input.pack()
self.user_input_given = tk.IntVar(master=self.user_input,
name='piece_type',
value=-1)
# Create buttons/fields
self.buttons = [[], [], [], [], [], [], [], []]
self.fields = [[], [], [], [], [], [], [], []]
for x, y in product(range(8), range(8)):
field_color = (x + y) % 2
self.fields[x].append(
tk.Frame(self.board_frame,
height=50,
width=50,
background=Gui.color_dict[field_color]))
self.fields[x][y].propagate(False)
self.fields[x][y].grid(column=x, row=8 - y)
self.buttons[x].append(
tk.Button(self.fields[x][y],
background=Gui.color_dict[field_color],
activebackground='#f2ff00',
borderwidth=0,
font='Courier 30'))
self.buttons[x][y].pack(fill='both', expand=True)
self.draw()
self.select_piece()
self.root.mainloop()
def ask_promotion_type(self):
"""Asks the user which piece to promote"""
self.user_input.bind('<Return>', self.promote2input)
self.test_frame.configure(text='Promote to type:')
self.user_input.configure(state='normal')
self.reset_buttons()
self.user_input_given.set(-1)
self.user_input.wait_variable(name='piece_type')
user_input = self.user_input_given.get()
self.user_input.delete(0, len(self.user_input.get()))
self.user_input.configure(state='readonly')
return user_input
def promote2input(self, event) -> int:
"""Gets the entered text from the entry box"""
type_dict = {
'king': 0,
'queen': 1,
'rook': 2,
'bishop': 3,
'knight': 4,
'pawn': 5
}
promotion_type = type_dict[self.user_input.get()]
# self.board.promote(piece, promotion_type)
self.user_input.bind('<Return>')
self.user_input_given.set(promotion_type)
def select_piece(self):
"""Select piece to move"""
color = self.board.turn_color
for x, rows in enumerate(self.buttons):
for y, button in enumerate(rows):
piece = self.board.find_piece(Position(x, y))
if piece.color == color and \
piece.moves != [] and \
piece.moves != None:
func = partial(self.show_moves, piece)
button.configure(command=func)
def show_moves(self, piece: Piece):
"""Marks the fields where the selected piece can move to"""
self.reset_buttons()
for move in piece.moves:
self.buttons[move.x][move.y].configure(background='#f2ff00',
command=partial(
self.select_move, piece,
move))
def select_move(self, piece: Piece, position):
"""Runs when player selects where to move to"""
self.reset_buttons()
self.board.recalculate(piece, position, self.ask_promotion_type)
self.board.delete_self_check()
self.board.turn_counter += 1
self.board.turn_color = int(not self.board.turn_color)
self.draw()
if self.board.check == True:
self.board.check_mate = \
self.board.ischeckmate(self.board.turn_color)
if self.board.check_mate == True:
self.test_frame.configure(text='Check mate!')
else:
self.test_frame.configure(text='Check!')
else:
message = Gui.turn_color_dict[self.board.turn_color] + \
', it\'s your turn'
self.test_frame.configure(text=message)
self.select_piece()
def draw(self):
"""Draws pieces on the board"""
for (x, y) in product(range(8), range(8)):
piece = self.board.find_piece(Position(x, y))
if piece.color != None:
self.buttons[x][y].config(
text=Gui.piece_type_dict[piece.type][piece.color])
else:
self.buttons[x][y].config(text='')
def reset_buttons(self):
"""Resets the buttons colors and commands"""
for x, y in product(range(8), range(8)):
button = self.buttons[x][y]
button.configure(command=False,
background=Gui.color_dict[(x + y) % 2])
