扩展 Element 和 ElementTree 的功能

Question

我想从xml.etree 扩展Element 和ElementTree 类的功能并将它们与xml.etree.ElementTree.parse() 一起使用。

经过几次尝试，我已经设法为这个问题创建了一个解决方案，但我想知道是否有更好的解决方案，或者这个解决方案是否有任何隐患。

FooElementTree.py

import xml.etree
from xml.etree.ElementTree import Element
from xml.etree.ElementTree import ElementTree

class FooElement(Element):
    def __repr__(self):
        return "<FooElement %s at 0x%x>" % (repr(self.tag), id(self))

class FooElementTree(ElementTree):
    pass

xml.etree.ElementTree.Element = FooElement
xml.etree.ElementTree.ElementTree = FooElementTree
from xml.etree.ElementTree import parse

用法：

>>> import FooElementTree
>>> e = FooElementTree.parse('xml.cfg')
>>> e
<FooElementTree.FooElementTree object at 0x023AB650>
>>> r = e.getroot()
>>> r
<FooElement 'configuration' at 0x23c5470>

Answer 1

这就是热补丁工作的方式。但请注意，必须在导入任何其他也使用 ElementTree 的模块（或子模块）之前应用补丁。