save project

my_music.py

 import pygame
 from pygame import locals
-
+import os 
+path=input('请输入文件地址')
 pygame.init()  # 初始化
 # 创建窗口
 screen = pygame.display.set_mode((640, 480))
@@ -11,13 +12,23 @@ stop_img = pygame.image.load('stop.png')  # 暂停按钮
 last_img = pygame.image.load('last.png')  # 上一曲按钮
 next_img = pygame.image.load('next.png')  # 下一曲按钮
 logo_img = pygame.image.load('logo.png')  # 下一曲按钮
-num=-1      
-musics=['歌曲1.wav','歌曲2.wav','歌曲3.wav','歌曲4.ogg','test.wav']
+num=-1
 volume = 0.2
 pygame.mixer.music.set_volume(volume)  # 初始播放音量
 click = 0
-play_button = stop_img
-
+play_button = stop_img  
+false=0
+try:    
+    musics=[]
+    filelist=os.listdir(path)
+    for i in filelist:
+        if i[-4:]=='.wav'or i[-4:]=='.ogg':
+            musics.append(i)
+except:
+    musics=['歌曲1.wav','歌曲2.wav','歌曲3.wav','歌曲4.ogg','test.wav']        
+    false=1
+else:
+    false=0
 while True:
     for event in pygame.event.get():
         if event.type == locals.QUIT:
@@ -58,9 +69,11 @@ while True:
             num=0
         if num<0:
             num=4
-        pygame.mixer.music.load(musics[num]) 
+        if false==1:
+            pygame.mixer.music.load(musics[num]) 
+        else:
+            pygame.mixer.music.load(path+'\\'+musics[num])
         pygame.mixer.music.play()
-
     # 绘制画面
     screen.blit(bg_img, (0, 0))      # 填充背景
     screen.blit(play_button, (270, 330))   # 暂停按钮