blob: 2963a0294c965d05028fe82b531da9232f686d48 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
from lxml import etree
from beautifulsoup.element import Comment
from beautifulsoup.builder import TreeBuilder
class LXMLTreeBuilder(TreeBuilder):
def __init__(self, parser_class=etree.HTMLParser, self_closing_tags=[]):
self.parser = parser_class(target=self)
self.self_closing_tags = self_closing_tags
self.soup = None
def isSelfClosingTag(self, name):
return name in self.self_closing_tags
def feed(self, markup):
self.parser.feed(markup)
self.parser.close()
def close(self):
pass
def start(self, name, attrs):
self.soup.handle_starttag(name, attrs)
def end(self, name):
self.soup.handle_endtag(name)
def data(self, content):
self.soup.handle_data(content)
def comment(self, content):
"Handle comments as Comment objects."
self.soup.endData()
self.soup.handle_data(content)
self.soup.endData(Comment)
|
