57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
from __future__ import absolute_import
|
|
from django.core.exceptions import ImproperlyConfigured
|
|
from django.utils.encoding import smart_unicode
|
|
|
|
from compressor.exceptions import ParserError
|
|
from compressor.parser import ParserBase
|
|
from compressor.utils.decorators import cached_property
|
|
|
|
|
|
class LxmlParser(ParserBase):
|
|
|
|
def __init__(self, content):
|
|
try:
|
|
from lxml.html import fromstring, soupparser
|
|
from lxml.etree import tostring
|
|
self.fromstring = fromstring
|
|
self.soupparser = soupparser
|
|
self.tostring = tostring
|
|
except ImportError, err:
|
|
raise ImproperlyConfigured("Error while importing lxml: %s" % err)
|
|
except Exception, err:
|
|
raise ParserError("Error while initializing Parser: %s" % err)
|
|
super(LxmlParser, self).__init__(content)
|
|
|
|
@cached_property
|
|
def tree(self):
|
|
content = '<root>%s</root>' % self.content
|
|
tree = self.fromstring(content)
|
|
try:
|
|
ignore = self.tostring(tree, encoding=unicode)
|
|
except UnicodeDecodeError:
|
|
tree = self.soupparser.fromstring(content)
|
|
return tree
|
|
|
|
def css_elems(self):
|
|
return self.tree.xpath('link[@rel="stylesheet"]|style')
|
|
|
|
def js_elems(self):
|
|
return self.tree.findall('script')
|
|
|
|
def elem_attribs(self, elem):
|
|
return elem.attrib
|
|
|
|
def elem_content(self, elem):
|
|
return smart_unicode(elem.text)
|
|
|
|
def elem_name(self, elem):
|
|
return elem.tag
|
|
|
|
def elem_str(self, elem):
|
|
elem_as_string = smart_unicode(
|
|
self.tostring(elem, method='html', encoding=unicode))
|
|
if elem.tag == 'link':
|
|
# This makes testcases happy
|
|
return elem_as_string.replace('>', ' />')
|
|
return elem_as_string
|