start_objects.py

import pygame
import initvar
import placed_objects
from load_images_sounds import *
import logging

log = logging.getLogger(__name__)

START_SPRITES = pygame.sprite.Group()

class StartPiecesBehind(pygame.sprite.Sprite):
    def __init__(self, image_dir, pos):
        pygame.sprite.Sprite.__init__(self)
        self.image = image_dir
        self.rect = self.image.get_rect()
        self.rect.topleft = pos
        START_SPRITES.add(self)
    def draw(self, screen):
        screen.blit(self.image, (self.rect.topleft))
        
StartPiecesBehind(IMAGES["SPR_WHITE_PAWN"], initvar.STARTPOS['white_pawn'])
StartPiecesBehind(IMAGES["SPR_WHITE_BISHOP"], initvar.STARTPOS['white_bishop'])
StartPiecesBehind(IMAGES["SPR_WHITE_KNIGHT"], initvar.STARTPOS['white_knight'])
StartPiecesBehind(IMAGES["SPR_WHITE_ROOK"], initvar.STARTPOS['white_rook'])
StartPiecesBehind(IMAGES["SPR_WHITE_QUEEN"], initvar.STARTPOS['white_queen'])
StartPiecesBehind(IMAGES["SPR_WHITE_KING"], initvar.STARTPOS['white_king'])
StartPiecesBehind(IMAGES["SPR_BLACK_PAWN"], initvar.STARTPOS['black_pawn'])
StartPiecesBehind(IMAGES["SPR_BLACK_BISHOP"], initvar.STARTPOS['black_bishop'])
StartPiecesBehind(IMAGES["SPR_BLACK_KNIGHT"], initvar.STARTPOS['black_knight'])
StartPiecesBehind(IMAGES["SPR_BLACK_ROOK"], initvar.STARTPOS['black_rook'])
StartPiecesBehind(IMAGES["SPR_BLACK_QUEEN"], initvar.STARTPOS['black_queen'])
StartPiecesBehind(IMAGES["SPR_BLACK_KING"], initvar.STARTPOS['black_king'])

class StartPawn(pygame.sprite.Sprite):
    def __init__(self, color, pos):
        pygame.sprite.Sprite.__init__(self)
        self.color = color
        if self.color == "white":
            self.image = IMAGES["SPR_WHITE_PAWN"]
        elif self.color == "black":
            self.image = IMAGES["SPR_BLACK_PAWN"]
        self.rect = self.image.get_rect() 
        self.rect.topleft = pos
        START_SPRITES.add(self)
        self.clickable = True
    def update(self, game_mode):
        if game_mode == 0:
            self.clickable = True
        else:
            self.clickable = False

class StartBishop(pygame.sprite.Sprite):
    def __init__(self, color, pos):
        pygame.sprite.Sprite.__init__(self)
        self.color = color
        if self.color == "white":
            self.image = IMAGES["SPR_WHITE_BISHOP"]
        elif self.color == "black":
            self.image = IMAGES["SPR_BLACK_BISHOP"]
        self.rect = self.image.get_rect()
        self.rect.topleft = pos
        START_SPRITES.add(self)
        self.clickable = True
    def update(self, game_mode):
        if game_mode == 0:
            self.clickable = True
        else:
            self.clickable = False
    
class StartKnight(pygame.sprite.Sprite):
    def __init__(self, color, pos):
        pygame.sprite.Sprite.__init__(self)
        self.color = color
        if self.color == "white":
            self.image = IMAGES["SPR_WHITE_KNIGHT"]
        elif self.color == "black":
            self.image = IMAGES["SPR_BLACK_KNIGHT"]
        self.rect = self.image.get_rect()
        self.rect.topleft = pos
        START_SPRITES.add(self)
        self.clickable = True
    def update(self, game_mode):
        if game_mode == 0:
            self.clickable = True
        else:
            self.clickable = False

class StartRook(pygame.sprite.Sprite):
    def __init__(self, color, pos):
        pygame.sprite.Sprite.__init__(self)
        self.color = color
        if self.color == "white":
            self.image = IMAGES["SPR_WHITE_ROOK"]
        elif self.color == "black":
            self.image = IMAGES["SPR_BLACK_ROOK"]
        self.rect = self.image.get_rect()
        self.rect.topleft = pos
        START_SPRITES.add(self)
        self.clickable = True
    def update(self, game_mode):
        if game_mode == 0:
            self.clickable = True
        else:
            self.clickable = False
    
class StartQueen(pygame.sprite.Sprite):
    def __init__(self, color, pos):
        pygame.sprite.Sprite.__init__(self)
        self.color = color
        if self.color == "white":
            self.image = IMAGES["SPR_WHITE_QUEEN"]
        elif self.color == "black":
            self.image = IMAGES["SPR_BLACK_QUEEN"]
        self.rect = self.image.get_rect()
        self.rect.topleft = pos
        START_SPRITES.add(self)
        self.clickable = True
    def update(self, game_mode):
        if game_mode == 0:
            self.clickable = True
        else:
            self.clickable = False

class StartKing(pygame.sprite.Sprite):
    def __init__(self, color, pos):
        pygame.sprite.Sprite.__init__(self)
        self.color = color
        if self.color == "white":
            self.image = IMAGES["SPR_WHITE_KING"]
        elif self.color == "black":
            self.image = IMAGES["SPR_BLACK_KING"]
        self.rect = self.image.get_rect()
        self.rect.topleft = pos
        START_SPRITES.add(self)
        self.clickable = True
    def update(self, game_mode):
        if game_mode == 0:
            self.clickable = True
        else:
            self.clickable = False
    
class Start():
    start_dict = {}
    start_dict['white_pawn'] = StartPawn("white", initvar.STARTPOS['white_pawn'])
    start_dict['white_bishop'] = StartBishop("white", initvar.STARTPOS['white_bishop'])
    start_dict['white_knight'] = StartKnight("white", initvar.STARTPOS['white_knight'])        
    start_dict['white_rook'] = StartRook("white", initvar.STARTPOS['white_rook'])      
    start_dict['white_queen'] = StartQueen("white", initvar.STARTPOS['white_queen'])      
    start_dict['white_king'] = StartKing("white", initvar.STARTPOS['white_king'])      
    start_dict['black_pawn'] = StartPawn("black", initvar.STARTPOS['black_pawn'])
    start_dict['black_bishop'] = StartBishop("black", initvar.STARTPOS['black_bishop'])
    start_dict['black_knight'] = StartKnight("black", initvar.STARTPOS['black_knight'])      
    start_dict['black_rook'] = StartRook("black", initvar.STARTPOS['black_rook'])      
    start_dict['black_queen'] = StartQueen("black", initvar.STARTPOS['black_queen'])      
    start_dict['black_king'] = StartKing("black", initvar.STARTPOS['black_king'])
    def restart_start_positions():
        Start.start_dict['white_pawn'].rect.topleft = initvar.STARTPOS['white_pawn']
        Start.start_dict['white_bishop'].rect.topleft = initvar.STARTPOS['white_bishop']
        Start.start_dict['white_knight'].rect.topleft = initvar.STARTPOS['white_knight']
        Start.start_dict['white_rook'].rect.topleft = initvar.STARTPOS['white_rook']
        Start.start_dict['white_queen'].rect.topleft = initvar.STARTPOS['white_queen']
        Start.start_dict['white_king'].rect.topleft = initvar.STARTPOS['white_king']
        Start.start_dict['black_pawn'].rect.topleft = initvar.STARTPOS['black_pawn']
        Start.start_dict['black_bishop'].rect.topleft = initvar.STARTPOS['black_bishop']
        Start.start_dict['black_knight'].rect.topleft = initvar.STARTPOS['black_knight']
        Start.start_dict['black_rook'].rect.topleft = initvar.STARTPOS['black_rook']
        Start.start_dict['black_queen'].rect.topleft = initvar.STARTPOS['black_queen']
        Start.start_dict['black_king'].rect.topleft = initvar.STARTPOS['black_king']
        
class Dragging():
    drag_piece_name = ""
    def dragging_all_false():
        Dragging.drag_piece_name = ""
    def start_drag(piece_name):
        Start.restart_start_positions()
        Dragging.dragging_all_false()
        Dragging.drag_piece_name = piece_name
    def update_drag_piece_and_all_start_pieces_positions(mousepos):
        def drag_and_replace_start_obj_image(name_of_piece, mouse_pos):
            if Dragging.drag_piece_name == name_of_piece:
                Start.start_dict[name_of_piece].rect.topleft = (mousepos[0]-(Start.start_dict[name_of_piece].image.get_width()/2),
                                          mousepos[1]-(Start.start_dict[name_of_piece].image.get_height()/2))
            else:
                Start.start_dict[name_of_piece].rect.topleft = initvar.STARTPOS[name_of_piece]
        drag_and_replace_start_obj_image("white_pawn", mousepos)
        drag_and_replace_start_obj_image("white_bishop", mousepos)
        drag_and_replace_start_obj_image("white_knight", mousepos)
        drag_and_replace_start_obj_image("white_rook", mousepos)
        drag_and_replace_start_obj_image("white_queen", mousepos)
        drag_and_replace_start_obj_image("white_king", mousepos)
        drag_and_replace_start_obj_image("black_pawn", mousepos)
        drag_and_replace_start_obj_image("black_bishop", mousepos)
        drag_and_replace_start_obj_image("black_knight", mousepos)
        drag_and_replace_start_obj_image("black_rook", mousepos)
        drag_and_replace_start_obj_image("black_queen", mousepos)
        drag_and_replace_start_obj_image("black_king", mousepos)   
    def dragging_to_placed_no_dups(mouse_coord):
        for piece_list in [placed_objects.PlacedPawn.white_pawn_list, placed_objects.PlacedBishop.white_bishop_list, 
                           placed_objects.PlacedKnight.white_knight_list, placed_objects.PlacedRook.white_rook_list, 
                           placed_objects.PlacedQueen.white_queen_list, placed_objects.PlacedKing.white_king_list,
                           placed_objects.PlacedPawn.black_pawn_list, placed_objects.PlacedBishop.black_bishop_list, 
                           placed_objects.PlacedKnight.black_knight_list, placed_objects.PlacedRook.black_rook_list, 
                           placed_objects.PlacedQueen.black_queen_list, placed_objects.PlacedKing.black_king_list]:
            # If there is already a piece on grid then don't create new Placed object
            for piece in piece_list:
                if piece.coordinate == mouse_coord:
                    return
        # Created Placed objects at the snapped grid location of the piece that's being dragged
        if Dragging.drag_piece_name == "white_pawn":
            if int(mouse_coord[1]) != 1 and int(mouse_coord[1]) != 8:
                placed_objects.PlacedPawn(mouse_coord, "white")
            else:
                log.info("You are not allowed to place a pawn on rank " + mouse_coord[1])
        elif Dragging.drag_piece_name == "white_bishop":
            placed_objects.PlacedBishop(mouse_coord, "white")
        elif Dragging.drag_piece_name == "white_knight":
            placed_objects.PlacedKnight(mouse_coord, "white")
        elif Dragging.drag_piece_name == "white_rook":
            placed_objects.PlacedRook(mouse_coord, "white")
        elif Dragging.drag_piece_name == "white_queen":
            placed_objects.PlacedQueen(mouse_coord, "white")
        elif Dragging.drag_piece_name == "white_king":
            if not placed_objects.PlacedKing.white_king_list:
                placed_objects.PlacedKing(mouse_coord, "white")
            else:
                log.info("You can only have one white king.")
        elif Dragging.drag_piece_name == "black_pawn":
            if int(mouse_coord[1]) != 1 and int(mouse_coord[1]) != 8:
                placed_objects.PlacedPawn(mouse_coord, "black")
            else:
                log.info("You are not allowed to place a pawn on rank " + mouse_coord[1])
        elif Dragging.drag_piece_name == "black_bishop":
            placed_objects.PlacedBishop(mouse_coord, "black")
        elif Dragging.drag_piece_name == "black_knight":
            placed_objects.PlacedKnight(mouse_coord, "black")
        elif Dragging.drag_piece_name == "black_rook":
            placed_objects.PlacedRook(mouse_coord, "black")
        elif Dragging.drag_piece_name == "black_queen":
            placed_objects.PlacedQueen(mouse_coord, "black")
        elif Dragging.drag_piece_name == "black_king":
            if not placed_objects.PlacedKing.black_king_list:
                placed_objects.PlacedKing(mouse_coord, "black")
            else:
                log.info("You can only have one black king.")