55 lines
1.4 KiB
Python
55 lines
1.4 KiB
Python
import re
|
|
from typing import Optional
|
|
|
|
from draftjs_exporter.engines.base import Attr, DOMEngine
|
|
from draftjs_exporter.types import HTML, Tag
|
|
|
|
try:
|
|
from lxml import etree, html
|
|
except ImportError:
|
|
pass
|
|
|
|
NSMAP = {"xlink": "http://www.w3.org/1999/xlink"}
|
|
|
|
RENDER_RE = re.compile(r"</?fragment>")
|
|
|
|
|
|
class DOM_LXML(DOMEngine):
|
|
"""
|
|
lxml implementation of the DOM API.
|
|
"""
|
|
|
|
@staticmethod
|
|
def create_tag(type_: Tag, attr: Optional[Attr] = None) -> etree.Element:
|
|
nsmap = None
|
|
|
|
if attr:
|
|
if "xlink:href" in attr:
|
|
attr[f"{{{NSMAP['xlink']}}}href"] = attr.pop("xlink:href")
|
|
nsmap = NSMAP
|
|
|
|
return etree.Element(type_, attrib=attr, nsmap=nsmap)
|
|
|
|
@staticmethod
|
|
def parse_html(markup: HTML) -> etree.Element:
|
|
return html.fromstring(markup)
|
|
|
|
@staticmethod
|
|
def append_child(elt: etree.Element, child: etree.Element) -> None:
|
|
if hasattr(child, "tag"):
|
|
elt.append(child)
|
|
else:
|
|
c = etree.Element("fragment")
|
|
c.text = child
|
|
elt.append(c)
|
|
|
|
@staticmethod
|
|
def render(elt: etree.Element) -> HTML:
|
|
return RENDER_RE.sub(
|
|
"", etree.tostring(elt, method="html", encoding="unicode")
|
|
)
|
|
|
|
@staticmethod
|
|
def render_debug(elt: etree.Element) -> HTML:
|
|
return etree.tostring(elt, method="html", encoding="unicode")
|