You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
1.6 KiB
58 lines
1.6 KiB
from django.core.exceptions import ImproperlyConfigured
|
|
from django.utils.encoding import smart_str
|
|
from django.utils.functional import cached_property
|
|
|
|
from compressor.exceptions import ParserError
|
|
from compressor.parser import ParserBase
|
|
|
|
|
|
class LxmlParser(ParserBase):
|
|
"""
|
|
LxmlParser will use `lxml.html` parser to parse rendered contents of
|
|
{% compress %} tag.
|
|
"""
|
|
|
|
def __init__(self, content):
|
|
try:
|
|
from lxml.html import fromstring
|
|
from lxml.etree import tostring
|
|
except ImportError as err:
|
|
raise ImproperlyConfigured("Error while importing lxml: %s" % err)
|
|
except Exception as err:
|
|
raise ParserError("Error while initializing parser: %s" % err)
|
|
|
|
self.fromstring = fromstring
|
|
self.tostring = tostring
|
|
super().__init__(content)
|
|
|
|
@cached_property
|
|
def tree(self):
|
|
"""
|
|
Document tree.
|
|
"""
|
|
content = "<root>%s</root>" % self.content
|
|
tree = self.fromstring(content)
|
|
self.tostring(tree, encoding=str)
|
|
return tree
|
|
|
|
def css_elems(self):
|
|
return self.tree.xpath(
|
|
'//link[re:test(@rel, "^stylesheet$", "i")]|style',
|
|
namespaces={"re": "http://exslt.org/regular-expressions"},
|
|
)
|
|
|
|
def js_elems(self):
|
|
return self.tree.findall("script")
|
|
|
|
def elem_attribs(self, elem):
|
|
return elem.attrib
|
|
|
|
def elem_content(self, elem):
|
|
return smart_str(elem.text)
|
|
|
|
def elem_name(self, elem):
|
|
return elem.tag
|
|
|
|
def elem_str(self, elem):
|
|
return smart_str(self.tostring(elem, method="html", encoding=str))
|