import os
import shutil
import sys

def copy_mp4_files(source_dir, target_dir):
    # 确保目标目录存在
    if not os.path.exists(target_dir):
        os.makedirs(target_dir)

    # 遍历源目录
    for file_name in os.listdir(source_dir):
        # 检查文件扩展名是否为.mp4
        if file_name.lower().endswith('.mp4'):
            # 构造源文件的完整路径
            file_path = os.path.join(source_dir, file_name)
            # 构造目标文件的路径
            target_file_path = os.path.join(target_dir, file_name)
            # 复制文件
            shutil.copy2(file_path, target_file_path)
            print(f"Copied: {file_path} to {target_file_path}")

def main():
    # 检查命令行参数数量
    if len(sys.argv) != 3:
        print("Usage: python script.py <source_directory> <target_directory>")
        sys.exit(1)

    source_directory = sys.argv[1]  # 从命令行获取源文件夹路径
    target_directory = sys.argv[2]  # 从命令行获取目标文件夹路径

    copy_mp4_files(source_directory, target_directory)

if __name__ == "__main__":
    main()