menu.py

import pygame
import sys


class Menu:
    game_type = "easy"
    game_side = "white"
    game_AI = 0

    done = True
    font_menu = pygame.font.SysFont("comicsansms", 50)
    punkt = 0

    window = pygame.display.set_mode((920, 720))
    pygame.display.set_caption(u"Hahki")
    pygame.display.set_icon(pygame.image.load('pic/DBlack.gif').convert())
    mainscreen = pygame.Surface((720, 720))
    rightscreen = pygame.Surface((280, 720))

    def __init__(self, punkts):

        self.i_menu = pygame.image.load('pic/menu.png')
        self.i_rmenu = pygame.image.load('pic/menurightscreen.png')
        self.punkts = punkts

    def render(self, poverhnost, font, num_punkt):
        for e in self.punkts:
            if num_punkt == e[5]:
                poverhnost.blit(font.render(e[2], 1, e[4]), (e[0], e[1]))
            else:
                poverhnost.blit(font.render(e[2], 1, e[3]), (e[0], e[1]))

    def collaide_punkt(self, mp):
        for i in self.punkts:
            if mp[0] > i[0] and mp[0] < (i[0] + 155) and\
                            mp[1] > i[1] and mp[1] < (i[1] + 50):
                self.punkt = i[5]

    def done_with_punkt(self):
        pass

    def run(self):

        while self.done:

            mp = pygame.mouse.get_pos()
            self.collaide_punkt(mp)
            self.render(self.mainscreen, self.font_menu, self.punkt)

            for e in pygame.event.get():
                if e.type == pygame.QUIT:
                    sys.exit()
                if e.type == pygame.KEYDOWN:
                    if e.key == pygame.K_ESCAPE:
                        sys.exit()
                    if e.key == pygame.K_UP:
                        if self.punkt > 0:
                            self.punkt -= 1
                    if e.key == pygame.K_DOWN:
                        if self.punkt < (len(self.punkts) - 1):
                            self.punkt += 1
                if (e.type == pygame.MOUSEBUTTONDOWN and e.button == 1) or \
                        (e.type == pygame.KEYDOWN and e.key == pygame.K_RIGHT):
                    return self.done_with_punkt()

            self.window.blit(self.mainscreen, (0, 0))
            self.window.blit(self.rightscreen, (720, 0))
            self.mainscreen.blit(self.i_menu, (0, 0))
            self.rightscreen.blit(self.i_rmenu, (0, 0))
            pygame.display.flip()