"""
Outils d'export et téléchargement pour l'API Grist.
Ce module contient des outils MCP pour exporter et télécharger
des documents et tables Grist dans différents formats.
Corrections v0.2.0:
- download_table_csv: utilise download_doc_csv avec tableId parameter
- download_document_excel: gestion du paramètre tableId obligatoire
"""
import base64
import logging
from typing import Any, Dict, List, Optional, Union
from fastmcp import Context
from ..client import get_client
# Configurer le logger
logger = logging.getLogger("grist_mcp_server")
def register_export_tools(mcp_server):
"""
Enregistre tous les outils d'export sur le serveur MCP.
Args:
mcp_server: L'instance du serveur MCP sur laquelle enregistrer les outils.
"""
mcp_server.tool()(download_document_sqlite)
mcp_server.tool()(download_document_excel)
mcp_server.tool()(download_table_csv)
async def download_document_sqlite(
doc_id: str,
nohistory: bool = False,
template: bool = False,
ctx: Context = None
) -> Dict[str, Any]:
"""
Télécharge un document Grist au format SQLite.
Prérequis:
- list_documents: Pour obtenir un doc_id valide
Args:
doc_id: L'ID du document
nohistory: Si True, exclut l'historique des modifications
template: Si True, télécharge comme modèle (sans données utilisateur)
Returns:
Dict avec statut, message et contenu encodé en base64
"""
logger.info(f"Tool called: download_document_sqlite with doc_id: {doc_id}")
try:
client = get_client(ctx)
if not client:
return {
"success": False,
"message": "Client Grist non configuré"
}
content = await client.download_doc(doc_id, nohistory=nohistory, template=template)
# Encoder le contenu binaire en base64
encoded_content = base64.b64encode(content).decode('utf-8')
return {
"success": True,
"message": f"Document {doc_id} téléchargé avec succès au format SQLite",
"content_type": "application/x-sqlite3",
"filename": f"{doc_id}.sqlite",
"content_base64": encoded_content,
"size_bytes": len(content)
}
except Exception as e:
logger.error(f"Error downloading document as SQLite: {e}")
return {
"success": False,
"message": f"Erreur lors du téléchargement du document au format SQLite: {str(e)}"
}
async def download_document_excel(
doc_id: str,
table_id: Optional[str] = None,
header: str = "label",
ctx: Context = None
) -> Dict[str, Any]:
"""
Télécharge un document Grist au format Excel.
Prérequis:
- list_documents: Pour obtenir un doc_id valide
Args:
doc_id: L'ID du document
table_id: L'ID de la table à exporter (optionnel, première table par défaut)
header: Format des en-têtes (label, id, ou none)
Returns:
Dict avec statut, message et contenu encodé en base64
"""
logger.info(f"Tool called: download_document_excel with doc_id: {doc_id}, table_id: {table_id}")
try:
client = get_client(ctx)
if not client:
return {
"success": False,
"message": "Client Grist non configuré"
}
if header not in ["label", "id", "none"]:
return {
"success": False,
"message": "Format d'en-tête invalide. Doit être: label, id, ou none"
}
# Si pas de table_id, récupérer la première table du document
if not table_id:
tables = await client.list_tables(doc_id)
if not tables:
return {
"success": False,
"message": "Aucune table trouvée dans le document"
}
table_id = tables[0].id
logger.info(f"Auto-selected table: {table_id}")
content = await client.download_doc_xlsx(doc_id, table_id=table_id, header=header)
# Encoder le contenu binaire en base64
encoded_content = base64.b64encode(content).decode('utf-8')
return {
"success": True,
"message": f"Table {table_id} du document {doc_id} téléchargée avec succès au format Excel",
"content_type": "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
"filename": f"{table_id}.xlsx",
"content_base64": encoded_content,
"size_bytes": len(content)
}
except Exception as e:
logger.error(f"Error downloading document as Excel: {e}")
return {
"success": False,
"message": f"Erreur lors du téléchargement du document au format Excel: {str(e)}"
}
async def download_table_csv(
doc_id: str,
table_id: str,
header: str = "label",
ctx: Context = None
) -> Dict[str, Any]:
"""
Télécharge une table Grist au format CSV.
Prérequis:
- list_documents: Pour obtenir un doc_id valide
- list_tables: Pour obtenir un table_id valide
Args:
doc_id: L'ID du document
table_id: L'ID de la table
header: Format des en-têtes (label, id, ou none)
Returns:
Dict avec statut, message et contenu CSV
"""
logger.info(f"Tool called: download_table_csv with doc_id: {doc_id}, table_id: {table_id}")
try:
client = get_client(ctx)
if not client:
return {
"success": False,
"message": "Client Grist non configuré"
}
if header not in ["label", "id", "none"]:
return {
"success": False,
"message": "Format d'en-tête invalide. Doit être: label, id, ou none"
}
# CORRECTION v0.2.0: utilise download_doc_csv avec le paramètre table_id
content = await client.download_doc_csv(doc_id, table_id=table_id, header=header)
return {
"success": True,
"message": f"Table {table_id} du document {doc_id} téléchargée avec succès au format CSV",
"content_type": "text/csv",
"filename": f"{table_id}.csv",
"content": content,
"size_bytes": len(content.encode('utf-8'))
}
except Exception as e:
logger.error(f"Error downloading table as CSV: {e}")
return {
"success": False,
"message": f"Erreur lors du téléchargement de la table au format CSV: {str(e)}"
}
