Pygame 媒体播放器：使用 Tkinter 选择文件和全屏功能

import pygame
import tkinter as tk
from tkinter import filedialog
import os
import threading


class PygamePlayer:
    def __init__(self):
        self.screen_size = (800, 600)
        os.environ['SDL_VIDEODRIVER'] = 'dummy'
        pygame.display.init()
        pygame.init()  # 初始化pygame模块
        pygame.display.set_caption('PygamePlayer')  # 设置窗口标题
        self.screen = pygame.display.set_mode(self.screen_size, pygame.NOFRAME)  # 设置pygame显示屏幕
        self.video_surface = pygame.Surface((640, 480))  # 创建视频界面
        self.clock = pygame.time.Clock()  # 创建一个pygame时间对象
        self.duration = 0  # 视频文件总时长，毫秒
        self.current_time = 0  # 当前播放时间，毫秒
        self.paused = True  # 是否暂停
        self.full_screen = False  # 是否全屏播放
        self.loaded = False  # 是否已经载入媒体文件
        self.volume = 0.5  # 声音大小
        self.media_type = None  # 媒体类型，音频或视频

    def load_media(self, path):
        try:
            if os.path.isfile(path):  # 判断路径是不是文件
                self.loaded = True  # 文件已载入标志位
                if path.endswith('.mp4') or path.endswith('.avi') or path.endswith('.mkv'):  # 如果是视频文件
                    self.media_type = 'video'
                    self.media = pygame.movie.Movie(path)  # 加载视频
                    self.media.set_display(self.video_surface, (0, 0, 640, 480))  # 设置视频显示区域
                    self.duration = self.media.get_length()  # 获取视频时长

                elif path.endswith('.mp3') or path.endswith('.wav'):  # 如果是音频文件
                    self.media_type = 'audio'
                    pygame.mixer.music.load(path)  # 加载音频文件
                    self.duration = pygame.mixer.Sound(path).get_length() * 1000  # 获取音频时长，转换为毫秒
        except:
            print('文件加载失败！')
            self.choose_file()

    def play(self):
        if self.media_type == 'video':
            self.media.play()  # 播放视频
            self.paused = False
        elif self.media_type == 'audio':
            pygame.mixer.music.set_volume(self.volume)
            pygame.mixer.music.play()  # 播放音频
            self.paused = False

    def pause(self):
        if self.media_type == 'video':
            self.media.stop()  # 停止视频
            self.paused = True
        elif self.media_type == 'audio':
            pygame.mixer.music.pause()  # 暂停音频
            self.paused = True

    def rewind(self):
        if self.media_type == 'audio':
            current_time = pygame.mixer.music.get_pos() - 10000  # 当前时间减去10秒
            if current_time < 0:
                current_time = 0
            pygame.mixer.music.set_pos(current_time)  # 设置当前播放时间
        elif self.media_type == 'video':
            current_time = self.media.get_time() - 10000
            if current_time < 0:
                current_time = 0
            self.media.set_time(current_time)

    def fast_forward(self):
        if self.media_type == 'audio':
            current_time = pygame.mixer.music.get_pos() + 10000  # 当前时间加上10秒
            if current_time > self.duration:
                current_time = self.duration
            pygame.mixer.music.set_pos(current_time)  # 设置当前播放时间
        elif self.media_type == 'video':
            current_time = self.media.get_time() + 10000
            if current_time > self.duration:
                current_time = self.duration
            self.media.set_time(current_time)

    def toggle_full_screen(self):
        self.full_screen = not self.full_screen  # 切换全屏标志位
        if self.full_screen:
            self.screen = pygame.display.set_mode((0, 0), pygame.FULLSCREEN | pygame.DOUBLEBUF)  # 全屏播放
        else:
            self.screen = pygame.display.set_mode(self.screen_size)  # 恢复原来的窗口大小

    def choose_file(self):
        def choose_file_thread():
            root = tk.Tk()
            root.withdraw()  # 隐藏tk窗口
            file_path = filedialog.askopenfilename()  # 打开文件选择窗口
            if file_path:
                self.load_media(file_path)
                self.play()

        if not self.loaded:
            threading.Thread(target=choose_file_thread).start()  # 打开线程选择文件

    def manage_keys(self, keys):
        if keys[pygame.K_SPACE]:
            if self.paused:
                self.play()
            else:
                self.pause()
        if keys[pygame.K_RIGHT]:
            self.fast_forward()
        if keys[pygame.K_LEFT]:
            self.rewind()
        if keys[pygame.K_ESCAPE]:
            self.toggle_full_screen()
        if keys[pygame.K_q]:
            self.quit()

    def quit(self):
        pygame.quit()
        quit()

    def run(self):
        while True:
            if self.loaded:
                # 处理事件
                for event in pygame.event.get():
                    if event.type == pygame.QUIT:
                        self.quit()
                    if event.type == pygame.KEYDOWN:
                        self.manage_keys(pygame.key.get_pressed())
                    if event.type == pygame.MOUSEBUTTONDOWN:
                        if event.button == 1:
                            self.choose_file()
                # 更新时间
                if self.media_type == 'audio':
                    self.current_time = pygame.mixer.music.get_pos()
                elif self.media_type == 'video':
                    self.current_time = self.media.get_time()
                    if self.media.get_time() >= self.duration:  # 视频播放完毕，重新播放
                        self.media.stop()
                        self.media.play()
                # 绘制
                if self.full_screen:
                    self.video_surface = pygame.transform.scale(self.video_surface, self.screen_size)
                else:
                    self.video_surface = pygame.transform.scale(self.video_surface, (640, 480))
                self.screen.blit(self.video_surface, (0, 0))
                # 在屏幕上绘制当前时间和总时间
                time_text = '%02d:%02d:%02d / %02d:%02d:%02d' % (
                    self.current_time // 3600000, 
                    self.current_time // 60000 % 60, 
                    self.current_time // 1000 % 60, 
                    self.duration // 3600000, 
                    self.duration // 60000 % 60, 
                    self.duration // 1000 % 60
                )
                font = pygame.font.SysFont(None, 30)  # 设置字体和字体大小
                time_surface = font.render(time_text, True, (255, 255, 255))  # 创建时间显示surface
                self.screen.blit(time_surface, (10, 10))  # 在屏幕上绘制时间
                pygame.display.update()  # 更新屏幕
            self.clock.tick(60)  # 设置帧率为60


if __name__ == '__main__':
    player = PygamePlayer()  # 创建播放器对象
    player.run()  # 运行播放器

代码改进：

在 choose_file() 方法中添加了判断，如果用户取消选择文件，则不重新选择。

    def choose_file(self):
        def choose_file_thread():
            root = tk.Tk()
            root.withdraw()  # 隐藏tk窗口
            file_path = filedialog.askopenfilename()  # 打开文件选择窗口
            if file_path:
                self.load_media(file_path)
                self.play()

        if not self.loaded:
            threading.Thread(target=choose_file_thread).start()  # 打开线程选择文件

改进后的代码解决了用户一直选择错误文件导致程序陷入死循环的问题。