testChooseDefenders.py

import time

import arcade
import constants as c
from game import Game


class TestChooseDefenders(arcade.Section):
    def __init__(self, left:int, bottom:int, width:int, height:int):
        super().__init__(left, bottom, width, height, modal=True)
        self.manager = arcade.gui.UIManager()
        self.manager.enable()
        self.chosen_defenders = []
        self.defenders_list = []
        self.v_box = arcade.gui.UIBoxLayout()

        for defender in c.defenders_data.keys():
            self.defenders_list.append(defender)

        while self.defenders_list:
            i = 0
            tmp = arcade.gui.UIBoxLayout(vertical=False)
            for column in range(5):
                if len(self.defenders_list) != 0:
                    defender = c.defenders_data[self.defenders_list[0]]
                    button = arcade.gui.UITextureButton(texture=arcade.load_texture("GUI/"+defender['name']+".png"),
                                                        texture_hovered=arcade.load_texture(
                                                            "GUI/"+defender['name']+"-hover.png"),
                                                        texture_pressed=arcade.load_texture(
                                                            "GUI/"+defender['name']+"-selected.png"),
                                                        width=75, height=75)
                    button.defenderId = self.defenders_list[0]
                    tmp.add(button.with_space_around(bottom=20))
                    button.on_click = self.on_choose
                    self.defenders_list.pop(0)

                else:
                    break
            self.v_box.add(tmp)

        self.manager.add(
            arcade.gui.UIAnchorWidget(
                anchor_x="center_x",
                anchor_y="center_y",
                child=self.v_box)
        )

        texture = arcade.load_texture(":resources:onscreen_controls/flat_dark/start.png")
        self.start_button = arcade.gui.UITextureButton(texture=texture)
        self.start_button.on_click = self.start_game

        self.manager.add(
            arcade.gui.UIAnchorWidget(
                anchor_x="right",
                anchor_y="bottom",
                child=self.start_button)
        )

    def on_draw(self):
        self.manager.draw()

    def on_choose(self, event):
        if event.source.defenderId not in self.chosen_defenders:
            self.chosen_defenders.append(event.source.defenderId)
        else:
            self.chosen_defenders.remove(event.source.defenderId)
        for i in self.chosen_defenders:
            print("CHOOSE DEFENDERS:", i, end=', ')
        print()
        if len(self.chosen_defenders) == c.NUMBER_OF_DEFENDERS:
            # self.start_button.texture.image = ":resources:onscreen_controls/shaded_light/start.png"
            print("ready")

    def start_game(self, event):
        if len(self.chosen_defenders) == c.NUMBER_OF_DEFENDERS:
            self.enabled = False


    def on_show_view(self):
        print("#####IN ONSHOWVIEW CHOOSE_DEFENDERS")