import osimport ffmpeg
from ffmpeg import output, rundef convert_wemtv_to_mp4(input_file, output_dir):"""将 WEMTV 文件转换为 MP4 文件并保存Args:input_file (str): 输入 WEMTV 文件路径output_dir (str): 输出 MP4 文件保存目录"""try:if  os.path.exists(input_file):print(f"input_file:存在")# 检查输出目录是否存在if not os.path.exists(output_dir):os.makedirs(output_dir)print(f"output_dir:不存在")# 构建输出文件路径output_file = os.path.join(output_dir, output_dir + ".mp4")print(f"input_file:{input_file}")# 创建 FFmpeg 处理流stream = input(input_file)print(f"stream:{stream}")stream = output(stream, output_file, codec='libx264', f='mp4')# 执行转换run(stream)print(f"转换成功: {output_file}")except Exception as e:print(f"转换失败: {input_file}")print(e)# 示例用法
input_file = 'path/to/input.wemtv'
output_file = 'path/to/output.mp4'
output_dir = 'directory'
convert_wemtv_to_mp4(input_file, output_dir)