go-api/api/playwright.py at ed61db0df9fc7b180c0fe8852438e4bca9557fa4 · IFRCGo/go-api · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
import json
import pathlib
import tempfile
import time

from django.conf import settings
from django.core.files.base import ContentFile
from playwright.sync_api import sync_playwright

from .utils import DebugPlaywright

footer_template = """
        <div class="footer" style="width: 100%;font-size: 8px;color: #FEFEFE; bottom: 10px; position: absolute;">
        <div style="float: left; margin-top: 10px; margin-left: 40px;">
            Page <span class="pageNumber"></span> / <span class="totalPages"></span>
        </div>
        <div style="float: right; margin-right: 40px;">
            <svg
                xmlns="http://www.w3.org/2000/svg"
                viewBox="0 0 89.652 89.654"
                height="48"
                width="48"
            >
            <path
                d="M50.284 18.637a5.14 5.14 0 00-5.136-5.135 5.139 5.139 0 00-5.135 5.135 5.141 5.141 0 005.135 5.138 5.146 5.146 0 005.136-5.138M28.416 63.032a5.143 5.143 0 00-5.138 5.138 5.14 5.14 0 005.138 5.133 5.14 5.14 0 005.136-5.133 5.143 5.143 0 00-5.136-5.138M45.151 34.057a7.021 7.021 0 00-7.02 7.025 7.02 7.02 0 0014.04 0 7.021 7.021 0 00-7.02-7.025M61.883 63.032a5.143 5.143 0 00-5.135 5.138 5.138 5.138 0 005.135 5.133 5.14 5.14 0 005.136-5.133 5.143 5.143 0 00-5.136-5.138"
                class="st1"
                fill="#F5333F"
            />
            <path
                d="M61.883 75.769c-4.19 0-7.601-3.41-7.601-7.602 0-2.32 1.05-4.4 2.696-5.794L49.726 50.26a10.205 10.205 0 01-4.575 1.085c-1.648 0-3.196-.397-4.577-1.085l-7.252 12.113a7.571 7.571 0 012.693 5.794c0 4.191-3.408 7.602-7.599 7.602-4.19 0-7.601-3.41-7.601-7.602 0-4.19 3.41-7.601 7.601-7.601.984 0 1.926.196 2.791.54l7.303-12.2a10.236 10.236 0 01-3.63-7.827c0-5.254 3.947-9.58 9.038-10.189v-4.762c-3.606-.59-6.368-3.72-6.368-7.49 0-4.192 3.41-7.602 7.601-7.602s7.599 3.41 7.599 7.601c0 3.77-2.762 6.9-6.366 7.49v4.763c5.093.611 9.038 4.935 9.038 10.19a10.23 10.23 0 01-3.633 7.826l7.306 12.2a7.544 7.544 0 012.791-.54c4.191 0 7.599 3.41 7.599 7.601s-3.41 7.602-7.602 7.602m-49.286-34.65c0-5.485 3.44-10.057 9.194-10.057 4.194 0 7.715 2.236 8.226 6.562h-3.281c-.32-2.524-2.524-3.818-4.945-3.818-4.117 0-5.834 3.627-5.834 7.313s1.717 7.313 5.834 7.313c3.44.056 5.32-2.016 5.376-5.268h-5.106v-2.556h8.173v10.11h-2.151l-.51-2.257c-1.803 2.043-3.44 2.715-5.78 2.715-5.754 0-9.196-4.57-9.196-10.057M44.826 0C20.07 0 0 20.069 0 44.828c0 24.755 20.071 44.826 44.826 44.826 24.757 0 44.826-20.071 44.826-44.826C89.652 20.068 69.582 0 44.826 0"
                class="st1"
                fill="#F5333F"
            />
             </svg>
        </div>
        </div>
    """  # noqa


def build_storage_state(tmp_dir, user, token, language="en"):
    temp_file = pathlib.Path(tmp_dir, "storage_state.json")
    temp_file.touch()

    state = {
        "origins": [
            {
                "origin": settings.GO_WEB_INTERNAL_URL + "/",
                "localStorage": [
                    {
                        "name": "user",
                        "value": json.dumps(
                            {
                                "id": user.id,
                                "username": user.username,
                                "firstName": user.first_name,
                                "lastName": user.last_name,
                                "token": token.key,
                            }
                        ),
                    },
                    {"name": "language", "value": json.dumps(language)},
                ],
            }
        ]
    }
    with open(temp_file, "w") as f:
        json.dump(state, f)
    return temp_file


def render_pdf_from_url(
    *,
    url: str,
    user,
    token,
    language: str = "en",
    timeout: int = 300_000,
):
    """
    Renders a URL to PDF using Playwright.
    Returns a Django ContentFile.
    """
    with tempfile.TemporaryDirectory() as tmp_dir:
        storage_state = build_storage_state(
            tmp_dir=tmp_dir,
            user=user,
            token=token,
            language=language,
        )

        with sync_playwright() as playwright:
            browser = playwright.chromium.connect(settings.PLAYWRIGHT_SERVER_URL)

            try:
                context = browser.new_context(storage_state=storage_state)
                page = context.new_page()

                if settings.DEBUG_PLAYWRIGHT:
                    DebugPlaywright.debug(page)

                page.goto(url, timeout=timeout)
                time.sleep(5)
                # NOTE: Use wait_for_load_state instead of sleep?
                # page.wait_for_load_state("networkidle", timeout=timeout)
                page.wait_for_selector(
                    "#pdf-preview-ready",
                    state="attached",
                    timeout=timeout,
                )

                pdf_bytes = page.pdf(
                    display_header_footer=True,
                    prefer_css_page_size=True,
                    print_background=True,
                    footer_template=footer_template,
                    header_template="<p></p>",
                )
            finally:
                browser.close()

    return ContentFile(pdf_bytes)