mirror of
https://github.com/linyqh/NarratoAI.git
synced 2025-12-12 11:22:51 +00:00
fix(video): 改进背景音乐随机选择功能
- 检查背景音乐目录是否存在 -支持 mp3 和 flac两种音频格式 - 增加找不到音乐文件时的错误处理
This commit is contained in:
parent
97c3ed1fbe
commit
6b436cba06
@ -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 ""
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user