50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
import shutil
|
||
from pathlib import Path
|
||
from pydub import AudioSegment
|
||
|
||
|
||
def check_dependencies():
|
||
|
||
dependencies = ["ffmpeg", "ffprobe"]
|
||
missing = []
|
||
for dep in dependencies:
|
||
if not shutil.which(dep):
|
||
missing.append(dep)
|
||
if missing:
|
||
print(f"缺少必要依赖: {', '.join(missing)}")
|
||
print("请安装缺失的依赖项。在Ubuntu上可以使用: sudo apt-get install ffmpeg")
|
||
return False
|
||
return True
|
||
|
||
|
||
def convert_wav_to_mp3(source_dir):
|
||
|
||
source_path = Path(source_dir)
|
||
if not source_path.exists() or not source_path.is_dir():
|
||
print(f"源目录不存在或不是一个目录: {source_dir}")
|
||
return False
|
||
wav_files = list(source_path.glob("**/*.wav"))
|
||
if not wav_files:
|
||
print(f"在目录 {source_dir} 中没有找到WAV文件")
|
||
return False
|
||
print(f"找到 {len(wav_files)} 个WAV文件，开始转换...")
|
||
for wav_file in wav_files:
|
||
try:
|
||
mp3_file = wav_file.with_suffix(".mp3")
|
||
sound = AudioSegment.from_wav(str(wav_file))
|
||
sound = sound.set_frame_rate(16000).set_sample_width(2).set_channels(1)
|
||
sound.export(str(mp3_file), format="mp3", bitrate="16k")
|
||
print(f"已转换: {wav_file} -> {mp3_file}")
|
||
except Exception as e:
|
||
print(f"转换文件 {wav_file} 时出错: {str(e)}")
|
||
print("转换完成！")
|
||
return True
|
||
|
||
|
||
if __name__ == "__main__":
|
||
if not check_dependencies():
|
||
print("缺少必要依赖，程序退出")
|
||
exit(1)
|
||
source_directory = "/home/ubuntu/TalkingQ_URL/assets/roles/zhuchi"
|
||
convert_wav_to_mp3(source_directory)
|