save project

diy1.py

 import pygame
 from pygame import locals
+import os
 
 pygame.init()  # 初始化
 # 创建窗口

my_music.py

 import pygame
 from pygame import locals
-
+import os
 pygame.init()  # 初始化
 # 创建窗口
 screen = pygame.display.set_mode((640, 480))
@@ -13,9 +13,15 @@ next_img = pygame.image.load('next.png')  # 下一曲按钮
 logo_img = pygame.image.load('logo.png')  # 下一曲按钮
 
 # 载入音乐   
-filelist = ['歌曲1.wav','歌曲2.wav','歌曲3.wav','歌曲1.ogg']
-unm = 0
+file_list = []
+path = "C:\\Users\\Administrator\\Desktop\\test"
+filelist = os.listdir(path)
+for i in filelist:
+    if i[-4:] == ".ogg" or i[-4:] == ".wav":
+        file_list.append(i)
+
 
+unm = -1
 volume = 0.2
 pygame.mixer.music.set_volume(volume)  # 初始播放音量
 click = 0