用lxml编辑html代码而不用

时间:2013-11-10 09:23:55

标签: python lxml

假设我有html代码。它坏了(它是从另一个程序给出的)。我必须找到其中的所有图像,并将src属性更改为某些代码。这就是全部 - 它应该在决赛中保持不变。

这是我的代码:

# -*- coding: utf-8 -*-

from lxml import etree
from StringIO import StringIO

def test(text):
    text = analyze_dir_template_imgs(text)
    return text

def analyze_dir_template_imgs(text):
    ts = [
        {
            'name':'img',
            'path':"//img[@src]",
            'attrib': 'src',
        },
    ]

    parser = etree.HTMLParser()

    for t in ts:
        tree = etree.parse(StringIO(text), parser)
        path = t['path']
        name = t['name']
        attrib = t.get('attrib', False)

        text = str_repr(tree.getroot())

        #text_to_file(text, 'temp/template_dir.html', 0, 'wb')

        links = tree.xpath(path)

        i = 0
        for l in links:
            i+=1
            #print
            #print l
            y0 = l.attrib[attrib]
            print y0

            x0 = str_repr(l)
            if name=='img':
                repl_to = '[image]'

                x0_new = x0.replace(y0, repl_to)

                text = text.replace(x0, x0_new)
    print text
    return text

def str_repr(el, enc='utf-8'):
    text = etree.tostring(el, pretty_print=True, method='xml', encoding=enc)
    return text

if __name__ == "__main__":
    text = u'''
<a href="[link]">[name]</a>
            </h2>

<img width="280" style= "margin: 10px; float:left;" alt ="привет мир" src = "/[[template]]/image.jpg"><p>[[body]]</p>

<div class="jcomments-links">
<a href="[link]" class="readmore-link">more...</a> </div>



<div class="item-separator">'''

    test(text)

但问题是在此过程中lxml会更正html代码。我的意思是结果是

<html><body>
<a href="%5Blink%5D">[name]</a>
...

但我的结果应该从

开始
<a href="[link]">[name]</a>

因为它是初始代码的开始。

这很糟糕,因为我的任务只是改变src,这个新的更改代码将被发送到另一个脚本,它将收集不同的代码片段。

另一个问题是,y0 = l.attrib[attrib]提供了/[[template]]/image.jpg,但x0 = str_repr(l)提供了<img width="280" style="margin: 10px; float:left;" alt="привет мир" src="/%5B%5Btemplate%5D%5D/image.jpg"><p>[[body]]</p> - 正如您所见,[等特殊符号更改为{{1}但是我希望得到原始html代码中的这部分。

如何使用lxml正确解决此问题?感谢。

1 个答案:

答案 0 :(得分:2)

替换以下行:

text = etree.tostring(el, pretty_print=True, method='html', encoding=enc)

使用:

text = etree.tostring(el, pretty_print=True, method='xml', encoding=enc)

>>> root = etree.fromstring('<a href="[link]">more...</a>')
>>> etree.tostring(root, method='html')
'<a href="%5Blink%5D">more...</a>'
>>> etree.tostring(root, method='xml')
'<a href="[link]">more...</a>'