如何在lxml.html树中插入HTML元素

时间:2013-06-21 12:49:39

标签: python lxml lxml.html

我正在使用python 3.3和lxml 3.2.0

问题: 我在变量webpageString = "<html><head></head><body>webpage content</body></html>"中有一个网页 我想在两个标头标签之间插入一个css链接标签,以便我得到 webpageString = "<html><head><link rel='stylesheet' type='text/css'></head><body>webpage content</body></html>"

我写了以下代码:

def addCssCode(self):
    tree = html.fromstring(self.article)
    headTag = tree.xpath("//head")
    #htmlTag = tree.getroot()

    if headTag is None:
        pass    #insert the head tag first

    cssLinkString = "<link rel='stylesheet' type='text/css' href='"+ self.cssLocation+"'>"
    headTag[0].insert(1, html.HtmlElement(cssLinkString))
    print(cssLinkString)
    self.article = html.tostring(tree).decode("utf-8")

导致插入 -

    <HtmlElement>&lt; link rel='stylesheet' type='text/css' href='cssCode.css' &gt;</HtmlElement>

我也尝试在下一页解决相同的问题,但它也没有用。 python lxml append element after another element

我该如何解决这个问题? 感谢

1 个答案:

答案 0 :(得分:0)

使用.insert / .append方法。

import lxml.html

def add_css_code(webpageString, linkString):
    root = lxml.html.fromstring(webpageString)
    link = lxml.html.fromstring(linkString).find('.//link')
    head = root.find('.//head')
    title = head.find('title')
    if title == None:
        where = 0
    else:
        where = head.index(title) + 1
    head.insert(where, link)
    return lxml.html.tostring(root)

webpageString1 = "<html><head><title>test</title></head><body>webpage content</body></html>"
webpageString2 = "<html><head></head><body>webpage content</body></html>"
linkString = "<link rel='stylesheet' type='text/css'>"

print(add_css_code(webpageString1, linkString))
print(add_css_code(webpageString2, linkString))