mirror of
https://github.com/Kozea/WeasyPrint.git
synced 2024-10-05 00:21:15 +03:00
377 lines
12 KiB
Python
377 lines
12 KiB
Python
"""
|
||
weasyprint.html
|
||
---------------
|
||
|
||
Specific handling for some HTML elements, especially replaced elements.
|
||
|
||
Replaced elements (eg. <img> elements) are rendered externally and
|
||
behave as an atomic opaque box in CSS. In general, they may or may not
|
||
have intrinsic dimensions. But the only replaced elements currently
|
||
supported in WeasyPrint are images with intrinsic dimensions.
|
||
|
||
:copyright: Copyright 2011-2019 Simon Sapin and contributors, see AUTHORS.
|
||
:license: BSD, see LICENSE for details.
|
||
|
||
"""
|
||
|
||
import logging
|
||
import re
|
||
from urllib.parse import urljoin
|
||
|
||
from . import CSS, ROOT
|
||
from .css import get_child_text
|
||
from .formatting_structure import boxes
|
||
from .logger import LOGGER
|
||
from .urls import get_url_attribute
|
||
|
||
# XXX temporarily disable logging for user-agent stylesheet
|
||
level = LOGGER.level
|
||
LOGGER.setLevel(logging.ERROR)
|
||
|
||
HTML5_UA_COUNTER_STYLE = {}
|
||
HTML5_UA_STYLESHEET = CSS(
|
||
filename=(ROOT / 'css' / 'html5_ua.css'),
|
||
counter_style=HTML5_UA_COUNTER_STYLE)
|
||
HTML5_PH_STYLESHEET = CSS(filename=(ROOT / 'css' / 'html5_ph.css'))
|
||
|
||
LOGGER.setLevel(level)
|
||
|
||
|
||
# http://whatwg.org/C#space-character
|
||
HTML_WHITESPACE = ' \t\n\f\r'
|
||
HTML_SPACE_SEPARATED_TOKENS_RE = re.compile('[^%s]+' % HTML_WHITESPACE)
|
||
|
||
|
||
def ascii_lower(string):
|
||
r"""Transform (only) ASCII letters to lower case: A-Z is mapped to a-z.
|
||
|
||
:param string: An Unicode string.
|
||
:returns: A new Unicode string.
|
||
|
||
This is used for `ASCII case-insensitive
|
||
<http://whatwg.org/C#ascii-case-insensitive>`_ matching.
|
||
|
||
This is different from the :meth:`~py:str.lower` method of Unicode strings
|
||
which also affect non-ASCII characters,
|
||
sometimes mapping them into the ASCII range:
|
||
|
||
>>> keyword = u'Bac\N{KELVIN SIGN}ground'
|
||
>>> assert keyword.lower() == u'background'
|
||
>>> assert ascii_lower(keyword) != keyword.lower()
|
||
>>> assert ascii_lower(keyword) == u'bac\N{KELVIN SIGN}ground'
|
||
|
||
"""
|
||
# This turns out to be faster than unicode.translate()
|
||
return string.encode('utf8').lower().decode('utf8')
|
||
|
||
|
||
def element_has_link_type(element, link_type):
|
||
"""
|
||
Return whether the given element has a ``rel`` attribute with the
|
||
given link type.
|
||
|
||
:param link_type: Must be a lower-case string.
|
||
|
||
"""
|
||
return any(ascii_lower(token) == link_type for token in
|
||
HTML_SPACE_SEPARATED_TOKENS_RE.findall(element.get('rel', '')))
|
||
|
||
|
||
# Maps HTML tag names to function taking an HTML element and returning a Box.
|
||
HTML_HANDLERS = {}
|
||
|
||
|
||
def handle_element(element, box, get_image_from_uri, base_url):
|
||
"""Handle HTML elements that need special care.
|
||
|
||
:returns: a (possibly empty) list of boxes.
|
||
"""
|
||
if box.element_tag in HTML_HANDLERS:
|
||
return HTML_HANDLERS[element.tag](
|
||
element, box, get_image_from_uri, base_url)
|
||
else:
|
||
return [box]
|
||
|
||
|
||
def handler(tag):
|
||
"""Return a decorator registering a function handling ``tag`` elements."""
|
||
def decorator(function):
|
||
"""Decorator registering a function handling ``tag`` elements."""
|
||
HTML_HANDLERS[tag] = function
|
||
return function
|
||
return decorator
|
||
|
||
|
||
def make_replaced_box(element, box, image):
|
||
"""Wrap an image in a replaced box.
|
||
|
||
That box is either block-level or inline-level, depending on what the
|
||
element should be.
|
||
|
||
"""
|
||
if box.style['display'] in ('block', 'list-item', 'table'):
|
||
type_ = boxes.BlockReplacedBox
|
||
else:
|
||
# TODO: support images with 'display: table-cell'?
|
||
type_ = boxes.InlineReplacedBox
|
||
new_box = type_(element.tag, box.style, element, image)
|
||
# TODO: check other attributes that need to be copied
|
||
# TODO: find another solution
|
||
new_box.string_set = box.string_set
|
||
new_box.bookmark_label = box.bookmark_label
|
||
return new_box
|
||
|
||
|
||
@handler('img')
|
||
def handle_img(element, box, get_image_from_uri, base_url):
|
||
"""Handle ``<img>`` elements, return either an image or the alt-text.
|
||
|
||
See: http://www.w3.org/TR/html5/embedded-content-1.html#the-img-element
|
||
|
||
"""
|
||
src = get_url_attribute(element, 'src', base_url)
|
||
alt = element.get('alt')
|
||
if src:
|
||
image = get_image_from_uri(src)
|
||
if image is not None:
|
||
return [make_replaced_box(element, box, image)]
|
||
else:
|
||
# Invalid image, use the alt-text.
|
||
if alt:
|
||
box.children = [boxes.TextBox.anonymous_from(box, alt)]
|
||
return [box]
|
||
elif alt == '':
|
||
# The element represents nothing
|
||
return []
|
||
else:
|
||
assert alt is None
|
||
# TODO: find some indicator that an image is missing.
|
||
# For now, just remove the image.
|
||
return []
|
||
else:
|
||
if alt:
|
||
box.children = [boxes.TextBox.anonymous_from(box, alt)]
|
||
return [box]
|
||
else:
|
||
return []
|
||
|
||
|
||
@handler('embed')
|
||
def handle_embed(element, box, get_image_from_uri, base_url):
|
||
"""Handle ``<embed>`` elements, return either an image or nothing.
|
||
|
||
See: https://www.w3.org/TR/html5/embedded-content-0.html#the-embed-element
|
||
|
||
"""
|
||
src = get_url_attribute(element, 'src', base_url)
|
||
type_ = element.get('type', '').strip()
|
||
if src:
|
||
image = get_image_from_uri(src, type_)
|
||
if image is not None:
|
||
return [make_replaced_box(element, box, image)]
|
||
# No fallback.
|
||
return []
|
||
|
||
|
||
@handler('object')
|
||
def handle_object(element, box, get_image_from_uri, base_url):
|
||
"""Handle ``<object>`` elements, return either an image or the fallback
|
||
content.
|
||
|
||
See: https://www.w3.org/TR/html5/embedded-content-0.html#the-object-element
|
||
|
||
"""
|
||
data = get_url_attribute(element, 'data', base_url)
|
||
type_ = element.get('type', '').strip()
|
||
if data:
|
||
image = get_image_from_uri(data, type_)
|
||
if image is not None:
|
||
return [make_replaced_box(element, box, image)]
|
||
# The element’s children are the fallback.
|
||
return [box]
|
||
|
||
|
||
def integer_attribute(element, box, name, minimum=1):
|
||
"""Read an integer attribute from the HTML element and set it on the box.
|
||
|
||
"""
|
||
value = element.get(name, '').strip()
|
||
if value:
|
||
try:
|
||
value = int(value)
|
||
except ValueError:
|
||
pass
|
||
else:
|
||
if value >= minimum:
|
||
setattr(box, name, value)
|
||
|
||
|
||
@handler('colgroup')
|
||
def handle_colgroup(element, box, _get_image_from_uri, _base_url):
|
||
"""Handle the ``span`` attribute."""
|
||
if isinstance(box, boxes.TableColumnGroupBox):
|
||
if any(child.tag == 'col' for child in element):
|
||
box.span = None # sum of the children’s spans
|
||
else:
|
||
integer_attribute(element, box, 'span')
|
||
box.children = (
|
||
boxes.TableColumnBox.anonymous_from(box, [])
|
||
for _i in range(box.span))
|
||
return [box]
|
||
|
||
|
||
@handler('col')
|
||
def handle_col(element, box, _get_image_from_uri, _base_url):
|
||
"""Handle the ``span`` attribute."""
|
||
if isinstance(box, boxes.TableColumnBox):
|
||
integer_attribute(element, box, 'span')
|
||
if box.span > 1:
|
||
# Generate multiple boxes
|
||
# http://lists.w3.org/Archives/Public/www-style/2011Nov/0293.html
|
||
return [box.copy() for _i in range(box.span)]
|
||
return [box]
|
||
|
||
|
||
@handler('th')
|
||
@handler('td')
|
||
def handle_td(element, box, _get_image_from_uri, _base_url):
|
||
"""Handle the ``colspan``, ``rowspan`` attributes."""
|
||
if isinstance(box, boxes.TableCellBox):
|
||
# HTML 4.01 gives special meaning to colspan=0
|
||
# http://www.w3.org/TR/html401/struct/tables.html#adef-rowspan
|
||
# but HTML 5 removed it
|
||
# http://www.w3.org/TR/html5/tabular-data.html#attr-tdth-colspan
|
||
# rowspan=0 is still there though.
|
||
integer_attribute(element, box, 'colspan')
|
||
integer_attribute(element, box, 'rowspan', minimum=0)
|
||
return [box]
|
||
|
||
|
||
@handler('a')
|
||
def handle_a(element, box, _get_image_from_uri, base_url):
|
||
"""Handle the ``rel`` attribute."""
|
||
box.is_attachment = element_has_link_type(element, 'attachment')
|
||
return [box]
|
||
|
||
|
||
def find_base_url(html_document, fallback_base_url):
|
||
"""Return the base URL for the document.
|
||
|
||
See http://www.w3.org/TR/html5/urls.html#document-base-url
|
||
|
||
"""
|
||
first_base_element = next(iter(html_document.iter('base')), None)
|
||
if first_base_element is not None:
|
||
href = first_base_element.get('href', '').strip()
|
||
if href:
|
||
return urljoin(fallback_base_url, href)
|
||
return fallback_base_url
|
||
|
||
|
||
def get_html_metadata(wrapper_element, base_url):
|
||
"""
|
||
Relevant specs:
|
||
|
||
http://www.whatwg.org/html#the-title-element
|
||
http://www.whatwg.org/html#standard-metadata-names
|
||
http://wiki.whatwg.org/wiki/MetaExtensions
|
||
http://microformats.org/wiki/existing-rel-values#HTML5_link_type_extensions
|
||
|
||
"""
|
||
title = None
|
||
description = None
|
||
generator = None
|
||
keywords = []
|
||
authors = []
|
||
created = None
|
||
modified = None
|
||
attachments = []
|
||
for element in wrapper_element.query_all('title', 'meta', 'link'):
|
||
element = element.etree_element
|
||
if element.tag == 'title' and title is None:
|
||
title = get_child_text(element)
|
||
elif element.tag == 'meta':
|
||
name = ascii_lower(element.get('name', ''))
|
||
content = element.get('content', '')
|
||
if name == 'keywords':
|
||
for keyword in map(strip_whitespace, content.split(',')):
|
||
if keyword not in keywords:
|
||
keywords.append(keyword)
|
||
elif name == 'author':
|
||
authors.append(content)
|
||
elif name == 'description' and description is None:
|
||
description = content
|
||
elif name == 'generator' and generator is None:
|
||
generator = content
|
||
elif name == 'dcterms.created' and created is None:
|
||
created = parse_w3c_date(name, content)
|
||
elif name == 'dcterms.modified' and modified is None:
|
||
modified = parse_w3c_date(name, content)
|
||
elif element.tag == 'link' and element_has_link_type(
|
||
element, 'attachment'):
|
||
url = get_url_attribute(element, 'href', base_url)
|
||
attachment_title = element.get('title', None)
|
||
if url is None:
|
||
LOGGER.error('Missing href in <link rel="attachment">')
|
||
else:
|
||
attachments.append((url, attachment_title))
|
||
return dict(title=title, description=description, generator=generator,
|
||
keywords=keywords, authors=authors,
|
||
created=created, modified=modified,
|
||
attachments=attachments)
|
||
|
||
|
||
def strip_whitespace(string):
|
||
"""Use the HTML definition of "space character",
|
||
not all Unicode Whitespace.
|
||
|
||
http://www.whatwg.org/html#strip-leading-and-trailing-whitespace
|
||
http://www.whatwg.org/html#space-character
|
||
|
||
"""
|
||
return string.strip(HTML_WHITESPACE)
|
||
|
||
|
||
# YYYY (eg 1997)
|
||
# YYYY-MM (eg 1997-07)
|
||
# YYYY-MM-DD (eg 1997-07-16)
|
||
# YYYY-MM-DDThh:mmTZD (eg 1997-07-16T19:20+01:00)
|
||
# YYYY-MM-DDThh:mm:ssTZD (eg 1997-07-16T19:20:30+01:00)
|
||
# YYYY-MM-DDThh:mm:ss.sTZD (eg 1997-07-16T19:20:30.45+01:00)
|
||
|
||
W3C_DATE_RE = re.compile('''
|
||
^
|
||
[ \t\n\f\r]*
|
||
(?P<year>\\d\\d\\d\\d)
|
||
(?:
|
||
-(?P<month>0\\d|1[012])
|
||
(?:
|
||
-(?P<day>[012]\\d|3[01])
|
||
(?:
|
||
T(?P<hour>[01]\\d|2[0-3])
|
||
:(?P<minute>[0-5]\\d)
|
||
(?:
|
||
:(?P<second>[0-5]\\d)
|
||
(?:\\.\\d+)? # Second fraction, ignored
|
||
)?
|
||
(?:
|
||
Z | # UTC
|
||
(?P<tz_hour>[+-](?:[01]\\d|2[0-3]))
|
||
:(?P<tz_minute>[0-5]\\d)
|
||
)
|
||
)?
|
||
)?
|
||
)?
|
||
[ \t\n\f\r]*
|
||
$
|
||
''', re.VERBOSE)
|
||
|
||
|
||
def parse_w3c_date(meta_name, string):
|
||
"""http://www.w3.org/TR/NOTE-datetime"""
|
||
if W3C_DATE_RE.match(string):
|
||
return string
|
||
else:
|
||
LOGGER.warning(
|
||
'Invalid date in <meta name="%s"> %r', meta_name, string)
|