gui.py

import chessboard
from tkinter import *
import tkinter as tk
import chess
from bot import bot_move, checking, is_piece_under_attack
import tkinter.messagebox
from PIL import ImageTk,Image  
from timer import Timer


button_bg = '#A66D4F'
button_font_color = 'black'
button_font_size = 15


class GUI:
    pieces = {}
    selected_piece = None
    focused = None
    images = {}
    color1 = "#DDB88C"
    color2 = "#A66D4F"
    highlightcolor = "khaki"
    rows = 8
    columns = 8
    dim_square = 64

    def __init__(self, parent, chessboard,board,root):
        self.chessboard = chessboard
        self.parent = parent
        # /////////////////////////////////
        self.play_mode = None
        self.counter = 0
        self.board = board
        self.posi1 = None
        self.posi2 = None
        self.root = root
        self.player_color = "white"
        self.bot_color = "black"
        self.focus_piece = None
        self.timer = None
        self.start_timer = False
        self.parent = None
        # ////////////////////////////
        # Adding Top Menu
        # self.menubar = tk.Menu(root)
        # self.filemenu = tk.Menu(self.menubar, tearoff=0)
        # self.filemenu.add_command(label="New Game", command=self.new_game)
        # self.menubar.add_cascade(label="File", menu=self.filemenu)
        # self.root.config(menu=self.menubar)

        # Adding Frame
        self.btmfrm = tk.Frame(parent, height=64)
        self.info_label = tk.Label(self.btmfrm,text="   White to Start the Game  ",fg=self.color2)
        self.info_label.pack(side=tk.RIGHT, padx=8, pady=5)
        self.btmfrm.pack(fill="x", side=tk.BOTTOM)
        canvas_width = self.columns * self.dim_square
        canvas_height = self.rows * self.dim_square
        self.canvas = tk.Canvas(parent, width=canvas_width,
                               height=canvas_height)
        self.canvas.pack(padx=8, pady=8)
        self.draw_board()
        self.canvas.bind("<Button-1>", self.square_clicked)

    def new_game(self):
        # $$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$
        self.board = chess.Board()
        if self.start_timer:
            self.timer.frame.destroy()
            self.timer = None
            self.start_timer = False
            self.parent = None
        # $$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$
        self.chessboard.show(chessboard.START_PATTERN)
        self.draw_board()
        self.draw_pieces()
        self.info_label.config(text="   White to Start the Game  ", fg='red')

    def square_clicked(self, event=None,posi=None):
        if not posi:
            col_size = row_size = self.dim_square
            selected_column = int(event.x / col_size)
            selected_row = 7 - int(event.y / row_size)
        pos = posi or self.chessboard.alpha_notation((selected_row, selected_column))
        try:
            piece = self.chessboard[pos]
            # &&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&
            self.focus_piece = pos
            # &&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&
        except:
            pass
        if self.selected_piece:
            # //////////////////////////
            if self.play_mode == 'single_player' and self.counter == 2:
                self.counter = 0
            # //////////////////////////
            self.shift(self.selected_piece[1], pos)
            self.selected_piece = None
            self.focused = None
            self.pieces = {}
            self.draw_board()
            self.draw_pieces()
            # ///////////////////////////////////////////////
            check = checking(self.board)
            if check == 'White won' or check == "Black won":
                tkinter.messagebox.showinfo("winner",f"{check}")
                self.clicked_restart()
            # ////////////////////////////////////////////
        self.focus(pos)
        self.draw_board()
        #////////////////////////////////////
        if self.chessboard.player_turn == self.bot_color and self.play_mode == 'single_player':
            self.bot()
       #////////////////////////////////// 

    def shift(self, p1, p2):
        piece = self.chessboard[p1]
        try:
            dest_piece = self.chessboard[p2]
        except:
            dest_piece = None
        if dest_piece is None or dest_piece.color != piece.color:
            try:
                self.chessboard.shift(p1, p2)
                # //////////////////
                if self.start_timer:
                    self.timer.frame.destroy()
                    self.clicked_timer(self.parent)
                # ////////////////////
            except chessboard.ChessError as error:
                self.info_label["text"] = error.__class__.__name__
            else:
                turn = ('white' if piece.color == 'black' else 'black')
                self.info_label[
                    "text"] = '' + piece.color.capitalize() + "  :  " + p1 + p2 + '    ' + turn.capitalize() + '\'s turn'
                # ////////////////////////////
                if self.chessboard.player_turn == self.bot_color and self.play_mode == "single_player":
                    move = p1[0].lower()
                    move += p1[1]
                    move += p2[0].lower()
                    move += p2[1]
                    self.board.push_san(move)
                if self.play_mode == "multi_player":
                    move = p1[0].lower()
                    move += p1[1]
                    move += p2[0].lower()
                    move += p2[1]
                    self.board.push_san(move)
                # ///////////////////////////

    def focus(self, pos):
        # /////////////////////
        if pos is not self.focus_piece:
            self.focus_piece = None
        # /////////////////////
        try:
            piece = self.chessboard[pos]
        except:
            piece = None
        if piece is not None and (piece.color == self.chessboard.player_turn):
            self.selected_piece = (self.chessboard[pos], pos)
            self.focused = list(map(self.chessboard.num_notation,
                               (self.chessboard[pos].moves_available(pos))))

    def draw_board(self):
        color = self.color2
        for row in range(self.rows):
            color = self.color1 if color == self.color2 else self.color2
            for col in range(self.columns):
                x1 = (col * self.dim_square)
                y1 = ((7 - row) * self.dim_square)
                x2 = x1 + self.dim_square
                y2 = y1 + self.dim_square
                if (self.focused is not None and (row, col) in self.focused):
                    self.canvas.create_rectangle(x1, y1, x2, y2,
                                                 fill=self.highlightcolor,
                                                 tags="area")
                else:
                    self.canvas.create_rectangle(x1, y1, x2, y2, fill=color,
                                                 tags="area")
                color = self.color1 if color == self.color2 else self.color2
        for name in self.pieces:
            self.pieces[name] = (self.pieces[name][0], self.pieces[name][1])
            x0 = (self.pieces[name][1] * self.dim_square) + int(
                self.dim_square / 2)
            y0 = ((7 - self.pieces[name][0]) * self.dim_square) + int(
                self.dim_square / 2)
            self.canvas.coords(name, x0, y0)
        self.canvas.tag_raise("occupied")
        self.canvas.tag_lower("area")

    def draw_pieces(self):
        self.canvas.delete("occupied")
        for coord, piece in self.chessboard.items():
            x, y = self.chessboard.num_notation(coord)
            if piece is not None:
                filename = "pieces_image/%s%s.png" % (
                piece.shortname.lower(), piece.color)
                piecename = "%s%s%s" % (piece.shortname, x, y)
                if filename not in self.images:
                    self.images[filename] = tk.PhotoImage(file=filename)
                self.canvas.create_image(0, 0, image=self.images[filename],
                                         tags=(piecename, "occupied"),
                                         anchor="c")
                x0 = (y * self.dim_square) + int(self.dim_square / 2)
                y0 = ((7 - x) * self.dim_square) + int(self.dim_square / 2)
                self.canvas.coords(piecename, x0, y0)
    # ///////////////////////////////////////////
    def bot(self):
        if self.counter == 0:
            move = bot_move(self.board,1)
            move = str(move)
            self.board.push_san(move)
            self.posi1 = move[0].upper()
            self.posi1 += move[1] 
            self.posi2 = move[2].upper()
            self.posi2 += move[3] 
            self.counter += 1
            self.square_clicked(None,self.posi1)
        elif self.counter == 1:
            self.counter += 1
            self.square_clicked(None,self.posi2)

    def clicked_restart(self):
        self.new_game()
        
    def clicked_attack(self):
        if self.focus_piece:
            massege,result = is_piece_under_attack(self.board,self.focus_piece.lower())
            if not result :
                tkinter.messagebox.showinfo("Under Attack",f"{self.focus_piece} {massege}")
            else :
                tkinter.messagebox.showinfo("Under Attack",f"{self.focus_piece} {massege} by : {result}")

    def clicked_timer(self,parent):
        self.parent = parent
        self.timer = Timer()
        self.timer.root = parent
        self.timer.main()
    #//////////////////////////////////////////////////// 

def main(chessboard):

    board = chess.Board()
    root = tk.Tk()
    root.title("Chess")
    root.geometry('250x200')
    # &&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&
    main_frame = Frame(root)
    main_frame.grid(row=0, column=0, sticky="nswe")
    chessframe = Frame(main_frame)
    chessframe.grid(row=0, column=0, sticky="nswe")
    left_frame = Frame(main_frame)
    left_frame.grid(row=0, column=1, sticky="nswe")
    game_option_frame = Frame(left_frame)
    game_option_frame.pack_propagate(0)
    game_option_frame.pack(side='top', padx=0, pady=0)
    single_player_option_frame = Frame(main_frame)
    single_player_option_frame.grid(row=0, column=2, sticky="nswe")
    multi_player_option_frame = Frame(main_frame)
    multi_player_option_frame.grid(row=0, column=3, sticky="nswe")
    gui = GUI(chessframe, chessboard,board,root)
    gui.draw_board()
    gui.draw_pieces()

    def clicked_virual():
        pass

    def back():
        root.geometry('250x200')
        gui.clicked_restart()
        chessframe.grid_remove()
        single_player_option_frame.grid_remove()
        left_frame.grid()

    def clicked_draw():
        player = gui.chessboard.player_turn
        if player == 'white':
            winner = "black"
        else:
            winner = "white"
        tkinter.messagebox.showinfo("winner",f"The winner is {winner}")
        gui.clicked_restart()
    
    def clicked_single():
        chessframe.grid()
        single_player_option_frame.grid()
        root.geometry('800x650')
        gui.play_mode = "single_player"
        left_frame.grid_remove()

    def clicked_multi():
        chessframe.grid()
        multi_player_option_frame.grid()
        root.geometry('800x650')
        gui.play_mode = "multi_player"
        left_frame.grid_remove()
    
    def clicked_rules():
        back_ground_color = 'gray'
        text_color_1 = 'black'
        text_color_2 = 'white'
        font_size = 15
        window = tk.Tk()
        window.title("Chess")
        window.geometry('910x300')
        window.configure(bg=back_ground_color)


        lbl = Label(window,font=("Arial Bold", 30),bg=back_ground_color, fg=text_color_2, anchor="e", justify=LEFT, text="chess Rules")
        lbl.grid(column=0, row=0)

        # -----------------------------
        lbl = Label(window,font=("Antique Olive", font_size), fg=text_color_1,bg=back_ground_color, text="King can move exactly one square horizontally, vertically, or diagonally. \n At most once in every game, each king is allowed to make a special move, known as castling.")
        lbl.grid(column=0, row=1)

        lbl = Label(window,font=("Antique Olive", font_size), fg=text_color_2,bg=back_ground_color, text="Queen can move any number of vacant squares diagonally, horizontally, or vertically.")
        lbl.grid(column=0, row=2)

        lbl = Label(window,font=("Antique Olive", font_size), fg=text_color_1,bg=back_ground_color, text="Rook can move any number of vacant squares vertically or horizontally. It also is moved while castling.")
        lbl.grid(column=0, row=3)

        lbl = Label(window,font=("Antique Olive", font_size), fg=text_color_2,bg=back_ground_color, text="Bishop can move any number of vacant squares in any diagonal direction.")
        lbl.grid(column=0, row=4)

        lbl = Label(window,font=("Antique Olive", font_size), fg=text_color_1,bg=back_ground_color, text="Knight can move one square along any rank or file and then at an angle. The knight´s movement \n can also be viewed as an “L” or “7″ laid out at any horizontal or vertical angle.")
        lbl.grid(column=0, row=5)

        lbl = Label(window,font=("Antique Olive", font_size), fg=text_color_2,bg=back_ground_color, text="Pawns can move forward one square, if that square is unoccupied , They can capture an enemy piece \n on either of the two spaces adjacent to the space in front of them")
        lbl.grid(column=0, row=6)
        # -----------------------------

    lbl = Label(game_option_frame, text="                                                                                 ")
    lbl.grid(column=0, row=0)
    btn = Button(game_option_frame,font=("Antique Olive", button_font_size), text="One player", command=clicked_single, bg = button_bg, fg=button_font_color,width=18,height=1)
    btn.grid(column=0, row=1)
    lbl = Label(game_option_frame,font=("Arial Bold", 5),text="       ")
    lbl.grid(column=0, row=2)
    btn_2 = Button(game_option_frame,font=("Antique Olive", button_font_size), text="Multiplayer", command=clicked_multi, bg = button_bg,  fg=button_font_color,width=18,height=1)
    btn_2.grid(column=0, row=3)
    lbl = Label(game_option_frame,font=("Arial Bold", 5),text="       ")
    lbl.grid(column=0, row=4)
    btn_3 = Button(game_option_frame, text="Game rules",font=("Antique Olive", button_font_size), command=clicked_rules, bg = button_bg,  fg=button_font_color,width=18,height=1)
    btn_3.grid(column=0, row=5)

    lbl = Label(single_player_option_frame, text="                                                                                 ")
    lbl.grid(column=0, row=0)
    btn = Button(single_player_option_frame,font=("Antique Olive", button_font_size), text="Restart game", command=gui.clicked_restart, bg=button_bg,  fg=button_font_color,width=18,height=1)
    btn.grid(column=0, row=1)
    lbl = Label(single_player_option_frame,font=("Arial Bold", 5),text="       ")
    lbl.grid(column=0, row=2)
    btn_2 = Button(single_player_option_frame,font=("Antique Olive", button_font_size), text="Pieces under attack", command=gui.clicked_attack, bg=button_bg,  fg=button_font_color,width=18,height=1)
    btn_2.grid(column=0, row=3)
    lbl = Label(single_player_option_frame,font=("Arial Bold", 5),text="       ")
    lbl.grid(column=0, row=4)    
    # btn_3 = Button(single_player_option_frame, text="virual mouse", command=clicked_virual, bg="white", fg="red",width=20,height=3)
    # btn_3.grid(column=0, row=3)
    text1 = tk.StringVar()
    text1.set("choose black")

    def color():
        if text1.get() == "choose black":
            gui.player_color = "black"
            gui.bot_color = "white"
            text1.set("choose white")
            gui.clicked_restart()
            gui.bot()
        else:
            gui.player_color = "white"
            gui.bot_color = "black"
            text1.set("choose black")
            gui.clicked_restart()

    btn_5 = Button(single_player_option_frame,font=("Antique Olive", button_font_size), textvariable=text1, command=color, bg=button_bg, fg=button_font_color,width=18,height=1)
    btn_5.grid(column=0, row=5)
    lbl = Label(single_player_option_frame,font=("Arial Bold", 5),text="       ")
    lbl.grid(column=0, row=6)
    btn_4 = Button(single_player_option_frame,font=("Antique Olive", button_font_size), text="Back", command=back, bg=button_bg,  fg=button_font_color ,width=18,height=1)
    btn_4.grid(column=0, row=7)
    # btn_6 = Button(single_player_option_frame, text="choose white", command=color_white, bg="white", fg="red",width=20,height=3)
    # btn_6.grid(column=0, row=6)

    lbl = Label(multi_player_option_frame, text="                                                                                 ")
    lbl.grid(column=0, row=0)
    btn_1 = Button(multi_player_option_frame,font=("Antique Olive", button_font_size), text="Forfeit ", command=clicked_draw, bg=button_bg,  fg=button_font_color,width=18,height=1)
    btn_1.grid(column=0, row=1)
    lbl = Label(multi_player_option_frame,font=("Arial Bold", 5),text="       ")
    lbl.grid(column=0, row=2)
    text = tk.StringVar()
    text.set("Show timer")

    def start_timer():
        if text.get() == "Show timer":
            text.set("Stop timer")
            gui.start_timer = True
            gui.clicked_timer(root)
        else:
            text.set("Show timer")
            gui.timer.frame.destroy()
            gui.timer = None
            gui.start_timer = False
            gui.parent = None


    btn_2 = Button(multi_player_option_frame,font=("Antique Olive", button_font_size), textvariable=text, command=start_timer, bg = button_bg,  fg=button_font_color,width=18,height=1)
    btn_2.grid(column=0, row=3)
    lbl = Label(multi_player_option_frame,font=("Arial Bold", 5),text="       ")
    lbl.grid(column=0, row=4)
    btn_3 = Button(multi_player_option_frame,font=("Antique Olive", button_font_size), text="Pieces under attack", command=gui.clicked_attack, bg = button_bg,  fg=button_font_color,width=18,height=1)
    btn_3.grid(column=0, row=5)
    # btn_4 = Button(multi_player_option_frame, text="virual mouse", command=clicked_virual, bg="white", fg="red",width=20,height=3)
    # btn_4.grid(column=0, row=4)
    lbl = Label(multi_player_option_frame,font=("Arial Bold", 5),text="       ")
    lbl.grid(column=0, row=6)
    btn_5 = Button(multi_player_option_frame, text="Back",font=("Antique Olive", button_font_size), command=back, bg = button_bg ,  fg=button_font_color,width=18,height=1)
    btn_5.grid(column=0, row=7)

    chessframe.grid_remove()
    single_player_option_frame.grid_remove()
    # &&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&
    root.mainloop()
    
if __name__ == "__main__":
    game = chessboard.Board()
    main(game)