import os
import shutil

def copy_video_files_rename_duplicates(source_root_dir, target_dir, video_extensions):
    """
    指定されたソースディレクトリの配下にある動画ファイルを
    ターゲットディレクトリにコピーする関数。
    サブフォルダの奥深くにあるファイルも全て探索します。
    コピー先に同名ファイルが存在する場合は、別名（例: filename (1).ext）でコピーします。

    Args:
        source_root_dir (str): コピー元のルートディレクトリ。
        target_dir (str): コピー先のディレクトリ。
        video_extensions (list): 対象とする動画ファイルの拡張子リスト（例: ['.mp4', '.mov']）。
                                 ドットから始まる小文字で指定。
    """
    # ターゲットディレクトリが存在しない場合は作成
    if not os.path.exists(target_dir):
        try:
            os.makedirs(target_dir)
            print(f"作成しました: {target_dir}")
        except OSError as e:
            print(f"エラー: ターゲットディレクトリを作成できませんでした - {target_dir} ({e})")
            return

    copied_count = 0
    error_count = 0

    print(f"'{source_root_dir}' から '{target_dir}' へ動画ファイルをコピーします...")
    print("同名ファイルが存在する場合は、別名でコピーします。")

    # source_root_dir 自身とその配下の全てのディレクトリとファイルを再帰的に探索
    # これにより「配下の配下の配下ファイル」も対象となります
    for dirpath, dirnames, filenames in os.walk(source_root_dir):
        # 現在処理中のディレクトリがターゲットディレクトリ自身の場合はスキップ
        # (ターゲットがソースのサブフォルダにある場合に無限ループや意図しないコピーを防ぐため)
        # 正規化された絶対パスで比較して確実に判定します
        if os.path.abspath(dirpath) == os.path.abspath(target_dir):
            print(f"スキップ (処理対象外のディレクトリ): {dirpath} (ここはコピー先フォルダです)")
            # このディレクトリ配下の探索を止めるためにdirnamesをクリア
            dirnames[:] = []
            continue

        for filename in filenames:
            # ファイルの拡張子を取得し、小文字に変換して比較
            file_ext_lower = os.path.splitext(filename)[1].lower()

            if file_ext_lower in video_extensions:
                source_file_path = os.path.join(dirpath, filename)

                # コピー先のファイルパスを決定
                target_file_path_original = os.path.join(target_dir, filename)
                current_target_file_path = target_file_path_original

                # 同名ファイルが存在するか確認し、存在すれば別名を生成
                if os.path.exists(current_target_file_path):
                    base, ext = os.path.splitext(filename)
                    counter = 1
                    # 新しいファイル名を生成: base (counter).ext
                    new_filename = f"{base} ({counter}){ext}"
                    current_target_file_path = os.path.join(target_dir, new_filename)

                    # 生成した別名がさらに存在するかチェックし、存在しなくなるまでカウンターを増やす
                    while os.path.exists(current_target_file_path):
                        counter += 1
                        new_filename = f"{base} ({counter}){ext}"
                        current_target_file_path = os.path.join(target_dir, new_filename)

                    # 別名でコピーを実行
                    try:
                        shutil.copy2(source_file_path, current_target_file_path)
                        print(f"コピー完了 (別名): {source_file_path} -> {current_target_file_path}")
                        copied_count += 1
                    except Exception as e:
                        print(f"エラー (別名コピー失敗): {source_file_path} -> {current_target_file_path} - {e}")
                        error_count += 1
                else:
                    # 同名ファイルが存在しない場合、そのままコピー
                    try:
                        shutil.copy2(source_file_path, current_target_file_path)
                        print(f"コピー完了: {source_file_path} -> {current_target_file_path}")
                        copied_count += 1
                    except Exception as e:
                        print(f"エラー (コピー失敗): {source_file_path} - {e}")
                        error_count += 1
            # else:
            #     print(f"スキップ (対象外の拡張子): {os.path.join(dirpath, filename)}")


    print("\n--- 処理結果 ---")
    print(f"コピーされたファイル数: {copied_count}")
    print(f"エラーが発生したファイル数: {error_count}")
    print("処理が完了しました。")

if __name__ == "__main__":
    # --- 設定ここから ---
    # コピー元のルートディレクトリを指定してください
    source_directory = r"D:\動画制作\素材"
    # コピー先のディレクトリを指定してください
    destination_directory = r"D:\動画制作\素材\カラーグレーディング前素材"

    # 対象とする動画ファイルの拡張子リスト (ドットから始まる小文字で)
    # 必要に応じて追加・削除してください
    target_extensions = [
        '.mp4', '.mov', '.avi', '.mkv', '.wmv',
        '.flv', '.webm', '.mpg', '.mpeg', '.m2ts', '.mts'
        # 例: '.mts', '.m2ts' などを追加できます
    ]
    # --- 設定ここまで ---

    # コピー元ディレクトリの存在チェック
    if not os.path.isdir(source_directory):
        print(f"エラー: コピー元ディレクトリが見つかりません - {source_directory}")
    else:
        copy_video_files_rename_duplicates(source_directory, destination_directory, target_extensions)

import os
import shutil

def copy_video_files_rename_duplicates(source_root_dir, target_dir, video_extensions):
    # ... 関数本体 ...

    # ターゲットディレクトリが存在しない場合は作成
    if not os.path.exists(target_dir):
        try:
            os.makedirs(target_dir)
            print(f"作成しました: {target_dir}")
        except OSError as e:
            print(f"エラー: ターゲットディレクトリを作成できませんでした - {target_dir} ({e})")
            return

    for dirpath, dirnames, filenames in os.walk(source_root_dir):
        # ... ファイル処理 ...

        # 現在処理中のディレクトリがターゲットディレクトリ自身の場合はスキップ
        if os.path.abspath(dirpath) == os.path.abspath(target_dir):
            print(f"スキップ (処理対象外のディレクトリ): {dirpath} (ここはコピー先フォルダです)")
            dirnames[:] = [] # このディレクトリ配下の探索を止める
            continue

        for filename in filenames:
            file_ext_lower = os.path.splitext(filename)[1].lower()

            if file_ext_lower in video_extensions:
                # ... 動画ファイルに対する処理 ...

                source_file_path = os.path.join(dirpath, filename)
                target_file_path_original = os.path.join(target_dir, filename)
                current_target_file_path = target_file_path_original

                # 同名ファイルが存在するか確認し、存在すれば別名を生成
                if os.path.exists(current_target_file_path):
                    # ... 別名生成ロジック ...
                else:
                    # ... そのままコピー ...

                if os.path.exists(current_target_file_path):
                    base, ext = os.path.splitext(filename)
                    counter = 1
                    new_filename = f"{base} ({counter}){ext}"
                    current_target_file_path = os.path.join(target_dir, new_filename)

                    while os.path.exists(current_target_file_path):
                        counter += 1
                        new_filename = f"{base} ({counter}){ext}"
                        current_target_file_path = os.path.join(target_dir, new_filename)

                    # 別名でコピーを実行
                    try:
                        shutil.copy2(source_file_path, current_target_file_path)
                        # ... 成功/失敗の表示 ...
                    except Exception as e:
                        # ... エラー処理 ...

                    # ... 別名でコピーを実行 (重複時) ...
                    try:
                        shutil.copy2(source_file_path, current_target_file_path)
                        print(f"コピー完了 (別名): {source_file_path} -> {current_target_file_path}")
                        copied_count += 1
                    except Exception as e:
                        print(f"エラー (別名コピー失敗): {source_file_path} -> {current_target_file_path} - {e}")
                        error_count += 1
                else:
                    # 同名ファイルが存在しない場合、そのままコピー
                    try:
                        shutil.copy2(source_file_path, current_target_file_path)
                        print(f"コピー完了: {source_file_path} -> {current_target_file_path}")
                        copied_count += 1
                    except Exception as e:
                        print(f"エラー (コピー失敗): {source_file_path} - {e}")
                        error_count += 1

    print("\n--- 処理結果 ---")
    print(f"コピーされたファイル数: {copied_count}")
    print(f"エラーが発生したファイル数: {error_count}")
    print("処理が完了しました。")

if __name__ == "__main__":
    # --- 設定ここから ---
    source_directory = r"D:\動画制作\素材"
    destination_directory = r"D:\動画制作\素材\カラーグレーディング前素材"
    target_extensions = [ ... ]
    # --- 設定ここまで ---

    if not os.path.isdir(source_directory):
        print(f"エラー: コピー元ディレクトリが見つかりません - {source_directory}")
    else:
        copy_video_files_rename_duplicates(source_directory, destination_directory, target_extensions)