fix(video): 改进背景音乐随机选择功能

- 检查背景音乐目录是否存在
-支持 mp3 和 flac两种音频格式
- 增加找不到音乐文件时的错误处理
This commit is contained in:
linyqh 2024-11-16 01:42:10 +08:00
parent 97c3ed1fbe
commit 6b436cba06

View File

@ -25,9 +25,23 @@ def get_bgm_file(bgm_type: str = "random", bgm_file: str = ""):
return bgm_file return bgm_file
if bgm_type == "random": if bgm_type == "random":
suffix = "*.mp3"
song_dir = utils.song_dir() 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 random.choice(files)
return "" return ""