datawrapper_mcp/handlers/publish.py

"""Handler for publishing Datawrapper charts."""

import json

from datawrapper import get_chart
from mcp.types import TextContent

from ..utils import get_api_token


async def publish_chart(arguments: dict) -> list[TextContent]:
    """Publish a chart to make it publicly accessible."""
    api_token = get_api_token()
    chart_id = arguments["chart_id"]

    # Get chart and publish using Pydantic instance method
    chart = get_chart(chart_id, access_token=api_token)
    chart.publish(access_token=api_token)

    result = {
        "chart_id": chart_id,
        "public_url": chart.get_public_url(),
        "message": "Chart published successfully!",
    }

    return [TextContent(type="text", text=json.dumps(result, indent=2))]