Compaction Small Files Mock in Python

from pathlib import Path
import tempfile
import os


def create_small_files(directory: Path, file_count: int = 5, lines_per_file: int = 3):
    """Create several small mock files with sample content."""
    directory.mkdir(exist_ok=True)
    for i in range(file_count):
        file_path = directory / f"part-{i:04d}.txt"
        content = "\n".join(f"row-{i}-{j}" for j in range(lines_per_file))
        file_path.write_text(content + "\n")


def compact_files(source_dir: Path, output_file: Path):
    """Merge all small files into a single compact file."""
    with output_file.open("w") as out_f:
        for file_path in sorted(source_dir.glob("*.txt")):
            out_f.write(file_path.read_text())


if __name__ == "__main__":
    with tempfile.TemporaryDirectory() as tmp_dir:
        src_dir = Path(tmp_dir) / "source"
        create_small_files(src_dir, file_count=3, lines_per_file=2)

        output_path = Path(tmp_dir) / "compacted.txt"
        compact_files(src_dir, output_path)

        print(f"Small files: {len(list(src_dir.glob('*.txt')))}")
        print(f"Compacted content:")
        print(output_path.read_text())