Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Beautiful Soup 4: How to replace a tag with text and another tag?

I want to replace a tag with another tag and put the contents of the old tag before the new one. For example:

I want to change this:

<html>
<body>
<p>This is the <span id="1">first</span> paragraph</p>
<p>This is the <span id="2">second</span> paragraph</p>
</body>
</html>

into this:

<html>
<body>
<p>This is the first<sup>1</sup> paragraph</p>
<p>This is the second<sup>2</sup> paragraph</p>
</body>
</html>

I can easily find all spans with find_all(), get the number from the id attribute and replace one tag with another tag using replace_with(), but how do I replace a tag with text and a new tag or insert text before a replaced tag?

like image 997
Nemo XXX Avatar asked Nov 18 '14 23:11

Nemo XXX


1 Answers

The idea is to find every span tag with id attribute (span[id] CSS Selector), use insert_after() to insert a sup tag after it and unwrap() to replace the tag with it's contents:

from bs4 import BeautifulSoup

data = """
<html>
<body>
<p>This is the <span id="1">first</span> paragraph</p>
<p>This is the <span id="2">second</span> paragraph</p>
</body>
</html>
"""

soup = BeautifulSoup(data)
for span in soup.select('span[id]'):
    # insert sup tag after the span
    sup = soup.new_tag('sup')
    sup.string = span['id']
    span.insert_after(sup)

    # replace the span tag with it's contents
    span.unwrap()

print soup

Prints:

<html>
<body>
<p>This is the first<sup>1</sup> paragraph</p>
<p>This is the second<sup>2</sup> paragraph</p>
</body>
</html>
like image 133
alecxe Avatar answered Oct 04 '22 07:10

alecxe