import os
import subprocess


def get_exists_files(output_dir):
    return [i[:-2] + "kv" for i in os.listdir(output_dir)]


def convert_mkv_to_mp4(input_dir, output_dir):
    # 确保输出目录存在
    os.makedirs(output_dir, exist_ok=True)

    exists_files = get_exists_files(output_dir)

    # 遍历输入目录中的所有文件
    for filename in os.listdir(input_dir):
        if filename in exists_files:
            continue

        if filename.endswith('.mkv'):
            input_file = os.path.join(input_dir, filename)
            output_file = os.path.join(output_dir, filename[:-4] + '.mp4')

            # 使用 ffmpeg 进行转换
            command = ['ffmpeg', '-i', input_file, output_file]
            subprocess.run(command)

            print(f'Converted: {input_file} to {output_file}')


# 使用示例
input_directory = r'E:\琅琊榜mkv'  # 替换为你的输入目录
output_directory = r'D:\琅琊榜'  # 替换为你的输出目录
convert_mkv_to_mp4(input_directory, output_directory)