diff --git a/my_music.py b/my_music.py
index 5076dbb..5d15105 100644
--- a/my_music.py
+++ b/my_music.py
@@ -1,5 +1,6 @@
 import pygame
 from pygame import locals
+import os
 
 pygame.init()  # 初始化
 # 创建窗口
@@ -13,7 +14,12 @@ next_img = pygame.image.load('next.png')  # 下一曲按钮
 logo_img = pygame.image.load('logo.png')  # 下一曲按钮
 
 # 载入音乐
-music_list=['歌曲1.wav','test.wav']
+music_list=[]
+path="D:\\Desktop\\test"
+filelist=os.listdir(path)
+for i in filelist:
+    if i[-4:]==".ogg"or i[-4:]=="wav":
+        music_list.append(i)
 num=-1      
 
 volume = 0.2
@@ -54,7 +60,7 @@ while True:
         num += 1
         if num>len(music_list)-1:
             num=0
-        pygame.mixer.music.load(music_list[num])
+        pygame.mixer.music.load(filelist[num])
         pygame.mixer.music.play()
 
     # 绘制画面