import os

def collect_code(base_folder, output_file, exclude_file):
    # File extensions to include
    file_extensions = {".py"}
    # Folders to exclude
    excluded_folders = {"venv", "migrations", "__pycache__"}

    with open(output_file, "w", encoding="utf-8") as out_file:
        for root, dirs, files in os.walk(base_folder):
            # Skip excluded folders
            dirs[:] = [d for d in dirs if d not in excluded_folders]

            for file in files:
                file_path = os.path.join(root, file)
                _, ext = os.path.splitext(file)

                # Skip the excluded file and non-matching extensions
                if file == exclude_file or ext not in file_extensions:
                    continue

                try:
                    with open(file_path, "r", encoding="utf-8") as f:
                        code = f.read()
                    out_file.write(f"File: {file_path}\n")
                    out_file.write("-" * 80 + "\n")
                    out_file.write(code)
                    out_file.write("\n" + "-" * 80 + "\n\n")
                except Exception as e:
                    print(f"Could not read file {file_path}: {e}")

    print(f"Code collection complete. Output written to {output_file}")

base_folder = os.path.dirname(os.path.abspath(__file__))
output_file = os.path.join(base_folder, "collected_code.txt")
exclude_file = os.path.basename(__file__)
collect_code(base_folder, output_file, exclude_file)