feat: export service — DOCX, PDF, MD generation and download API
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,79 @@
|
||||
import os
|
||||
import uuid
|
||||
|
||||
from fastapi import APIRouter, Depends, HTTPException
|
||||
from fastapi.responses import FileResponse
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from app.api.deps import get_current_user
|
||||
from app.database import get_db
|
||||
from app.models.export_file import ExportFile, ExportFormat
|
||||
from app.models.protocol import Protocol
|
||||
from app.models.user import User
|
||||
from app.schemas.export_file import ExportFileResponse, ExportRequest
|
||||
from app.services.export import export_docx, export_markdown, export_pdf
|
||||
|
||||
router = APIRouter(prefix="/api/exports", tags=["exports"])
|
||||
|
||||
EXPORTERS = {
|
||||
ExportFormat.md: export_markdown,
|
||||
ExportFormat.docx: export_docx,
|
||||
ExportFormat.pdf: export_pdf,
|
||||
}
|
||||
|
||||
MEDIA_TYPES = {
|
||||
ExportFormat.md: "text/markdown",
|
||||
ExportFormat.docx: "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
|
||||
ExportFormat.pdf: "application/pdf",
|
||||
}
|
||||
|
||||
|
||||
@router.post("/protocol/{protocol_id}", response_model=ExportFileResponse, status_code=201)
|
||||
async def create_export(
|
||||
protocol_id: uuid.UUID,
|
||||
data: ExportRequest,
|
||||
db: AsyncSession = Depends(get_db),
|
||||
user: User = Depends(get_current_user),
|
||||
):
|
||||
result = await db.execute(
|
||||
select(Protocol).where(Protocol.id == protocol_id)
|
||||
)
|
||||
protocol = result.scalar_one_or_none()
|
||||
if not protocol:
|
||||
raise HTTPException(404, "Protocol not found")
|
||||
|
||||
exporter = EXPORTERS[data.format]
|
||||
file_path = exporter(protocol.content, protocol.id)
|
||||
file_size = os.path.getsize(file_path)
|
||||
|
||||
export_file = ExportFile(
|
||||
protocol_id=protocol_id,
|
||||
format=data.format,
|
||||
file_path=file_path,
|
||||
file_size=file_size,
|
||||
)
|
||||
db.add(export_file)
|
||||
await db.commit()
|
||||
await db.refresh(export_file)
|
||||
return export_file
|
||||
|
||||
|
||||
@router.get("/{export_id}/download")
|
||||
async def download_export(
|
||||
export_id: uuid.UUID,
|
||||
db: AsyncSession = Depends(get_db),
|
||||
user: User = Depends(get_current_user),
|
||||
):
|
||||
result = await db.execute(
|
||||
select(ExportFile).where(ExportFile.id == export_id)
|
||||
)
|
||||
export_file = result.scalar_one_or_none()
|
||||
if not export_file or not os.path.exists(export_file.file_path):
|
||||
raise HTTPException(404, "Export file not found")
|
||||
|
||||
return FileResponse(
|
||||
path=export_file.file_path,
|
||||
media_type=MEDIA_TYPES[export_file.format],
|
||||
filename=f"protocol.{export_file.format.value}",
|
||||
)
|
||||
@@ -4,8 +4,10 @@ from fastapi import FastAPI
|
||||
from sqlalchemy import select
|
||||
|
||||
from app.api.auth import router as auth_router
|
||||
from app.api.exports import router as exports_router
|
||||
from app.api.protocols import router as protocols_router
|
||||
from app.api.recordings import router as recordings_router
|
||||
from app.api.settings import router as settings_router
|
||||
from app.api.templates import router as templates_router
|
||||
from app.database import async_session
|
||||
from app.models.prompt_template import PromptTemplate
|
||||
@@ -33,6 +35,8 @@ app.include_router(auth_router)
|
||||
app.include_router(recordings_router)
|
||||
app.include_router(protocols_router)
|
||||
app.include_router(templates_router)
|
||||
app.include_router(exports_router)
|
||||
app.include_router(settings_router)
|
||||
|
||||
|
||||
@app.get("/api/health")
|
||||
|
||||
@@ -0,0 +1,20 @@
|
||||
import uuid
|
||||
from datetime import datetime
|
||||
|
||||
from pydantic import BaseModel
|
||||
|
||||
from app.models.export_file import ExportFormat
|
||||
|
||||
|
||||
class ExportRequest(BaseModel):
|
||||
format: ExportFormat
|
||||
|
||||
|
||||
class ExportFileResponse(BaseModel):
|
||||
id: uuid.UUID
|
||||
protocol_id: uuid.UUID
|
||||
format: ExportFormat
|
||||
file_size: int
|
||||
created_at: datetime
|
||||
|
||||
model_config = {"from_attributes": True}
|
||||
@@ -0,0 +1,105 @@
|
||||
import json
|
||||
import os
|
||||
import uuid
|
||||
|
||||
import markdown
|
||||
from docx import Document
|
||||
from docx.shared import Pt
|
||||
from jinja2 import Template
|
||||
from weasyprint import HTML
|
||||
|
||||
from app.config import settings
|
||||
|
||||
SECTION_LABELS = {
|
||||
"date_participants_topic": "Дата, участники, тема встречи",
|
||||
"process_description": "Описание обследуемого процесса",
|
||||
"current_state": "Текущее состояние (as-is)",
|
||||
"problems": "Выявленные проблемы / узкие места",
|
||||
"agreements": "Договорённости и решения",
|
||||
"open_questions": "Открытые вопросы",
|
||||
"tasks": "Задачи с ответственными и сроками",
|
||||
"next_steps": "Следующие шаги",
|
||||
"client_goals": "Верхнеуровневые цели клиента",
|
||||
"project_parameters": "Параметры будущего проекта",
|
||||
"budget_constraints": "Бюджетные ограничения",
|
||||
"time_constraints": "Ограничения по срокам",
|
||||
"decision_makers": "Лица, принимающие решения (ЛПР)",
|
||||
"beneficiaries": "Бенефициары",
|
||||
"influencers": "Лица, влияющие на принятие решения",
|
||||
"discussed_topics": "Обсуждённые вопросы",
|
||||
"decisions": "Принятые решения",
|
||||
}
|
||||
|
||||
|
||||
def _format_section_value(value) -> str:
|
||||
if isinstance(value, list):
|
||||
if value and isinstance(value[0], dict):
|
||||
lines = []
|
||||
for item in value:
|
||||
parts = [f"{k}: {v}" for k, v in item.items()]
|
||||
lines.append("- " + ", ".join(parts))
|
||||
return "\n".join(lines)
|
||||
return "\n".join(f"- {item}" for item in value)
|
||||
return str(value)
|
||||
|
||||
|
||||
def _protocol_to_markdown(content: dict) -> str:
|
||||
lines = ["# Протокол встречи\n"]
|
||||
for key, value in content.items():
|
||||
if key == "raw":
|
||||
lines.append(str(value))
|
||||
continue
|
||||
label = SECTION_LABELS.get(key, key)
|
||||
lines.append(f"## {label}\n")
|
||||
lines.append(_format_section_value(value))
|
||||
lines.append("")
|
||||
return "\n".join(lines)
|
||||
|
||||
|
||||
def export_markdown(content: dict, protocol_id: uuid.UUID) -> str:
|
||||
os.makedirs(settings.export_dir, exist_ok=True)
|
||||
file_path = os.path.join(settings.export_dir, f"{protocol_id}.md")
|
||||
md_text = _protocol_to_markdown(content)
|
||||
with open(file_path, "w", encoding="utf-8") as f:
|
||||
f.write(md_text)
|
||||
return file_path
|
||||
|
||||
|
||||
def export_docx(content: dict, protocol_id: uuid.UUID) -> str:
|
||||
os.makedirs(settings.export_dir, exist_ok=True)
|
||||
file_path = os.path.join(settings.export_dir, f"{protocol_id}.docx")
|
||||
|
||||
doc = Document()
|
||||
doc.add_heading("Протокол встречи", level=0)
|
||||
|
||||
for key, value in content.items():
|
||||
if key == "raw":
|
||||
doc.add_paragraph(str(value))
|
||||
continue
|
||||
label = SECTION_LABELS.get(key, key)
|
||||
doc.add_heading(label, level=1)
|
||||
text = _format_section_value(value)
|
||||
for line in text.split("\n"):
|
||||
doc.add_paragraph(line, style="List Bullet" if line.startswith("- ") else None)
|
||||
|
||||
doc.save(file_path)
|
||||
return file_path
|
||||
|
||||
|
||||
def export_pdf(content: dict, protocol_id: uuid.UUID) -> str:
|
||||
os.makedirs(settings.export_dir, exist_ok=True)
|
||||
file_path = os.path.join(settings.export_dir, f"{protocol_id}.pdf")
|
||||
|
||||
md_text = _protocol_to_markdown(content)
|
||||
html_body = markdown.markdown(md_text)
|
||||
html_full = f"""
|
||||
<html><head><meta charset="utf-8">
|
||||
<style>
|
||||
body {{ font-family: sans-serif; margin: 40px; font-size: 14px; line-height: 1.6; }}
|
||||
h1 {{ font-size: 22px; }} h2 {{ font-size: 16px; color: #333; border-bottom: 1px solid #ccc; padding-bottom: 4px; }}
|
||||
ul {{ padding-left: 20px; }}
|
||||
</style></head>
|
||||
<body>{html_body}</body></html>
|
||||
"""
|
||||
HTML(string=html_full).write_pdf(file_path)
|
||||
return file_path
|
||||
@@ -0,0 +1,33 @@
|
||||
import pytest
|
||||
from app.services.export import _protocol_to_markdown, _format_section_value
|
||||
|
||||
|
||||
def test_format_section_value_string():
|
||||
assert _format_section_value("Hello world") == "Hello world"
|
||||
|
||||
|
||||
def test_format_section_value_list():
|
||||
result = _format_section_value(["item1", "item2"])
|
||||
assert "- item1" in result
|
||||
assert "- item2" in result
|
||||
|
||||
|
||||
def test_format_section_value_list_of_dicts():
|
||||
tasks = [{"task": "Do X", "responsible": "Alice", "deadline": "2026-04-10"}]
|
||||
result = _format_section_value(tasks)
|
||||
assert "task: Do X" in result
|
||||
assert "responsible: Alice" in result
|
||||
|
||||
|
||||
def test_protocol_to_markdown():
|
||||
content = {
|
||||
"date_participants_topic": "01.04.2026, Иванов, Петров",
|
||||
"problems": ["Проблема 1", "Проблема 2"],
|
||||
"next_steps": "Назначить следующую встречу",
|
||||
}
|
||||
md = _protocol_to_markdown(content)
|
||||
assert "# Протокол встречи" in md
|
||||
assert "## Дата, участники, тема встречи" in md
|
||||
assert "## Выявленные проблемы" in md
|
||||
assert "- Проблема 1" in md
|
||||
assert "## Следующие шаги" in md
|
||||
Reference in New Issue
Block a user