How to Generate a Mock devcontainer.json Config in Python

import json
from pathlib import Path


def create_devcontainer_config(
    image: str = "mcr.microsoft.com/devcontainers/python:3.11",
    name: str = "python-dev-container",
    ports: list[int] | None = None,
    post_create: str | None = None,
) -> dict:
    config = {
        "name": name,
        "image": image,
        "customizations": {
            "vscode": {
                "extensions": [
                    "ms-python.python",
                    "ms-python.vscode-pylance",
                    "ms-python.black-formatter",
                ]
            }
        },
        "settings": {
            "python.defaultInterpreterPath": "/usr/local/bin/python",
            "python.testing.pytestEnabled": True,
        },
    }

    if ports:
        config["forwardPorts"] = ports

    if post_create:
        config["postCreateCommand"] = post_create

    return config


if __name__ == "__main__":
    config = create_devcontainer_config(
        ports=[8000, 8501],
        post_create="pip install -r requirements.txt",
    )
    output_path = Path(".devcontainer/devcontainer.json")
    output_path.parent.mkdir(exist_ok=True)
    output_path.write_text(json.dumps(config, indent=2))
    print(f"Mock devcontainer config written to: {output_path}")
    print(json.dumps(config, indent=2))