You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
30 lines
919 B
Python
30 lines
919 B
Python
6 years ago
|
from __future__ import absolute_import, division, unicode_literals
|
||
|
|
||
|
from . import base
|
||
|
|
||
|
from collections import OrderedDict
|
||
|
|
||
|
|
||
|
def _attr_key(attr):
|
||
|
"""Return an appropriate key for an attribute for sorting
|
||
|
|
||
|
Attributes have a namespace that can be either ``None`` or a string. We
|
||
|
can't compare the two because they're different types, so we convert
|
||
|
``None`` to an empty string first.
|
||
|
|
||
|
"""
|
||
|
return (attr[0][0] or ''), attr[0][1]
|
||
|
|
||
|
|
||
|
class Filter(base.Filter):
|
||
|
"""Alphabetizes attributes for elements"""
|
||
|
def __iter__(self):
|
||
|
for token in base.Filter.__iter__(self):
|
||
|
if token["type"] in ("StartTag", "EmptyTag"):
|
||
|
attrs = OrderedDict()
|
||
|
for name, value in sorted(token["data"].items(),
|
||
|
key=_attr_key):
|
||
|
attrs[name] = value
|
||
|
token["data"] = attrs
|
||
|
yield token
|