lazy import copyreg

lazy from lxml import html
lazy from lxml.html._element import HtmlElement


def _reduce_html_element(elem: HtmlElement):
    raw_html = html.tostring(elem)
    return _rebuild_html_element, (raw_html,)


def _rebuild_html_element(raw_html: str) -> HtmlElement:
    return html.fromstring(raw_html)


def register() -> None:
    copyreg.pickle(HtmlElement, _reduce_html_element)