auto save

my_music.py

 import pygame
 from pygame import locals
+import os
 pygame.init()
 screen=pygame.display.set_mode((640,480))
 bg_img=pygame.image.load('background.png')
@@ -12,7 +13,12 @@ pygame.mixer.music.load('歌曲4.ogg')
 volume=0.2
 pygame.mixer.music.set_volume(volume)
 click=0
-music_list=['歌曲1.wav','歌曲2.wav','歌曲3.wav','歌曲4.ogg']
+music_list=[]
+path ='C:\\Users\\XMBC\\Desktop\\test'
+filelist = os.listdir(path)
+for i in filelist:
+    if i[-4:]=='.wav' or i[-4:]=='.ogg':
+        music_list.append(i)
 num=-1
 play_button=stop_img
 while True:
@@ -23,7 +29,7 @@ while True:
             num+=1
             if num> len(music_list)-1:
                 num=0
-                pygame.mixer.music.load(music_list[num])
+            pygame.mixer.music.load(path+'\\'+music_list[num])
             pygame.mixer.music.play()
         if event.type==locals.KEYDOWN:
             if event.key==locals.K_UP: