import pygame
import pygame_menu
import configparser


def set_difficulty(value, _):
    global difficulty
    difficulty = value[0][0]


def start_the_game():
    config = configparser.ConfigParser()
    config.read('conf/config.ini')
    config['Settings']['difficulty'] = difficulty
    with open('conf/config.ini', 'w') as configfile:
        config.write(configfile)
    menu.toggle()


difficulty = "Простая"
pygame.init()
surface = pygame.display.set_mode((600, 400))
mytheme = pygame_menu.themes.THEME_DARK.copy()
mytheme.background_color = pygame_menu.baseimage.BaseImage(image_path='styles/background.jpg')
menu = pygame_menu.Menu('Welcome', 600, 400, theme=mytheme)
menu.add.label("Сложность")
menu.add.selector('', [('Простая', 1), ('Средняя', 2), ('Сложная', 3), ('Проигрывать весело', 4)],
                  onchange=set_difficulty)
menu.add.button('Play', start_the_game)
menu.add.button('Quit', pygame_menu.events.EXIT)
menu.mainloop(surface)