import json
from fastmcp import FastMCP
from wan_neng_copy_word import clone_document as word_to_dict
from wan_neng_copy_word_pro import clone_document
from wan_neng_copy_word import clone_document as get_para_style
from gen_all_styles import gen_all_styles

mcp = FastMCP(name="MyServer")

# 基礎(chǔ)問候工具
@mcp.tool
def greet(name: str) -> str:
    """Greet a user by name."""
    return f"Hello, {name}!"

# Word 轉(zhuǎn) JSON 工具
@mcp.tool
def word_to_json(word_path: str) -> str:
    """Convert a word document to json."""
    body_s, body_p = word_to_dict(word_path)
    return json.dumps(body_p)

# JSON 轉(zhuǎn) Word 工具
@mcp.tool
def json_to_word(word_path: str, json_data: str) -> str:
    """Convert a json to word document."""
    try:
        body_ws, _ = get_para_style('demo_template.docx')
    except:
        gen_all_styles()
        body_ws, _ = get_para_style('demo_template.docx')

    body_s, _ = get_para_style(word_path)
    clone_document(body_s, json.loads(json_data), body_ws, 'cloned_example.docx')
    return 'cloned_example.docx'

# 啟動(dòng) MCP 服務(wù)
if __name__ == "__main__":
    mcp.run(transport="streamable-http", host="127.0.0.1", port=9000)

import asyncio
from fastmcp import Client

# MCP 服務(wù)配置
config = {
    "mcpServers": {
        "document-service": {
            "url": "http://127.0.0.1:9000/mcp",
            "transport": "streamable-http"
        }
    }
}

# 創(chuàng)建客戶端實(shí)例
client = Client(config)

async def main():
    async with client:
        # 讀取 JSON 數(shù)據(jù)
        with open("1.json", "r", encoding="utf-8") as f:
            body_p = f.read()
        
        # 調(diào)用 JSON 轉(zhuǎn) Word 工具
        result = await client.call_tool(
            "json_to_word", 
            {"word_path": "1.docx", "json_data": body_p}
        )
        print(f"生成文檔路徑: {result}")

if __name__ == "__main__":
    asyncio.run(main())

pip install fastmcp python-docx

python my_server.py

python client_test.py

# 可擴(kuò)展的異常處理示例
try:
    # 文件操作代碼
except FileNotFoundError as e:
    return {"error": f"Missing file: {str(e)}"}
except json.JSONDecodeError:
    return {"error": "Invalid JSON input"}

def validate_word_file(path):
    if not os.path.exists(path):
        raise ValueError("Template file not found")
    if not path.endswith('.docx'):
        raise ValueError("Invalid file format")