当程序员纠结中午应该吃什么，那就用pygame来解决吧

写多了kotlin和android，最近想搞点小东西，于是拿出了长期没有宠爱的python，打算搞个小项目

想想应该写什么，对了，该吃饭了，诶，刚好，写一个能随机选择吃什么的小程序吧，只需要点击按钮，就会随机出现菜谱，然后再点一下，就会得出今天吃什么的结论

思路是这样的，读入一个txt文件，文件中写满想吃的东西，做到数据和代码区分，然后开始写UI，UI通过按钮点击随机展示美食即可

麻辣香锅

糖醋排骨

红烧肉

...

    import pygame
import random

    ​
class App:

        def __init__(self):

            # 初始化 Pygame

            pygame.init()

    ​

            # 创建窗口和设置标题

            self.window_size = (600, 300)

            self.window = pygame.display.set_mode(self.window_size)

            pygame.display.set_caption("What to Eat Today")

    ​

            # 设置字体对象

            self.font = pygame.font.Font('myfont.ttf', 32)

    ​

            # 加载菜单数据

            self.menu = []

            with open("menu.txt", "r") as f:

                for line in f:

                    line = line.strip()

                    if line != "":

                        self.menu.append(line)

                    print(line) # 打印数据

    ​
if __name__ == "__main__":

        app = App()

    ​

运行一下

nice，文件已经读入

这个时候的UI是一闪而过的，因为程序瞬间就执行完毕了，ok，那么我们就需要用一个循环维持UI窗口，然后设置开始选择按键，以及键盘控制按键，同时设置变量

today_menu表示今天吃的东西，

btn_start_stop表示按键文字，

cur_menu表示正处于随机中的美食，当我们按下开始按键时，cur_menu开始变换，当我们按下关闭按键时，cur_menu的数据就赋值到today_menu中，

show_wheel表示当前正处于随机中，还是已经结束了

只要增加一个无限循环，一切就会好起来的

            # 随机选择一道菜

            self.today_menu = ""

            self.btn_start_stop = "start"

            self.cur_menu = ""

    ​

            # 游戏主循环

            self.running = True

            self.show_wheel = False

    ​

    ​

            # 开关程序

            while self.running:

                for event in pygame.event.get():

                    if event.type == pygame.QUIT:

                        self.running = False

                    elif event.type == pygame.MOUSEBUTTONDOWN:

                    ...

    ​

                    # 增加一个elif 按键s，show_wheel为true, 按下q, show_wheel为false

                    elif event.type == pygame.KEYDOWN:

                    ...

运行结果

现在已经有了窗口，接下来需要在上面画东西了

所用到的就是draw函数

        def draw(self):

            # 绘制界面

            self.window.fill((255, 255, 255))

    ​

            # 绘制菜单

            menu_surface = self.font.render(f"Today's Menu: {self.today_menu}", True, (0, 0, 0))

            ...

    ​

            # 绘制按钮

            button_size = min(self.window_size[0] // 4, self.window_size[1] // 4)

            ...

    ​

            btn_start = self.font.render(f"{self.btn_start_stop}", True, (0, 0, 0))

             # 缩小start 文字字号 以适应按钮

            btn_start = pygame.transform.scale(btn_start, (button_size // 3 * 2, button_size // 3 * 2))

            self.window.blit(btn_start, (button_x + button_size // 2 - btn_start.get_width() // 2, button_y + button_size // 2 - btn_start.get_height() // 2))

    ​

            # 滚轮动画

            ...

            pygame.display.update()

运行

上面的代码仅仅能够展示一个静态的页面，

虽然界面平平无奇，似乎只有两行字？不然，实际上暗藏玄🐔，只要我们加上这段，

            # 绘制滚轮动画

            if self.show_wheel:

                wheel_size = min(self.window_size) // 2

                wheel_x = self.window_size[0] // 2 - wheel_size // 2

                wheel_y = self.window_size[1] // 2 - wheel_size // 2

                wheel_rect = pygame.Rect(wheel_x, wheel_y, wheel_size, wheel_size)

                ...

    ​

                # 随机选择并显示菜谱

                menu_index = random.randint(0, len(self.menu) - 1)

                menu_surface = self.font.render(self.menu[menu_index], True, (0, 0, 0))

                self.window.blit(menu_surface, (wheel_x + wheel_size // 2 - menu_surface.get_width() // 2, wheel_y + wheel_size // 2 - menu_surface.get_height() // 2))

                self.cur_menu = self.menu[menu_index]

当我们点击“start"

发现中间的菜谱动了起来，动了起来！都是大家爱吃的，只需要点击右边的stop就可以固定结果！

真正麻烦的在于那个滚轮动画，可以想见，我们需要额外的一个无限循环，每一帧都要修改cur_menu，同时更新动画中的菜谱，然后点击stop后，将cur_menu赋值给到today_menu，最麻烦的不是这些逻辑，而是位置，滚轮动画的位置设置为窗口正中间，然后画了两条线，看起来更好看，有一种，狙击枪瞄准的感觉

最后，进行简单优化，比如设置定时关闭等，全代码如下，如果你也不知道吃什么，就用这段代码 + 在同目录自定义一个txt文件，把自己想吃的写上去吧

import pygame
import random

    ​
class App:

        def __init__(self):

            # 初始化 Pygame

            pygame.init()

    ​

            # 创建窗口和设置标题

            self.window_size = (600, 300)

            self.window = pygame.display.set_mode(self.window_size)

            pygame.display.set_caption("What to Eat Today")

    ​

            # 设置字体对象

            self.font = pygame.font.Font('myfont.ttf', 32)

    ​

            # 加载菜单数据

            self.menu = []

            with open("menu.txt", "r") as f:

                for line in f:

                    line = line.strip()

                    if line != "":

                        self.menu.append(line)

    ​

            # 随机选择一道菜

            self.today_menu = ""

            self.btn_start_stop = "start"

            self.cur_menu = ""

    ​

            # 游戏主循环

            self.running = True

            self.show_wheel = False

            self.wheel_count = 0     # 记录滚轮动画播放的帧数

    ​

    ​

            # 开关程序

            while self.running:

                for event in pygame.event.get():

                    if event.type == pygame.QUIT:

                        self.running = False

                    elif event.type == pygame.MOUSEBUTTONDOWN:

                        if not self.show_wheel:

                            self.show_wheel = True

                            self.wheel_count = 0  # 点击按钮后重置计数器为0

                            self.btn_start_stop = "stop"

                        else:

                            self.show_wheel = False

                            self.today_menu = self.cur_menu  # 点击停止赋值

                            self.btn_start_stop = "start"

    ​

                    # 增加一个elif 按键s，show_wheel为true, 按下q, show_wheel为false

                    elif event.type == pygame.KEYDOWN:

                        if event.key == pygame.K_s:  # 按下 s 键

                            self.show_wheel = True

                            self.wheel_count = 0  # 重置计数器为0

                        elif event.key == pygame.K_q:  # 按下 q 键

                            self.show_wheel = False

                            self.today_menu = self.cur_menu  # 停止赋值

    ​

                self.draw()

    ​

        def draw(self):

            # 绘制界面

            self.window.fill((255, 255, 255))

    ​

            # 绘制菜单

            menu_surface = self.font.render(f"Today's Menu: {self.today_menu}", True, (0, 0, 0))

            menu_x = self.window_size[0] // 2 - menu_surface.get_width() // 2

            menu_y = self.window_size[1] // 2 - menu_surface.get_height() // 2

            self.window.blit(menu_surface, (menu_x, menu_y))

    ​

            # 绘制按钮

            button_size = min(self.window_size[0] // 4, self.window_size[1] // 4)

            button_x = self.window_size[0] - button_size - 20

            button_y = self.window_size[1] - button_size - 20

            button_rect = pygame.Rect(button_x, button_y, button_size, button_size)

            pygame.draw.circle(self.window, (255, 0, 0), (button_x + button_size // 2, button_y + button_size // 2), button_size // 2)

            pygame.draw.rect(self.window, (255, 255, 255), button_rect.inflate(-button_size // 8, -button_size // 8))

    ​

            btn_start = self.font.render(f"{self.btn_start_stop}", True, (0, 0, 0))

             # 缩小start 文字字号 以适应按钮

            btn_start = pygame.transform.scale(btn_start, (button_size // 3 * 2, button_size // 3 * 2))

            self.window.blit(btn_start, (button_x + button_size // 2 - btn_start.get_width() // 2, button_y + button_size // 2 - btn_start.get_height() // 2))

    ​

            # 绘制滚轮动画

            if self.show_wheel:

                wheel_size = min(self.window_size) // 2

                wheel_x = self.window_size[0] // 2 - wheel_size // 2

                wheel_y = self.window_size[1] // 2 - wheel_size // 2

                wheel_rect = pygame.Rect(wheel_x, wheel_y, wheel_size, wheel_size)

                pygame.draw.circle(self.window, (0, 0, 0), (wheel_x + wheel_size // 2, wheel_y + wheel_size // 2), wheel_size // 2)

                pygame.draw.circle(self.window, (255, 255, 255), (wheel_x + wheel_size // 2, wheel_y + wheel_size // 2), wheel_size // 2 - 10)

                pygame.draw.line(self.window, (0, 0, 0), (wheel_x + 10, wheel_y + wheel_size // 2), (wheel_x + wheel_size - 10, wheel_y + wheel_size // 2))

                pygame.draw.line(self.window, (0, 0, 0), (wheel_x + wheel_size // 2, wheel_y + 10), (wheel_x + wheel_size // 2, wheel_y + wheel_size - 10))

    ​

                # 随机选择并显示菜谱

                menu_index = random.randint(0, len(self.menu) - 1)

                menu_surface = self.font.render(self.menu[menu_index], True, (0, 0, 0))

                self.window.blit(menu_surface, (wheel_x + wheel_size // 2 - menu_surface.get_width() // 2, wheel_y + wheel_size // 2 - menu_surface.get_height() // 2))

                self.cur_menu = self.menu[menu_index]

                # 播放一定帧数后停止动画

                self.wheel_count += 1

                if self.wheel_count > 500:

                    self.show_wheel = False

                    self.today_menu = self.cur_menu  # 自动停止赋值

            pygame.display.update()

    ​
if __name__ == "__main__":

        app = App()