【Python】フォルダ配下のコードを一括でまとめる方法 – LLMでのコード生成への活用

import os

def concatenate_files(
    root_dir='.', 
    excluded_dirs=None, 
    excluded_files=None, 
    output_file='concatenated_output.txt'
):
    """
    指定したディレクトリ配下のファイルを再帰的に読み込み、
    ファイルの相対パスと内容を連結したテキストを出力用テキストファイルに書き込みます。

    Parameters:
    -----------
    root_dir : str
        走査を開始するディレクトリ（デフォルトは現在ディレクトリ'.'）
    excluded_dirs : list or set
        走査から除外したいディレクトリ名のリスト/セット
    excluded_files : list or set
        走査から除外したいファイル名のリスト/セット
    output_file : str
        出力テキストファイルのパス
    """

    if excluded_dirs is None:
        excluded_dirs = []
    if excluded_files is None:
        excluded_files = []

    with open(output_file, 'w', encoding='utf-8') as out_f:
        for current_root, dirs, files in os.walk(root_dir):
            # 除外ディレクトリをスキップ
            dirs[:] = [d for d in dirs if d not in excluded_dirs]

            for filename in files:
                # 除外ファイルをスキップ
                if filename in excluded_files:
                    continue

                file_path = os.path.join(current_root, filename)

                # テキストファイルで読み込む際のエラーをハンドリング
                try:
                    with open(file_path, 'r', encoding='utf-8') as f:
                        content = f.read()

                    # ファイルパスと内容を区切って書き込み
                    out_f.write(f"=== File: {file_path} ===\n")
                    out_f.write(content)
                    out_f.write("\n\n---\n\n")
                except Exception as e:
                    print(f"Skipping file: {file_path} - {e}")

    print(f"ファイルの内容を '{output_file}' に書き出しました。")

if __name__ == '__main__':
    excluded_dirs = {'.git', 'venv', '__pycache__'}  # 除外したいディレクトリ
    excluded_files = {'.DS_Store', 'README.md'}      # 除外したいファイル
    
    concatenate_files(
        root_dir='.',
        excluded_dirs=excluded_dirs, 
        excluded_files=excluded_files, 
        output_file='concatenated_output.txt'
    )

=== File: ./example.py ===
( example.py の中身 )

---

=== File: ./subfolder/sample.txt ===
( sample.txt の中身 )

---

... (続く)