Add weasyprint.browser.

Start it with python -m weasyprint.browser Requires Flask (and thus Python 2)
2024-10-05 00:21:15 +03:00 · 2012-06-14 15:43:35 +02:00 · 2012-06-14 15:43:35 +02:00 · 6e6424e3a1
commit 6e6424e3a1
parent d2aeda4372
1 changed files with 135 additions and 0 deletions
--- a/weasyprint/browser.py
+++ b/weasyprint/browser.py
@ -0,0 +1,135 @@
+# coding: utf8
+"""
+    weasyprint.browser
+    ------------------
+
+    A WeasyPrint-based web browser. In your web browser.
+
+    :copyright: Copyright 2011-2012 Simon Sapin and contributors, see AUTHORS.
+    :license: BSD, see LICENSE for details.
+
+"""
+
+from __future__ import division, unicode_literals
+
+import io
+
+import cairo
+
+from weasyprint import HTML, CSS, draw
+from weasyprint.backends import PNGBackend
+from weasyprint.formatting_structure import boxes
+
+
+def find_links(box, links):
+    link = box.style.link
+    anchor = box.style.anchor
+    if link or anchor:
+        if link:
+            type_, href = box.style.link
+            if type_ == 'internal':
+                href = '#' + href
+            else:
+                href = '/' + href
+        else:
+            href = None
+        links.append((href, anchor, box.position_x, box.position_y,
+                      box.margin_width(), box.margin_height()))
+
+    if isinstance(box, boxes.ParentBox):
+        for child in box.children:
+            find_links(child, links)
+
+
+def surface_to_base64(surface):
+    file_obj = io.BytesIO()
+    surface.write_to_png(file_obj)
+    return 'data:image/png;base64,' + (
+        file_obj.getvalue().encode('base64').replace('\n', ''))
+
+
+def get_svg_pages(html, *stylesheets):
+    document = html._get_document(PNGBackend, stylesheets)
+    backend = PNGBackend(None)
+    result = []
+    for page in document.pages:
+        context = backend.start_page(page.outer_width, page.outer_height)
+        draw.draw_page(document, page, context)
+        width, height, surface = backend.pages.pop()
+        links = []
+        find_links(page, links)
+        result.append(dict(
+            width=width,
+            height=height,
+            links=links,
+            data_url=surface_to_base64(surface),
+        ))
+    return result
+
+
+def make_app():
+    # Keep here imports that are not required for the rest of WeasyPrint
+    from flask import Flask, request
+    from jinja2 import Template
+
+    app = Flask(__name__)
+
+    template = Template('''
+        <!doctype html>
+        <meta charset=utf-8>
+        <title>WeasyPrint Browser</title>
+        <style>
+            input { font: 1.5em sans-serif }
+            form { border-bottom: 1px #888 solid; padding: .5em }
+            section { -webkit-box-shadow: 0 0 10px 2px #aaa;
+                         -moz-box-shadow: 0 0 10px 2px #aaa;
+                          -ms-box-shadow: 0 0 10px 2px #aaa;
+                           -o-box-shadow: 0 0 10px 2px #aaa;
+                              box-shadow: 0 0 10px 2px #aaa;
+                      margin: 25px; position: relative }
+            a { position: absolute; display: block }
+            a[href]:hover, a[href]:focus { outline: 1px dotted }
+        </style>
+        <form onsubmit="window.location.href = '/' + this.url.value;
+                        return false;">
+            <input name=url style="width: 90%" value="{{ url }}" />
+            <input type=submit value=Go />
+        </form>
+        {% for page in pages %}
+            <section style="width: {{ page.width }}px;
+                            height: {{ page.height }}px">
+                <img src="{{ page.data_url }}">
+                {% for href, anchor, pos_x, pos_y, width, height
+                   in page.links %}
+                    <a{% if href %} href="{{ href }}"{% endif %}
+                      {% if anchor %} name="{{ anchor }}"{% endif %}
+                       style="width: {{ width }}px; height: {{ height }}px;
+                              left: {{ pos_x }}px; top: {{ pos_y }}px;"></a>
+                {% endfor %}
+            </section>
+        {% endfor %}
+    ''')
+
+    stylesheet = CSS(string='''
+        @page { -weasy-size: 640px; margin: 0 }
+    ''')
+
+    @app.route('/favicon.ico')
+    def favicon():
+        return 'No favicon yet.', 404
+
+    @app.route('/')
+    @app.route('/<path:url>')
+    def index(url=''):
+        if url:
+            if request.query_string:
+                url += '?' + request.query_string
+            pages = get_svg_pages(HTML(url), stylesheet)
+        else:
+            pages = []
+        return template.render(**locals())
+
+    return app
+
+if __name__ == '__main__':
+    make_app().run(debug=True)