img.py

from PIL import Image
from board import Board
import helper

PIECE_SIZE = 133
PIECES = {
            'p':    Image.open("./assets/PawnB.png"),
            'r':    Image.open("./assets/RookB.png"),
            'n':    Image.open("./assets/KnightB.png"),
            'b':    Image.open("./assets/BishopB.png"),
            'q':    Image.open("./assets/QueenB.png"),
            'k':    Image.open("./assets/KingB.png"),
            'k+':    Image.open("./assets/KingBCheck.png"),
            'k#':    Image.open("./assets/KingBCheckMate.png"),
            'P':    Image.open("./assets/PawnW.png"),
            'R':    Image.open("./assets/RookW.png"),
            'N':    Image.open("./assets/KnightW.png"),
            'B':    Image.open("./assets/BishopW.png"),
            'Q':    Image.open("./assets/QueenW.png"),
            'K':    Image.open("./assets/KingW.png"),
            'K+':    Image.open("./assets/KingWCheck.png"),
            'K#':    Image.open("./assets/KingWCheckMate.png"),
            "Move": Image.open("./assets/Move.png"),
            "Take": Image.open("./assets/Take.png")
        }

def draw_board(b: Board, moves = []) -> None:
    board = b.board_str
    player = b.rules["move"]

    if not board or len(board) != 8:
        Image.open("./assets/BoardError.jpg").save("./assets/RunningBoard.jpg")
        return

    final = Image.open("./assets/Board.jpg")

    if player == "b":
        board = helper.rotate_board(board)
        moves = helper.rotate_moves(moves)
        final = final.rotate(180)

    for i in range(8):
        for j in range(8):
            piece = board[i][j]
            if piece != ".":
                final.paste(PIECES[piece], (PIECE_SIZE*j + 30, PIECE_SIZE*i + 30), PIECES[piece])
                if (i, j) in moves:
                    final.paste(PIECES["Take"], (PIECE_SIZE*j + 30, PIECE_SIZE*i + 30), PIECES["Take"])
            elif (i, j) in moves:
                final.paste(PIECES["Move"], (PIECE_SIZE*j + 30, PIECE_SIZE*i + 30), PIECES["Move"])
            
    
    final.save("./assets/RunningBoard.jpg")


if __name__ == "__main__":
    pass