mirror of
https://github.com/rembo10/headphones.git
synced 2026-03-22 20:59:27 +00:00
6e2338184e
- Delete from Releases when deleting artist/album - Searcher - Size limits not quite working - Searcher - 1st newznab used even if disabled - Rutracker search stopped working for me, fixed by updating Beautiful Soup. Moved bs4 and html5lib to lib and ensured (I think) it’s imported from the right place
64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
from __future__ import absolute_import, division, unicode_literals
|
|
|
|
from xml.dom.pulldom import START_ELEMENT, END_ELEMENT, \
|
|
COMMENT, IGNORABLE_WHITESPACE, CHARACTERS
|
|
|
|
from . import _base
|
|
|
|
from ..constants import voidElements
|
|
|
|
|
|
class TreeWalker(_base.TreeWalker):
|
|
def __iter__(self):
|
|
ignore_until = None
|
|
previous = None
|
|
for event in self.tree:
|
|
if previous is not None and \
|
|
(ignore_until is None or previous[1] is ignore_until):
|
|
if previous[1] is ignore_until:
|
|
ignore_until = None
|
|
for token in self.tokens(previous, event):
|
|
yield token
|
|
if token["type"] == "EmptyTag":
|
|
ignore_until = previous[1]
|
|
previous = event
|
|
if ignore_until is None or previous[1] is ignore_until:
|
|
for token in self.tokens(previous, None):
|
|
yield token
|
|
elif ignore_until is not None:
|
|
raise ValueError("Illformed DOM event stream: void element without END_ELEMENT")
|
|
|
|
def tokens(self, event, next):
|
|
type, node = event
|
|
if type == START_ELEMENT:
|
|
name = node.nodeName
|
|
namespace = node.namespaceURI
|
|
attrs = {}
|
|
for attr in list(node.attributes.keys()):
|
|
attr = node.getAttributeNode(attr)
|
|
attrs[(attr.namespaceURI, attr.localName)] = attr.value
|
|
if name in voidElements:
|
|
for token in self.emptyTag(namespace,
|
|
name,
|
|
attrs,
|
|
not next or next[1] is not node):
|
|
yield token
|
|
else:
|
|
yield self.startTag(namespace, name, attrs)
|
|
|
|
elif type == END_ELEMENT:
|
|
name = node.nodeName
|
|
namespace = node.namespaceURI
|
|
if name not in voidElements:
|
|
yield self.endTag(namespace, name)
|
|
|
|
elif type == COMMENT:
|
|
yield self.comment(node.nodeValue)
|
|
|
|
elif type in (IGNORABLE_WHITESPACE, CHARACTERS):
|
|
for token in self.text(node.nodeValue):
|
|
yield token
|
|
|
|
else:
|
|
yield self.unknown(type)
|