diff --git a/my_music.py b/my_music.py
index 4339d87..434106c 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')  # 下一曲按钮
 
 # 载入音乐
-filelist=["歌曲1.wav","歌曲2.wav","歌曲3.wav","歌曲4.ogg"]
+music_list=["歌曲1.wav","歌曲2.wav","歌曲3.wav","歌曲4.ogg"]
+path="C:\\Users\\Administrator\\Desktop\\test"
+filelist=os.listdir(path)
+for i in filelist:
+    if i[-4:]=="wav" or i[-4:]=="ogg" or i[-4:]=="mp3":
+        music_list.append(i)
 #pygame.mixer.music.load('歌曲4.ogg')  # 载入音乐      
 num=-1
 volume = 0.2