Evo_war/progress_tab.py - main - Destian/Dev_python_27.09.2025

Evo_war/progress_tab.py
668 строк32 KB
Destian
new icons
10 сен 2024, 14:15
10 сен 2024, 14:154df08a4
import pygame
import time
import os

images_path = os.getcwd() + r'\icons'
images_path_lev1_arm = images_path + r'\progress\armor\1_level'
images_path_lev2_arm = images_path + r'\progress\armor\2_level'
images_path_lev1_sp = images_path + r'\progress\spez\1_level'
images_path_lev2_sp = images_path + r'\progress\spez\2_level'
images_path_lev1_sys = images_path + r'\progress\system\1_level'
images_path_lev2_sys = images_path + r'\progress\system\2_level'


class Button:
    def __init__(self, x, y, width, height, text, icon_path, action):
        self.rect = pygame.Rect(x, y, width, height)
        self.text = text
        self.icon = pygame.image.load(icon_path)
        self.icon = pygame.transform.scale(self.icon, (width - 20, height - 20))
        self.action = action
        self.enabled = True

    def draw(self, screen, font):
        if self.enabled:
            pygame.draw.rect(screen, (222, 184, 135), self.rect)
            pygame.draw.rect(screen, (0, 0, 0), self.rect, 2)
            icon_x = self.rect.x + (self.rect.width - self.icon.get_width()) // 2
            icon_y = self.rect.y + 5
            screen.blit(self.icon, (icon_x, icon_y))
            text_surf = font.render(self.text, True, (0, 0, 0))
            text_rect = text_surf.get_rect(center=(self.rect.centerx, self.rect.bottom - text_surf.get_height() // 2))
            screen.blit(text_surf, text_rect)

    def is_clicked(self, mouse_pos):
        return self.rect.collidepoint(mouse_pos)

    def handle_event(self, event):
        if event.type == pygame.MOUSEBUTTONDOWN and event.button == 1:
            if self.is_clicked(event.pos) and self.enabled:
                self.action()


class TextApplet:
    def __init__(self, x, y, width, height, text):
        self.rect = pygame.Rect(x, y, width, height)
        self.text = text
        self.enabled = True

    def draw(self, screen, font):
        if self.enabled:
            text_surf = font.render(self.text, True, (255, 255, 255))
            text_rect = text_surf.get_rect(topleft=(self.rect.x, self.rect.y))
            screen.blit(text_surf, text_rect)


class ProgressTab:
    def __init__(self, screen_width, screen_height, base):
        self.screen_width = screen_width
        self.screen_height = screen_height
        self.base = base
        self.font = pygame.font.Font(None, 24)
        self.label_text = "Выберите свою специализацию"
        self.show_progress_window = True
        self.armored_vehicles_open = False
        self.speznaz_open = False
        self.system_open = False
        self.arm_unit = []
        self.sys_unit = []
        self.sp_unit = []
        self.message = ""
        self.message_time = 0
        self.all_upgrades_purchased = False
        self.transition_start_time = None
        self.transition_duration = 0.5

        # General buttons
        self.special_forces_button = Button(350, 350, 150, 160, "Спецназ", images_path + "/progress/special_forces.png",
                                            self.special_forces_action)
        self.armored_vehicles_button = Button(550, 350, 150, 160, "Бронетехника", images_path + "/progress/armored_vehicles.png",
                                              self.armored_vehicles_action)
        self.computing_systems_button = Button(750, 350, 150, 160, "Выч. системы",
                                               images_path + "/progress/computing_systems.png", self.computing_systems_action)
        self.close_button = Button(955, 280, 50, 50, "", images_path + "/close.png", self.close_action)

        self.buttons = [self.special_forces_button, self.armored_vehicles_button, self.computing_systems_button,
                        self.close_button]

        # Armored vehicle upgrade buttons and text applets
        self.heavy_weapons_button = Button(350, 370, 60, 60, "", images_path_lev1_arm + "/big_weap.png",
                                           self.heavy_weapons_action)
        self.heavy_armor_button = Button(350, 440, 60, 60, "", images_path_lev1_arm + "/ammo.png",
                                         self.heavy_armor_action)
        self.unique_alloys_button = Button(350, 510, 60, 60, "", images_path_lev1_arm + "/splav.png",
                                           self.unique_alloys_action)
        self.advanced_machines_button = Button(350, 580, 60, 60, "", images_path_lev1_arm + "/stanki.png",
                                               self.advanced_machines_action)

        self.heavy_weapons_applet = TextApplet(420, 390, 200, 50, "Тяжелое оружие: 300 железа, 450 сырья, 500 золота")
        self.heavy_armor_applet = TextApplet(420, 460, 200, 50, "Тяжелая амуниция: 300 железа, 450 сырья, 500 золота")
        self.unique_alloys_applet = TextApplet(420, 530, 200, 50,
                                               "Уникальные сплавы: 300 железа, 450 сырья, 500 золота")
        self.advanced_machines_applet = TextApplet(420, 600, 200, 50,
                                                   "Улучшенные станки: 300 железа, 450 сырья, 500 золота")

        self.armored_buttons_lev1 = [self.heavy_weapons_button, self.heavy_armor_button, self.unique_alloys_button,
                                     self.advanced_machines_button, self.close_button]

        self.armored_applets_lev1 = [self.heavy_weapons_applet, self.heavy_armor_applet, self.unique_alloys_applet,
                                     self.advanced_machines_applet]

        self.dvg_buttons = Button(350, 370, 60, 60, "", images_path_lev2_arm + "/dvg.png", self.dvg_arm)
        self.sbor_buttons = Button(350, 440, 60, 60, "", images_path_lev2_arm + "/sbor.png", self.sbor_arm)
        self.kum_buttons = Button(350, 510, 60, 60, "", images_path_lev2_arm + "/kum.png", self.kum_arm)

        self.dvg_applet = TextApplet(420, 390, 200, 50, "ДВГ: 3500 железо, 1500 сырья, 3000 золота")
        self.sbor_applet = TextApplet(420, 460, 200, 50, "Новая сборочная линия: 4500 железо, 2000 сырья, 4000 золота")
        self.kum_applet = TextApplet(420, 530, 200, 50, "Кумулятивные боеприпасы: 1000 железо, 1500 сырья, 2500 золота")

        self.armored_buttons_lev2 = [self.dvg_buttons, self.sbor_buttons, self.kum_buttons]
        self.armored_applets_lev2 = [self.dvg_applet, self.sbor_applet, self.kum_applet]

        self.sp_ammo_buttons = Button(350, 370, 60, 60, "", images_path_lev1_sp + "/ammo.png", self.sp_ammo)
        self.sp_appar_buttons = Button(350, 440, 60, 60, "", images_path_lev1_sp + "/appar.png", self.sp_appar)
        self.sp_connect_buttons = Button(350, 510, 60, 60, "", images_path_lev1_sp + "/connect.png", self.sp_connect)
        self.sp_ekip_buttons = Button(350, 580, 60, 60, "", images_path_lev1_sp + "/ekip.png", self.sp_ekip)

        self.sp_ammo_applet = TextApplet(420, 390, 200, 50, "Улучшенные винтовки: 400 железа, 900 сырья, 350 золота")
        self.sp_appar_applet = TextApplet(420, 460, 200, 50, "Тепловизоры, ПНВ: 300 железа, 700 сырья, 1500 золота")
        self.sp_connect_applet = TextApplet(420, 530, 200, 50,
                                            "Спутниковая связь: 1000 железа, 1200 сырья, 3000 золота")
        self.sp_ekip_applet = TextApplet(420, 600, 200, 50,
                                         'Экипировка класса "Альфа": 150 железа, 700 сырья, 2600 золота')

        self.sp_buttons_lev1 = [self.sp_ammo_buttons, self.sp_appar_buttons, self.sp_connect_buttons,
                                self.sp_ekip_buttons, self.close_button]
        self.sp_applets_lev1 = [self.sp_ammo_applet, self.sp_appar_applet, self.sp_connect_applet, self.sp_ekip_applet]

        self.sp_aggit_buttons = Button(350, 370, 60, 60, "", images_path_lev2_sp + "/aggitation.png", self.sp_aggit)
        self.sp_bio_buttons = Button(350, 440, 60, 60, "", images_path_lev2_sp + "/bio.png", self.sp_bio)
        self.sp_electro_buttons = Button(350, 510, 60, 60, "", images_path_lev2_sp + "/electro.png", self.sp_electro)

        self.sp_aggit_applet = TextApplet(420, 390, 200, 50, "Аггитация: 100 железа, 1900 сырья, 9000 золота")
        self.sp_bio_applet = TextApplet(420, 460, 200, 50,
                                        "Биологическая модернизация: 50 железа, 4500 сырья, 25000 золота")
        self.sp_electro_applet = TextApplet(420, 530, 200, 50,
                                            "Экспериментальная электроника: 600 железа, 7000 сырья, 40000 золота")

        self.sp_buttons_lev2 = [self.sp_aggit_buttons, self.sp_bio_buttons, self.sp_electro_buttons]
        self.sp_applets_lev2 = [self.sp_aggit_applet, self.sp_bio_applet, self.sp_electro_applet]

        self.sys_fugas_buttons = Button(310, 370, 60, 60, "", images_path_lev1_sys + "/fugas.png", self.sys_fugas)
        self.sys_optika_buttons = Button(310, 440, 60, 60, "", images_path_lev1_sys + "/optika.png", self.sys_optika)
        self.sys_electro_reloaded_buttons = Button(310, 510, 60, 60, "", images_path_lev1_sys + "/electro_reloaded.png",
                                                   self.sys_electro_reloaded)

        self.sys_fugas_applet = TextApplet(390, 390, 200, 50, "Фугасные боеприпасы: 2000 железа, 800 сырья, 500 золота")
        self.sys_optika_applet = TextApplet(390, 460, 200, 50,
                                            "Разработка новой оптики: 1000 железа, 1500 сырья, 800 золота")
        self.sys_electro_reloaded_applet = TextApplet(390, 530, 200, 50,
                                                      "Электрическая система перезарядки: 500 железа, 2000 сырья, 1200 золота")

        self.sys_buttons_lev1 = [self.sys_fugas_buttons, self.sys_optika_buttons, self.sys_electro_reloaded_buttons,
                                 self.close_button]
        self.sys_applets_lev1 = [self.sys_fugas_applet, self.sys_optika_applet, self.sys_electro_reloaded_applet]

        self.sys_giroskop_buttons = Button(310, 370, 60, 60, "", images_path_lev2_sys + "/giroskop.png",
                                           self.sys_giroskop)
        self.sys_bik_system_buttons = Button(310, 440, 60, 60, "", images_path_lev2_sys + "/bik_system.png",
                                             self.sys_bik_system)

        self.sys_giroskop_applet = TextApplet(390, 390, 200, 50,
                                              "Внедрение гироскопа в снаряды: 2400 железа, 3000 сырья, 2300 золота")
        self.sys_bik_system_applet = TextApplet(390, 460, 200, 50,
                                                "Бикалиберная система на гаубицах: 4000 железа, 5000 сырья, 10000 золота")

        self.sys_buttons_lev2 = [self.sys_giroskop_buttons, self.sys_bik_system_buttons]
        self.sys_applets_lev2 = [self.sys_giroskop_applet, self.sys_bik_system_applet]

    def draw_rounded_rect(self, surface, color, rect, radius, border_color=None, border_width=0):
        rect = pygame.Rect(rect)
        color = pygame.Color(*color)
        rectangle = pygame.Surface(rect.size, pygame.SRCALPHA)
        pygame.draw.rect(rectangle, color, rectangle.get_rect(), border_radius=radius)

        if border_color and border_width > 0:
            border_rect = rectangle.get_rect().inflate(-border_width * 2, -border_width * 2)
            pygame.draw.rect(rectangle, border_color, border_rect, border_radius=radius)

        surface.blit(rectangle, rect.topleft)

    def start_transition(self):
        self.transition_start_time = time.time()

    def get_transition_progress(self):
        if self.transition_start_time is None:
            return 1
        progress = (time.time() - self.transition_start_time) / self.transition_duration
        return min(1, progress)

    def draw(self, screen):
        if not self.show_progress_window:
            return

        progress = self.get_transition_progress()
        alpha = int(255 * progress)
        bg_color = (30, 30, 60, alpha)
        border_color = (0, 0, 0)

        self.draw_rounded_rect(screen, bg_color, (315, 320, 640, 250), 20, border_color=bg_color, border_width=2)

        text_surf = self.font.render(self.label_text, True, (255, 255, 255))
        screen.blit(text_surf, (self.screen_width // 2 - text_surf.get_width() // 2, 330))

        if self.armored_vehicles_open:
            self.draw_rounded_rect(screen, bg_color, (270, 320, 740, 350), 20, border_color=bg_color, border_width=2)
            text_surf = self.font.render(self.label_text, True, (255, 255, 255))
            screen.blit(text_surf, (self.screen_width // 2 - text_surf.get_width() // 2, 330))
            for button in self.armored_buttons_lev1:
                button.draw(screen, self.font)

            for applet in self.armored_applets_lev1:
                applet.draw(screen, self.font)

            if self.check_armor_unit_one():
                for button in self.armored_buttons_lev2:
                    button.draw(screen, self.font)

                for applet in self.armored_applets_lev2:
                    applet.draw(screen, self.font)

        elif self.speznaz_open:
            self.draw_rounded_rect(screen, bg_color, (270, 320, 740, 350), 20, border_color=bg_color, border_width=2)
            text_surf = self.font.render(self.label_text, True, (255, 255, 255))
            screen.blit(text_surf, (self.screen_width // 2 - text_surf.get_width() // 2, 330))
            for button in self.sp_buttons_lev1:
                button.draw(screen, self.font)

            for applet in self.sp_applets_lev1:
                applet.draw(screen, self.font)

            if self.check_sp_unit_one():
                for button in self.sp_buttons_lev2:
                    button.draw(screen, self.font)
                for applet in self.sp_applets_lev2:
                    applet.draw(screen, self.font)

        elif self.system_open:
            self.draw_rounded_rect(screen, bg_color, (270, 320, 740, 350), 20, border_color=bg_color, border_width=2)
            text_surf = self.font.render(self.label_text, True, (255, 255, 255))
            screen.blit(text_surf, (self.screen_width // 2 - text_surf.get_width() // 2, 330))
            for button in self.sys_buttons_lev1:
                button.draw(screen, self.font)

            for applet in self.sys_applets_lev1:
                applet.draw(screen, self.font)

            if self.check_sys_unit_one():
                for button in self.sys_buttons_lev2:
                    button.draw(screen, self.font)
                for applet in self.sys_applets_lev2:
                    applet.draw(screen, self.font)

        else:
            for button in self.buttons:
                button.draw(screen, self.font)

        # Отображение временного сообщения
        if self.message:
            message_surf = self.font.render(self.message, True, (255, 255, 255))
            screen.blit(message_surf, (self.screen_width // 2 - message_surf.get_width() // 2, 640))

            if time.time() - self.message_time > 2:
                self.message = ""

        # Отображение статичного сообщения
        if self.all_upgrades_purchased:
            static_message_surf = self.font.render("Все улучшения были приобретены!", True, (255, 255, 255))
            screen.blit(static_message_surf, (self.screen_width // 2 - static_message_surf.get_width() // 2, 450))

    def handle_event(self, event):
        if not self.show_progress_window:
            return

        if self.armored_vehicles_open:
            for button in self.armored_buttons_lev1:
                button.handle_event(event)

            if self.check_armor_unit_one():
                for button in self.armored_buttons_lev2:
                    button.handle_event(event)

        elif self.speznaz_open:
            for button in self.sp_buttons_lev1:
                button.handle_event(event)

            if self.check_sp_unit_one():
                for button in self.sp_buttons_lev2:
                    button.handle_event(event)

        elif self.system_open:
            for button in self.sys_buttons_lev1:
                button.handle_event(event)

            if self.check_sys_unit_one():
                for button in self.sys_buttons_lev2:
                    button.handle_event(event)

        else:
            for button in self.buttons:
                button.handle_event(event)

    def armored_vehicles_action(self):
        if not self.armored_vehicles_open:
            self.armored_vehicles_open = True
            self.label_text = 'Выберите улучшение для Бронетехники'
        else:
            self.armored_vehicles_open = False
            self.label_text = "Выберите свою специализацию"

    def close_action(self):
        if self.heavy_weapons_button.enabled == False or self.heavy_armor_button.enabled == False or self.unique_alloys_button.enabled == False or self.advanced_machines_button.enabled == False \
                or self.sp_ammo_buttons.enabled == False or self.sp_appar_buttons.enabled == False or self.sp_connect_buttons.enabled == False or self.sp_ekip_buttons.enabled == False \
                or self.sys_fugas_buttons.enabled == False or self.sys_optika_buttons.enabled == False or self.sys_electro_reloaded_buttons.enabled == False:
            self.show_progress_window = False
        else:
            self.armored_vehicles_open = False
            self.speznaz_open = False
            self.system_open = False
            self.show_progress_window = False
            self.label_text = "Выберите свою специализацию"

    def heavy_weapons_action(self):
        if self.base.cashe(1, 2, 3):
            self.base.update_resources()
            print("Тяжелое оружие улучшено")
            self.heavy_weapons_button.enabled = False
            self.heavy_weapons_applet.enabled = False
            self.display_message("Тяжелое оружие улучшено!")
            self.arm_unit.append(1)
        else:
            print("Не хватает ресурсов")
        self.update_buttons_state()
        self.check_all_upgrades_purchased()

    def heavy_armor_action(self):
        if self.base.cashe(1, 2, 3):
            self.base.update_resources()
            print("Тяжелая амуниция улучшена")
            self.heavy_armor_button.enabled = False
            self.heavy_armor_applet.enabled = False
            self.display_message("Тяжелая амуниция улучшена!")
            self.arm_unit.append(1)
        else:
            print("Не хватает ресурсов")
        self.update_buttons_state()
        self.check_all_upgrades_purchased()

    def unique_alloys_action(self):
        if self.base.cashe(1, 2, 3):
            self.base.update_resources()
            print("Уникальные сплавы улучшены")
            self.unique_alloys_button.enabled = False
            self.unique_alloys_applet.enabled = False
            self.display_message("Уникальные сплавы улучшены!")
            self.arm_unit.append(1)
        else:
            print("Не хватает ресурсов")
        self.update_buttons_state()
        self.check_all_upgrades_purchased()

    def advanced_machines_action(self):
        if self.base.cashe(1, 2, 3):
            self.base.update_resources()
            print("Улучшенные станки установлены")
            self.advanced_machines_button.enabled = False
            self.advanced_machines_applet.enabled = False
            self.display_message("Улучшенные станки установлены!")
            self.arm_unit.append(1)
            print(f"arm_unit updated: {self.arm_unit}")
        else:
            print("Не хватает ресурсов")
        self.update_buttons_state()
        self.check_all_upgrades_purchased()

    def dvg_arm(self):
        if self.base.cashe(1, 2, 3):
            self.base.update_resources()
            print("Получены двигатели нового поколения")
            self.dvg_buttons.enabled = False
            self.dvg_applet.enabled = False
            self.display_message("Двигатели нового поколения получены!")
            self.arm_unit.append(1)
        else:
            print("Не хватает ресурсов")
        self.check_all_upgrades_purchased()

    def sbor_arm(self):
        if self.base.cashe(1, 2, 3):
            self.base.update_resources()
            print("Построена новая сборочная линия")
            self.sbor_buttons.enabled = False
            self.sbor_applet.enabled = False
            self.display_message("Новая сборочная линия построена!")
            self.arm_unit.append(1)
        else:
            print("Не хватает ресурсов")
        self.check_all_upgrades_purchased()

    def kum_arm(self):
        if self.base.cashe(1, 2, 3):
            self.base.update_resources()
            print("Кумулятивные боеприпасы получены!")
            self.kum_buttons.enabled = False
            self.kum_applet.enabled = False
            self.display_message("Кумулятивные боеприпасы получены!")
            self.arm_unit.append(1)
        else:
            print("Не хватает ресурсов")
        self.check_all_upgrades_purchased()

    def check_armor_unit_one(self):
        return (not self.heavy_weapons_button.enabled and not self.heavy_armor_button.enabled and
                not self.unique_alloys_button.enabled and not self.advanced_machines_button.enabled)

    def check_sp_unit_one(self):
        return (not self.sp_connect_buttons.enabled and not self.sp_ammo_buttons.enabled and
                not self.sp_appar_buttons.enabled and not self.sp_ekip_buttons.enabled)

    def check_armor_unit_two(self):
        return not self.dvg_buttons.enabled and not self.sbor_buttons.enabled and not self.kum_buttons.enabled

    def check_sp_unit_two(self):
        return not self.sp_electro_buttons.enabled and not self.sp_aggit_buttons.enabled and not self.sp_bio_buttons.enabled

    def check_sys_unit_one(self):
        return not self.sys_electro_reloaded_buttons.enabled and not self.sys_optika_buttons.enabled and not self.sys_fugas_buttons.enabled

    def check_sys_unit_two(self):
        return not self.sys_bik_system_buttons.enabled and not self.sys_giroskop_buttons.enabled

    def special_forces_action(self):
        if not self.speznaz_open:
            self.speznaz_open = True
            self.label_text = "Выберите улучшения Спецназа"
        else:
            self.speznaz_open = False
            self.label_text = "Выберите свою специализацию"

    def sp_ammo(self):
        if self.base.cashe(1, 2, 3):
            self.base.update_resources()
            self.sp_ammo_buttons.enabled = False
            self.sp_ammo_applet.enabled = False
            self.display_message("Улучшенные винтовки получены")
            self.sp_unit.append(1)
        else:
            self.display_message("Не хватает ресурсов!")
        self.update_buttons_state()
        self.check_all_upgrades_purchased()

    def sp_appar(self):
        if self.base.cashe(1, 2, 3):
            self.base.update_resources()
            self.sp_appar_buttons.enabled = False
            self.sp_appar_applet.enabled = False
            self.display_message("Полученна новая аппаратура")
            self.sp_unit.append(1)
        else:
            self.display_message("Не хватает ресурсов")
        self.update_buttons_state()
        self.check_all_upgrades_purchased()

    def sp_connect(self):
        if self.base.cashe(1, 2, 3):
            self.base.update_resources()
            self.sp_connect_buttons.enabled = False
            self.sp_connect_applet.enabled = False
            self.display_message("Теперь наш обзор стал шире")
            self.sp_unit.append(1)
        else:
            self.display_message("Не хватает ресурсов")
        self.update_buttons_state()
        self.check_all_upgrades_purchased()

    def sp_ekip(self):
        if self.base.cashe(1, 2, 3):
            self.base.update_resources()
            self.sp_ekip_buttons.enabled = False
            self.sp_ekip_applet.enabled = False
            self.display_message("Теперь нас будут замечать с меньшего расстояния")
            self.sp_unit.append(1)
        else:
            self.display_message("Не хватает ресурсов")
        self.update_buttons_state()
        self.check_all_upgrades_purchased()

    def sp_aggit(self):
        if self.base.cashe(1, 2, 3):
            self.base.update_resources()
            self.sp_aggit_buttons.enabled = False
            self.sp_aggit_applet.enabled = False
            self.display_message("Теперь мы можем саботировать экономику противника")
            self.sp_unit.append(1)
        else:
            self.display_message("Не хватает ресурсов")
        self.check_all_upgrades_purchased()

    def sp_bio(self):
        if self.base.cashe(1, 2, 3):
            self.base.update_resources()
            self.sp_bio_buttons.enabled = False
            self.sp_bio_applet.enabled = False
            self.display_message("Теперь наши солдаты стали значительно крепче")
            self.sp_unit.append(1)
        else:
            self.display_message("Не хватает ресурсов")
        self.check_all_upgrades_purchased()

    def sp_electro(self):
        if self.base.cashe(1, 2, 3):
            self.base.update_resources()
            self.sp_electro_buttons.enabled = False
            self.sp_electro_applet.enabled = False
            self.display_message("Теперь мы получаем данные об их передвижении")
            self.sp_unit.append(1)
        else:
            self.display_message("Не хватает ресурсов")
        self.check_all_upgrades_purchased()

    def computing_systems_action(self):
        if not self.system_open:
            self.system_open = True
            self.label_text = "Выберите улучшения Вычислительных систем"
        else:
            self.system_open = False
            self.label_text = "Выберите свою специализацию"

    def sys_fugas(self):
        if self.base.cashe(1, 2, 3):
            self.base.update_resources()
            self.sys_fugas_buttons.enabled = False
            self.sys_fugas_applet.enabled = False
            self.display_message("Урон по пехоте увеличен в 4.5 раза")
            self.sys_unit.append(1)
        else:
            self.display_message("Не хватает ресурсов")
        self.check_all_upgrades_purchased()

    def sys_optika(self):
        if self.base.cashe(1, 2, 3):
            self.base.update_resources()
            self.sys_optika_buttons.enabled = False
            self.sys_optika_applet.enabled = False
            self.display_message("Дальность обстрела орудий увеличена")
            self.sys_unit.append(1)
        else:
            self.display_message("Не хватает ресурсов")
        self.check_all_upgrades_purchased()

    def sys_electro_reloaded(self):
        if self.base.cashe(1, 2, 3):
            self.base.update_resources()
            self.sys_electro_reloaded_buttons.enabled = False
            self.sys_electro_reloaded_applet.enabled = False
            self.display_message("Увеличена скорость перезарядки орудий")
            self.sys_unit.append(1)
        else:
            self.display_message("Не хватает ресурсов")
        self.check_all_upgrades_purchased()

    def sys_giroskop(self):
        if self.base.cashe(1, 2, 3):
            self.base.update_resources()
            self.sys_giroskop_buttons.enabled = False
            self.sys_giroskop_applet.enabled = False
            self.display_message("Теперь снаряды будут точнее")
            self.sys_unit.append(1)
        else:
            self.display_message("Не хватает ресурсов")
        self.check_all_upgrades_purchased()

    def sys_bik_system(self):
        if self.base.cashe(1, 2, 3):
            self.base.update_resources()
            self.sys_bik_system_buttons.enabled = False
            self.sys_bik_system_applet.enabled = False
            self.display_message("Теперь гаубицы могут стрелять в любую точку карты")
            self.sys_unit.append(1)
        else:
            self.display_message("Не хватает ресурсов")
        self.check_all_upgrades_purchased()

    def display_message(self, text):
        self.message = text
        self.message_time = time.time()

    def open(self):
        self.show_progress_window = True

    def update_buttons_state(self):
        if self.check_armor_unit_one():
            for button in self.armored_buttons_lev2:
                button.enabled = True
            for applet in self.armored_applets_lev2:
                applet.enabled = True
        else:
            for button in self.armored_buttons_lev2:
                button.enabled = False
            for applet in self.armored_applets_lev2:
                applet.enabled = False

        if self.check_sp_unit_one():
            for button in self.sp_buttons_lev2:
                button.enabled = True
            for applet in self.sp_applets_lev2:
                applet.enabled = True

        else:
            for button in self.sp_buttons_lev2:
                button.enabled = False
            for applet in self.sp_applets_lev2:
                applet.enabled = False

        if self.check_sys_unit_one():
            for button in self.sys_buttons_lev2:
                button.enabled = True
            for applet in self.sys_applets_lev2:
                applet.enabled = True

        else:
            for button in self.sys_buttons_lev2:
                button.enabled = False
            for applet in self.sys_applets_lev2:
                applet.enabled = False

    def check_army_armor(self):
        if 3 > len(self.arm_unit) > 0:
            return 1
        elif 5 > len(self.arm_unit) >= 3:
            return 2
        elif len(self.arm_unit) >= 5:
            return 3
        else:
            return 0

    def check_army_sys(self):
        if 2 > len(self.sys_unit) > 0:
            return 1
        elif 4 > len(self.sys_unit) >= 2:
            return 2
        elif len(self.sys_unit) >= 4:
            return 3
        else:
            return 0

    def check_army_sp(self):
        if 2 > len(self.sp_unit) > 0:
            return 1
        elif 4 > len(self.sp_unit) >= 2:
            return 2
        elif len(self.sp_unit) >= 4:
            return 3
        else:
            return 0

    def check_all_upgrades_purchased(self):
        if self.check_armor_unit_one() and self.check_armor_unit_two() or self.check_sp_unit_one() and self.check_sp_unit_two() or self.check_sys_unit_one() and self.check_sys_unit_two():
            self.all_upgrades_purchased = True