YearProject/menu.py at master · Sanya239/YearProject · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
import pygame
import os
import buttons.button_class

game_folder = os.path.dirname(__file__)


class main_menu():

    def __init__(self, screen):
        self.general_mode =0
        self.screen = screen
        self.buttons = pygame.sprite.Group()

        self.multy_play_button = buttons.button_class.button()
        self.multy_play_button.rect.center = (400, 100)
        self.buttons.add(self.multy_play_button)

        self.single_play_button = buttons.button_class.button()
        self.single_play_button.rect.center = (400, 200)
        self.buttons.add(self.single_play_button)

        self.settings_button = buttons.button_class.button()
        self.settings_button.rect.center = (400,300)
        self.buttons.add(self.settings_button)

        screen.fill((0, 255, 0))
        self.buttons.draw(screen)


    def update(self,event):
            self.buttons.update(event)
            self.buttons.draw(self.screen)
            if self.multy_play_button.message == True:
                self.general_mode = 1
            elif self.single_play_button.message == True:
                print('aa')
                self.general_mode = 2