33 lines
1.0 KiB
Python
33 lines
1.0 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
from openpyxl import Workbook
|
|
|
|
from civilplan_mcp.config import get_settings
|
|
from civilplan_mcp.models import ProjectDomain
|
|
from civilplan_mcp.tools._base import wrap_response
|
|
|
|
|
|
def generate_schedule(
|
|
*,
|
|
project_name: str,
|
|
project_spec: dict[str, Any],
|
|
legal_procedures: dict[str, Any],
|
|
output_filename: str | None = None,
|
|
) -> dict[str, Any]:
|
|
output_dir = Path(project_spec.get("output_dir", get_settings().output_dir))
|
|
output_dir.mkdir(parents=True, exist_ok=True)
|
|
path = output_dir / (output_filename or f"{project_name}_schedule.xlsx")
|
|
|
|
workbook = Workbook()
|
|
sheet = workbook.active
|
|
sheet.title = "일정표"
|
|
sheet.append(["단계", "소요일", "비고"])
|
|
for phase, procedures in legal_procedures.get("phases", {}).items():
|
|
sheet.append([phase, len(procedures), "자동 생성"])
|
|
workbook.save(path)
|
|
|
|
return wrap_response({"status": "success", "file_path": str(path)}, ProjectDomain.복합)
|