Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Replacing elements with lxml.html

Tags:

python

lxml

I'm fairly new to lxml and HTML Parsers as a whole. I was wondering if there is a way to replace an element within a tree with another element...

For example I have:

body = """<code> def function(arg): print arg </code> Blah blah blah <code> int main() { return 0; } </code> """

doc = lxml.html.fromstring(body)
codeblocks = doc.cssselect('code')

for block in codeblocks:
  lexer = guess_lexer(block.text_content())
  hilited = highlight(block.text_content(), lexer, HtmlFormatter())
  doc.replace(block, hilited)

I want to do something along those lines, but this results in a "TypeError" because "hilited" isn't an lxml.etree._Element.

Is this feasible?

Regards,

like image 961
tsoporan Avatar asked Nov 28 '09 15:11

tsoporan


1 Answers

Regarding lxml,

In doc.replace(block, hilited)

block is the lxml's Element object, hilited is string, you cannot replace that.

There is 2 ways to do that

block.text=hilited 

or

body=body.replace(block.text,hilited)
like image 171
YOU Avatar answered Sep 21 '22 13:09

YOU