diff options
Diffstat (limited to 'bs4/builder')
| -rw-r--r-- | bs4/builder/__init__.py | 20 | ||||
| -rw-r--r-- | bs4/builder/_html5lib.py | 15 |
2 files changed, 32 insertions, 3 deletions
diff --git a/bs4/builder/__init__.py b/bs4/builder/__init__.py index 9f4f59e..4c22b86 100644 --- a/bs4/builder/__init__.py +++ b/bs4/builder/__init__.py @@ -1,8 +1,10 @@ from collections import defaultdict +import itertools import sys from bs4.element import ( CharsetMetaAttributeValue, ContentMetaAttributeValue, + whitespace_re ) __all__ = [ @@ -140,6 +142,24 @@ class TreeBuilder(object): def set_up_substitutions(self, tag): return False + def _replace_cdata_list_attribute_values(self, tag_name, attrs): + """Replaces class="foo bar" with class=["foo", "bar"] + + Modifies its input in place. + """ + if self.cdata_list_attributes: + universal = self.cdata_list_attributes.get('*', []) + tag_specific = self.cdata_list_attributes.get( + tag_name.lower(), []) + for cdata_list_attr in itertools.chain(universal, tag_specific): + if cdata_list_attr in dict(attrs): + # Basically, we have a "class" attribute whose + # value is a whitespace-separated list of CSS + # classes. Split it into a list. + value = attrs[cdata_list_attr] + values = whitespace_re.split(value) + attrs[cdata_list_attr] = values + return attrs class SAXTreeBuilder(TreeBuilder): """A Beautiful Soup treebuilder that listens for SAX events.""" diff --git a/bs4/builder/_html5lib.py b/bs4/builder/_html5lib.py index 2d6fe31..6001e38 100644 --- a/bs4/builder/_html5lib.py +++ b/bs4/builder/_html5lib.py @@ -139,11 +139,20 @@ class Element(html5lib.treebuilders._base.Node): return AttrList(self.element) def setAttributes(self, attributes): - if attributes is not None and attributes != {}: + if attributes is not None and len(attributes) > 0: + + converted_attributes = [] for name, value in list(attributes.items()): if isinstance(name, tuple): - name = NamespacedAttribute(*name) - self.element[name] = value + new_name = NamespacedAttribute(*name) + del attributes[name] + attributes[new_name] = value + + self.soup.builder._replace_cdata_list_attribute_values( + self.name, attributes) + for name, value in attributes.items(): + self.element[name] = value + # The attributes may contain variables that need substitution. # Call set_up_substitutions manually. # |