uv sync

from jxls_mcp.models import GenerateTemplateRequest, DataStruct, DataField
from jxls_mcp.services import TemplateGenerator

# 创建数据结构
data_struct = DataStruct(
    collectName="employees",
    itemVariable="employee",
    dataFields=[
        DataField(name="姓名", field="name"),
        DataField(name="年龄", field="age"),
        DataField(name="部门", field="department")
    ]
)

# 创建请求
request = GenerateTemplateRequest(
    templateName="employee_report",
    dataStruct=data_struct,
    dataFormat="json"
)

# 生成模板
generator = TemplateGenerator()
response = generator.generate_template(request)

print(f"生成成功: {response.success}")
print(f"文件路径: {response.templatePath}")

# 创建数组格式数据结构
data_struct = DataStruct(
    collectName="dataList",
    itemVariable="row",
    dataFields=[
        DataField(name="列1", index=0),
        DataField(name="列2", index=1),
        DataField(name="列3", index=2)
    ]
)

# 创建请求
request = GenerateTemplateRequest(
    templateName="data_export",
    dataStruct=data_struct,
    dataFormat="array"
)

# 生成模板
generator = TemplateGenerator()
response = generator.generate_template(request)

# 创建包含绑定和不绑定列的数据结构
data_struct = DataStruct(
    collectName="employees",
    itemVariable="employee",
    dataFields=[
        DataField(name="序号"),  # 不绑定数据，留空供手工填写
        DataField(name="姓名", field="name"),  # 绑定到name字段
        DataField(name="年龄", field="age"),  # 绑定到age字段
        DataField(name="部门", field="department"),  # 绑定到department字段
        DataField(name="备注")  # 不绑定数据，留空供手工填写
    ]
)

# 创建请求
request = GenerateTemplateRequest(
    templateName="employee_with_static_columns",
    dataStruct=data_struct,
    dataFormat="json"
)

# 生成模板
generator = TemplateGenerator()
response = generator.generate_template(request)

# 生成的Excel模板将包含：
# 第1行：序号 | 姓名 | 年龄 | 部门 | 备注
# 第2行：(空) | ${employee.name} | ${employee.age} | ${employee.department} | (空)

uv run jxls-mcp-server

{
  "name": "generateJxlsTemplate",
  "arguments": {
    "templateName": "sales_report",
    "dataStruct": {
      "collectName": "sales",
      "itemVariable": "sale",
      "dataFields": [
        {"name": "产品名称", "field": "product"},
        {"name": "销售额", "field": "amount"},
        {"name": "销售日期", "field": "date"}
      ]
    },
    "dataFormat": "json",
    "sampleData": [
      {"product": "产品A", "amount": 1000, "date": "2024-01-01"}
    ]
  }
}

{
  "collectName": "data",
  "itemVariable": "item", 
  "dataFields": [
    {
      "name": "列标题",
      "field": "fieldName",
      "index": 0
    }
  ]
}

A1: 姓名           B1: 年龄           C1: 城市
A2: ${item.name}   B2: ${item.age}    C2: ${item.city}

A1: 列1            B1: 列2            C1: 列3
A2: ${item[0]}     B2: ${item[1]}     C2: ${item[2]}

{
  "outputPath": "templates/my_template.xlsx"  // 使用指定文件名
}

{
  "outputPath": "C:/Users/username/Documents/my_template.xlsx"  // 使用指定文件名
}

{
  "outputPath": "C:/Users/username/Documents/templates"  // 自动生成: templateName_timestamp.xlsx
}

{
  "outputPath": "reports/sales_report"  // 自动添加.xlsx扩展名
}

{
  "success": true,
  "templatePath": "/path/to/generated/template.xlsx",
  "message": "JXLS模板生成成功",
  "jxlsAnnotations": {
    "area": "jx:area(lastCell=\"C2\")",
    "each": "jx:each(items=\"data\" var=\"item\" lastCell=\"C2\")"
  },
  "dataStruct": {
    "collectName": "data",
    "itemVariable": "item",
    "columnCount": 3,
    "lastCell": "C2"
  }
}

pytest

black src/ tests/
isort src/ tests/

mypy src/