from moviepy.editor import VideoFileClip
import os
import argparse
import sys

def extract_audio(video_path, output_path=None, output_format="mp3"):
    """
    從視頻文件中提取音軌并保存為音頻文件

    參數(shù):
        video_path (str): 輸入視頻文件的路徑
        output_path (str, optional): 輸出音頻文件的路徑。如果未提供，則使用輸入文件名+音頻格式
        output_format (str, optional): 輸出音頻格式，默認為 "mp3"

    返回:
        str: 輸出音頻文件的路徑
    """
    try:
        # 檢查輸入視頻文件是否存在
        if not os.path.exists(video_path):
            raise FileNotFoundError(f"視頻文件不存在: {video_path}")

        # 加載視頻文件
        video = VideoFileClip(video_path)
        
        # 如果未提供輸出路徑，則生成默認輸出路徑
        if output_path is None:
            base_name = os.path.splitext(video_path)[0]
            output_path = f"{base_name}.{output_format}"
        
        # 提取音頻
        audio = video.audio
        
        # 保存音頻
        print(f"正在從視頻中提取音軌并保存為 {output_format} 格式...")
        audio.write_audiofile(output_path)
        
        # 關(guān)閉資源
        audio.close()
        video.close()
        
        print(f"音軌提取完成，已保存至: {output_path}")
        return output_path
    
    except Exception as e:
        print(f"提取音軌時出錯: {e}")
        return None

def main():
    """命令行接口的主函數(shù)"""
    parser = argparse.ArgumentParser(description="從視頻中提取音軌")
    parser.add_argument("video_path", help="輸入視頻文件的路徑")
    parser.add_argument("-o", "--output", help="輸出音頻文件的路徑")
    parser.add_argument("-f", "--format", choices=["mp3", "wav", "ogg", "m4a"], 
                        default="mp3", help="輸出音頻格式，默認為 mp3")
    
    args = parser.parse_args()
    
    # 提取音軌
    output_file = extract_audio(args.video_path, args.output, args.format)
    
    if output_file:
        print(f"音軌提取成功: {output_file}")
        sys.exit(0)
    else:
        print("音軌提取失敗")
        sys.exit(1)

if __name__ == "__main__":
    # 如果作為腳本直接運行，則調(diào)用命令行接口
    main()    

pip install moviepy

python video_audio_extractor.py path/to/your/video.mp4

python video_audio_extractor.py path/to/your/video.mp4 -o path/to/output/audio.mp3

python video_audio_extractor.py path/to/your/video.mp4 -f wav