From 6b436cba06272652e11eac779e66fde55116febd Mon Sep 17 00:00:00 2001
From: linyqh <linyqemail@163.com>
Date: Sat, 16 Nov 2024 01:42:10 +0800
Subject: [PATCH] =?UTF-8?q?fix(video):=20=E6=94=B9=E8=BF=9B=E8=83=8C?=
 =?UTF-8?q?=E6=99=AF=E9=9F=B3=E4=B9=90=E9=9A=8F=E6=9C=BA=E9=80=89=E6=8B=A9?=
 =?UTF-8?q?=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 检查背景音乐目录是否存在
-支持 mp3 和 flac两种音频格式
- 增加找不到音乐文件时的错误处理
---
 app/services/video.py | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/app/services/video.py b/app/services/video.py
index 23bf644..1d270fa 100644
--- a/app/services/video.py
+++ b/app/services/video.py
@@ -25,9 +25,23 @@ def get_bgm_file(bgm_type: str = "random", bgm_file: str = ""):
         return bgm_file
 
     if bgm_type == "random":
-        suffix = "*.mp3"
         song_dir = utils.song_dir()
-        files = glob.glob(os.path.join(song_dir, suffix))
+
+        # 检查目录是否存在
+        if not os.path.exists(song_dir):
+            logger.warning(f"背景音乐目录不存在: {song_dir}")
+            return ""
+
+        # 支持 mp3 和 flac 格式
+        mp3_files = glob.glob(os.path.join(song_dir, "*.mp3"))
+        flac_files = glob.glob(os.path.join(song_dir, "*.flac"))
+        files = mp3_files + flac_files
+
+        # 检查是否找到音乐文件
+        if not files:
+            logger.warning(f"在目录 {song_dir} 中没有找到 MP3 或 FLAC 文件")
+            return ""
+
         return random.choice(files)
 
     return ""