mirror of
https://github.com/rembo10/headphones.git
synced 2026-03-23 13:19: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
70 lines
2.2 KiB
Python
70 lines
2.2 KiB
Python
from __future__ import absolute_import, division, unicode_literals
|
|
|
|
from genshi.core import QName
|
|
from genshi.core import START, END, XML_NAMESPACE, DOCTYPE, TEXT
|
|
from genshi.core import START_NS, END_NS, START_CDATA, END_CDATA, PI, COMMENT
|
|
|
|
from . import _base
|
|
|
|
from ..constants import voidElements, namespaces
|
|
|
|
|
|
class TreeWalker(_base.TreeWalker):
|
|
def __iter__(self):
|
|
# Buffer the events so we can pass in the following one
|
|
previous = None
|
|
for event in self.tree:
|
|
if previous is not None:
|
|
for token in self.tokens(previous, event):
|
|
yield token
|
|
previous = event
|
|
|
|
# Don't forget the final event!
|
|
if previous is not None:
|
|
for token in self.tokens(previous, None):
|
|
yield token
|
|
|
|
def tokens(self, event, next):
|
|
kind, data, pos = event
|
|
if kind == START:
|
|
tag, attribs = data
|
|
name = tag.localname
|
|
namespace = tag.namespace
|
|
converted_attribs = {}
|
|
for k, v in attribs:
|
|
if isinstance(k, QName):
|
|
converted_attribs[(k.namespace, k.localname)] = v
|
|
else:
|
|
converted_attribs[(None, k)] = v
|
|
|
|
if namespace == namespaces["html"] and name in voidElements:
|
|
for token in self.emptyTag(namespace, name, converted_attribs,
|
|
not next or next[0] != END
|
|
or next[1] != tag):
|
|
yield token
|
|
else:
|
|
yield self.startTag(namespace, name, converted_attribs)
|
|
|
|
elif kind == END:
|
|
name = data.localname
|
|
namespace = data.namespace
|
|
if name not in voidElements:
|
|
yield self.endTag(namespace, name)
|
|
|
|
elif kind == COMMENT:
|
|
yield self.comment(data)
|
|
|
|
elif kind == TEXT:
|
|
for token in self.text(data):
|
|
yield token
|
|
|
|
elif kind == DOCTYPE:
|
|
yield self.doctype(*data)
|
|
|
|
elif kind in (XML_NAMESPACE, DOCTYPE, START_NS, END_NS,
|
|
START_CDATA, END_CDATA, PI):
|
|
pass
|
|
|
|
else:
|
|
yield self.unknown(kind)
|