def setUp(self):
unittest.TestCase.setUp(self)
self.test_main_game = MainGame()
def run(self):
self.load_music()
self.running = True
self.speed = 2
self.fps = 30
self.clock = pygame.time.Clock()
self.logo = Logo(self.logo_group,
(self.width // 2 - 130, self.height // 4 - 80))
self.draw_logo = True
bg = BackgroundStars(self, 500, (1, 5))
while self.running:
for event in pygame.event.get():
if event.type == pygame.QUIT:
self.running = False
if event.type == pygame.USEREVENT:
if event.user_type == pygame_gui.UI_BUTTON_PRESSED:
if event.ui_element == self.btn_start: # ship choice
if self.music_on:
pygame.mixer.Sound(
'sounds/ButtonMenu.mp3').play()
self.hide_all_btns()
self.draw_logo = False
self.btn_next.show()
self.btn_previous.show()
self.feature_text.show()
self.dropdownmenu_difficulty.show()
self.feature_describe_text.show()
self.label_difficulty.show()
self.start_trigger.show()
self.feature_recomendation_text.show()
self.label_ship_name.show()
self.go_to_menu.show()
self.ship = self.ships[0](self.ships_group,
self.ship_pos)
elif event.ui_element == self.btn_settings: # open settings
if self.music_on:
pygame.mixer.Sound(
'sounds/ButtonMenu.mp3').play()
self.hide_all_btns()
self.dropdownmenu_screen_size.show()
self.btn_return.show()
self.dropdownmenu_sound.show()
self.label_change_screen_size.show()
self.label_change_music_on.show()
elif event.ui_element == self.start_trigger: # start game
game = MainGame(
screen_size=self.screen_size,
music_on=self.music_on,
difficulty=self.difficulty,
ship=self.ship_names[self.now_ship])
self.menu_init()
self.load_music()
elif event.ui_element == self.btn_quit: # quit game
if self.music_on:
pygame.mixer.Sound(
'sounds/ButtonQuit.mp3').play()
pygame.quit()
sys.exit(-1)
elif event.ui_element == self.btn_return: # close settings
if self.music_on:
pygame.mixer.Sound(
'sounds/ButtonQuit.mp3').play()
self.hide_all_btns()
self.btn_start.show()
self.btn_settings.show()
self.btn_quit.show()
self.label_change_screen_size.hide()
self.label_change_music_on.hide()
self.save_settings()
elif event.ui_element in self.dropdownmenu_screen_size.get_focus_set(
): # change screen size data
if self.music_on:
pygame.mixer.Sound(
'sounds/ButtonMenu.mp3').play()
if event.ui_element.text != '▼':
self.screen_size = list(
map(int, event.ui_element.text.split('*')))
elif event.ui_element in self.dropdownmenu_difficulty.get_focus_set(
): # change difficulty
if self.music_on:
pygame.mixer.Sound(
'sounds/ButtonMenu.mp3').play()
if event.ui_element.text != '▼':
self.difficulty = event.ui_element.text
elif event.ui_element in self.dropdownmenu_sound.get_focus_set(
): # change music data
if self.music_on:
pygame.mixer.Sound(
'sounds/ButtonMenu.mp3').play()
if event.ui_element.text == 'On':
self.music_on = True
pygame.mixer.music.play()
else:
self.music_on = False
pygame.mixer.music.pause()
if event.ui_element == self.btn_next: # next ship
if self.music_on:
pygame.mixer.Sound(
'sounds/ButtonMenu.mp3').play()
if self.now_ship + 1 < len(self.ships):
self.now_ship += 1
self.ship.kill()
self.ship = self.ships[self.now_ship](
self.ships_group, self.ship_pos)
self.feature_describe_text.set_text(
self.ship_describe[self.now_ship])
self.feature_recomendation_text.set_text(
self.ship_recommended[self.now_ship])
elif event.ui_element == self.btn_previous: # previous ship
if self.music_on:
pygame.mixer.Sound(
'sounds/ButtonMenu.mp3').play()
if self.now_ship - 1 >= 0:
self.now_ship -= 1
self.ship.kill()
self.ship = self.ships[self.now_ship](
self.ships_group, self.ship_pos)
self.feature_describe_text.set_text(
self.ship_describe[self.now_ship])
self.feature_recomendation_text.set_text(
self.ship_recommended[self.now_ship])
elif event.ui_element == self.go_to_menu:
if self.music_on:
pygame.mixer.Sound(
'sounds/ButtonQuit.mp3').play()
self.menu_init()
text_pos = (self.width // 2 -
self.label_ship_name.rect[2] // 2,
self.height // 5)
self.label_ship_name.rect = pygame.Rect(
text_pos, (self.label_ship_name.rect[2],
self.label_ship_name.rect[3]))
self.label_ship_name.set_text(
self.ships[self.now_ship].get_name(False))
self.manager.process_events(event)
self.manager.update(time_delta=self.clock.tick(self.fps) / 1000.0)
bg.draw()
self.manager.draw_ui(self.screen)
self.ships_group.draw(self.screen)
if self.draw_logo:
self.logo_group.draw(self.screen)
pygame.display.flip()