Draft for contract printing

back/app/contract/print/__init__.py

@@ -0,0 +1,96 @@
+from fastapi import APIRouter
+from fastapi.responses import HTMLResponse, FileResponse
+from fastapi.templating import Jinja2Templates
+
+from weasyprint import HTML, CSS
+from weasyprint.text.fonts import FontConfiguration
+
+from pathlib import Path
+
+from app.entity.models import Entity
+from app.template.models import ProvisionTemplate
+from ..schemas import ContractDraft
+
+
+async def build_model(model):
+    parties = []
+    for p in model.parties:
+        party = {
+            "entity": await Entity.get(p.entity_id),
+            "part": p.part
+        }
+        if p.representative_id:
+            party['representative'] = await Entity.get(p.representative_id)
+
+        parties.append(party)
+
+
+    model.parties = parties
+
+    provisions = []
+    for p in model.provisions:
+        if p.provision.type == "template":
+            provisions.append(await ProvisionTemplate.get(p.provision.provision_template_id))
+        else:
+            provisions.append(p.provision)
+    model.provisions = provisions
+
+    model.location = "Toulouse"
+    model.date = "01/01/1970"
+    return model
+
+
+BASE_PATH = Path(__file__).resolve().parent
+
+print_router = APIRouter()
+
+
+templates = Jinja2Templates(directory=str(BASE_PATH / "templates"))
+
+
+async def render_print(host, draft, lawyer):
+    template = templates.get_template("print.html")
+    return template.render({
+        "draft": draft,
+        "lawyer": lawyer,
+        "static_host": host
+    })
+
+
+async def render_css(host, draft):
+    template = templates.get_template("styles.css")
+    return template.render({
+        "draft": draft,
+        "static_host": host
+    })
+
+
+@print_router.get("/", response_class=HTMLResponse)
+async def create() -> str:
+    draft = await build_model(await ContractDraft.get("63e92534aafed8b509f229c4"))
+    lawyer = {
+        "firstname": "Nathaniel",
+        "lastname": "Toshi",
+    }
+
+    return await render_print('localhost', draft, lawyer)
+
+
+@print_router.get("/pdf", response_class=FileResponse)
+async def create_pdf() -> str:
+    draft = await build_model(await ContractDraft.get("63e92534aafed8b509f229c4"))
+    lawyer = {
+        "firstname": "Nathaniel",
+        "lastname": "Toshi",
+    }
+
+    font_config = FontConfiguration()
+    html = HTML(string=await render_print('nginx', draft, lawyer))
+    css = CSS(string=await render_css('nginx', draft), font_config=font_config)
+
+    html.write_pdf('out.pdf', stylesheets=[css], font_config=font_config)
+
+    return FileResponse(
+        "out.pdf",
+        media_type="application/pdf",
+        filename=draft.name)