main_menu.py

import pygame
import sqlite3


def text_format(message, text_font, text_size, text_color):
    new_font = pygame.font.Font(text_font, text_size)
    new_text = new_font.render(message, 0, text_color)
    return new_text


font = "data/fonts/WarPriest.ttf"
font_score = "data/fonts/WarPriestRotalic.ttf"


def main_menu(screen):
    """Начальное меню игры"""

    # Берем из базы данных топ-6 по времени
    con = sqlite3.connect('rating.db')
    cur = con.cursor()
    result = cur.execute("""SELECT time FROM columns;""").fetchall()
    score = []
    for i in result:
        score.append(int(i[0]) / 1000)
    score = sorted(score, reverse=True)
    score = score[:6]


    menu = True
    selected = "start"
    pygame.display.set_caption("Главное меню")
    while menu:
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                pygame.quit()
                quit()
            if event.type == pygame.KEYDOWN:
                if event.key == pygame.K_UP:
                    selected = "start"
                elif event.key == pygame.K_DOWN:
                    selected = "quit"
                if event.key == pygame.K_RETURN:
                    if selected == "start":
                        return
                    if selected == "quit":
                        pygame.quit()
                        quit()

        screen.fill((0, 0, 0))
        title = text_format("Dodging rocks", font, 80, (255, 255, 255))
        if selected == "start":
            text_start = text_format("START", font, 75, (255, 255, 255))
        else:
            text_start = text_format("START", font, 75, (50, 50, 50))
        if selected == "quit":
            text_quit = text_format("QUIT", font, 75, (255, 255, 255))
        else:
            text_quit = text_format("QUIT", font, 75, (50, 50, 50))

        text_score = text_format("rating", font, 50, (255, 255, 255))
        title_rect = title.get_rect()
        start_rect = text_start.get_rect()
        quit_rect = text_quit.get_rect()

        screen.blit(title, (380 - (title_rect[2] / 2), 60))
        screen.blit(text_start, (400 - (start_rect[2] / 2), 300))
        screen.blit(text_quit, (400 - (quit_rect[2] / 2), 360))
        screen.blit(text_score, (30, 200, 100, 100))

        # вывод в меню рейтинг времени
        height_score = 220
        for num in score:
            height_score += 40
            scores = text_format(f"{str(num)} second", font_score, 30, (100, 100, 100))
            screen.blit(scores, (30, height_score, 100, 100))

        pygame.display.update()
        pygame.time.Clock().tick(30)