piece.py

''' piece.py - this class holds attributes of the Piece object. '''

import pygame
class Piece:
    """Contains information about a checkers piece. A piece is a pawn at creation,
    use make_king() to elevate."""

    img_folder = "img/"
    def __init__(self, screen, is_white, pos, width, height, image):
        """Instantiates a piece object with given properties"""

        self.screen = screen
        self.is_king = False
        self.is_white = is_white
        self.is_new_king = False
        self.pos = pos
        self.width = width
        self.height = height
        size = width, height

        self.piece_image = image
        self.piece_image = pygame.transform.scale(self.piece_image, size)
        self.king_image = pygame.image.load(self.img_folder+"king.png")
        self.king_image = pygame.transform.scale(self.king_image, size)

    def make_king(self):
        """Sets this piece as king"""

        self.is_king = True

    def draw(self):
        self.screen.blit(self.piece_image, (self.pos[0]*self.width, (self.pos[1])*self.height))
        if self.is_king:
            self.screen.blit(self.king_image, (self.pos[0]*self.width, (self.pos[1])*self.height))

    def _get_pos(self):
        return self.pos

    def _set_pos(self, pos):
        self.pos = pos

    def _set_is_white(self, is_white):
        print("You can't change the color of a piece!")

    def __repr__(self):
        """Displays the position and type of the piece"""
        if self.is_white:
            color = "white"
        else:
            color = "black"
        return "{} at {}".format(color, self.pos)

    def get_color(self):
        if self.is_white:
            return "white"
        else:
            return "black"