Python/FFmpegConvertOgg.py at main · Leet-Lyn/Python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
# 请帮我写个中文的 Python 脚本，批注也是中文：
# 在脚本开始前询问我源文件夹位置（默认“d:\\Works\\Ins\\”）与目标文件夹位置（默认“d:\\Works\\Outs\\”）。
# 遍历源文件夹及其子文件夹位置中所有音频文件（mp3、m4a、wma、ogg、aac、ac3、rm、wav）。
# 使用 ffmpeg 压缩。
# 音频参数为：ogg 格式，遍历每个音轨，质量模式。q=4。
# 生成的文件重新用 mkvmerge 再生成同名文件到 目标文件夹位置，文件夹结构保持一致。

# 导入模块
import os
import subprocess

# 函数用于询问用户文件夹位置
def ask_folder_location(prompt, default_folder):
    """
    提示用户输入文件夹路径，如果按回车则使用默认路径。
    :param prompt: 提示信息
    :param default_folder: 默认文件夹路径
    :return: 有效的文件夹路径
    """
    folder_path = input(f"{prompt}（按回车键使用默认位置 {default_folder}）: ").strip() or default_folder
    while not os.path.isdir(folder_path):
        print("文件夹不存在，请重新输入。")
        folder_path = input(f"{prompt}（按回车键使用默认位置 {default_folder}）: ").strip() or default_folder
    return folder_path

# 获取源文件夹和目标文件夹位置
source_folder = ask_folder_location("请输入源文件夹位置", "d:\\Works\\Ins\\")
target_folder = ask_folder_location("请输入目标文件夹位置", "d:\\Works\\Outs\\")

# 支持的音频格式
audio_formats = (".mp3", ".m4a", ".wma", ".ogg", ".aac", ".ac3", ".rm", ".wav")

# 确保目标文件夹存在
os.makedirs(target_folder, exist_ok=True)

def compress_audio_to_ogg(source_path, target_folder, relative_path):
    """
    压缩音频文件为OGG格式并保存到目标文件夹，保持目录结构。
    成功后删除源文件。
    :param source_path: 源音频文件路径
    :param target_folder: 目标文件夹路径
    :param relative_path: 相对于源文件夹的路径
    """
    # 获取文件名（不含扩展名）
    file_name, _ = os.path.splitext(os.path.basename(source_path))

    # 在目标文件夹中创建相同的目录结构
    target_subfolder = os.path.join(target_folder, relative_path)
    os.makedirs(target_subfolder, exist_ok=True)

    # 目标OGG文件路径
    target_ogg_path = os.path.join(target_subfolder, f"{file_name}.ogg")

    print(f"正在压缩音频: {source_path}")

    # 构建ffmpeg命令：压缩音频为ogg格式
    ffmpeg_command = [
        "ffmpeg",
        "-i", source_path,           # 输入文件
        "-c:a", "libvorbis",         # 音频编码器为libvorbis（ogg）
        "-q:a", "4",                 # 音频质量设置为4
        "-map", "0:a",               # 映射所有音频流
        "-y",                        # 覆盖输出文件
        target_ogg_path
    ]

    try:
        # 执行ffmpeg命令，避免编码问题
        result = subprocess.run(
            ffmpeg_command,
            check=True,
            capture_output=True,
            encoding='utf-8',
            errors='ignore'
        )

        # 检查目标文件是否存在且大小合理
        if os.path.exists(target_ogg_path) and os.path.getsize(target_ogg_path) > 0:
            print(f"音频压缩完成: {target_ogg_path}")

            # 删除源文件
            try:
                os.remove(source_path)
                print(f"已删除源文件: {source_path}")
            except Exception as e:
                print(f"删除源文件时出错 {source_path}: {e}")
        else:
            print(f"警告: 输出文件可能有问题，保留源文件: {source_path}")

    except subprocess.CalledProcessError as e:
        print(f"FFmpeg压缩失败: {source_path}")
        if e.stderr:
            print(f"错误信息: {e.stderr}")
        return
    except FileNotFoundError:
        print("错误: 未找到ffmpeg，请确保ffmpeg已安装并添加到系统PATH")
        return
    except Exception as e:
        print(f"处理文件时出错: {source_path}")
        print(f"错误信息: {e}")
        return

# 遍历源文件夹及其子文件夹中的所有音频文件
processed_count = 0
error_count = 0

for root, dirs, files in os.walk(source_folder):
    for file in files:
        if file.lower().endswith(audio_formats):  # 检查文件扩展名（忽略大小写）
            source_path = os.path.join(root, file)

            # 计算相对于源文件夹的路径
            relative_path = os.path.relpath(root, source_folder)

            # 压缩音频为OGG格式
            compress_audio_to_ogg(source_path, target_folder, relative_path)

            # 更新计数器
            processed_count += 1

print(f"\n音频处理完成！")
print(f"已处理文件数量: {processed_count}")

input("按回车键退出...")