如何替换树中的元素？（元素树/降价）

Question

在elementtree的迭代期间如何替换元素？我正在为markdown编写一个treeprocessor，希望包装一个元素。在

<pre class='inner'>...</pre>

应该变成

^{pr2}$

我使用getiterator('pre')来查找元素，但我不知道如何包装它。问题是用新的包装器替换找到的元素，但将现有元素保留为子元素。在

Answer 1

这有点棘手。首先，您需要获得this previous question中描述的父元素。在

parent_map = dict((c, p) for p in tree.getiterator() for c in p)

如果您可以让markdown使用lxml，那么这就简单了一点，我相信lxml元素已经知道它们的父元素了。在

现在，当您从迭代中获取元素时，还可以获得父元素：

^{pr2}$

注意，我已经将迭代器从树变成了一个列表，我们不想在遍历树时修改它。那可能是个麻烦。在

最后，您可以移动元素：

def wrap_elem(parent, elem)
    parent_index = list(parent).index(elem)
    parent.remove(elem)
    new_elem = ET.Element('div', attrib={'class': 'wrapper'})
    parent.insert(parent_index, new_elem)
    new_elem.append(elem)

*请注意，我还没有完全测试过这段代码。。。如果你发现什么虫子就告诉我。在