6.5 将字典转换为XML

问题

你想使用一个Python字典存储数据,并将它转换成XML格式。

解决方案

尽管 xml.etree.ElementTree 库通常用来做解析工作,其实它也可以创建XML文档。例如,考虑如下这个函数:

  1. from xml.etree.ElementTree import Element
  2.  
  3. def dict_to_xml(tag, d):
  4. '''
  5. Turn a simple dict of key/value pairs into XML
  6. '''
  7. elem = Element(tag)
  8. for key, val in d.items():
  9. child = Element(key)
  10. child.text = str(val)
  11. elem.append(child)
  12. return elem

下面是一个使用例子:

  1. >>> s = { 'name': 'GOOG', 'shares': 100, 'price':490.1 }
  2. >>> e = dict_to_xml('stock', s)
  3. >>> e
  4. <Element 'stock' at 0x1004b64c8>
  5. >>>

转换结果是一个 Element 实例。对于I/O操作,使用 xml.etree.ElementTree 中的 tostring()函数很容易就能将它转换成一个字节字符串。例如:

  1. >>> from xml.etree.ElementTree import tostring
  2. >>> tostring(e)
  3. b'<stock><price>490.1</price><shares>100</shares><name>GOOG</name></stock>'
  4. >>>

如果你想给某个元素添加属性值,可以使用 set() 方法:

  1. >>> e.set('_id','1234')
  2. >>> tostring(e)
  3. b'<stock _id="1234"><price>490.1</price><shares>100</shares><name>GOOG</name>
  4. </stock>'
  5. >>>

如果你还想保持元素的顺序,可以考虑构造一个 OrderedDict 来代替一个普通的字典。请参考1.7小节。

讨论

当创建XML的时候,你被限制只能构造字符串类型的值。例如:

  1. def dict_to_xml_str(tag, d):
  2. '''
  3. Turn a simple dict of key/value pairs into XML
  4. '''
  5. parts = ['<{}>'.format(tag)]
  6. for key, val in d.items():
  7. parts.append('<{0}>{1}</{0}>'.format(key,val))
  8. parts.append('</{}>'.format(tag))
  9. return ''.join(parts)

问题是如果你手动的去构造的时候可能会碰到一些麻烦。例如,当字典的值中包含一些特殊字符的时候会怎样呢?

  1. >>> d = { 'name' : '<spam>' }
  2.  
  3. >>> # String creation
  4. >>> dict_to_xml_str('item',d)
  5. '<item><name><spam></name></item>'
  6.  
  7. >>> # Proper XML creation
  8. >>> e = dict_to_xml('item',d)
  9. >>> tostring(e)
  10. b'<item><name>&lt;spam&gt;</name></item>'
  11. >>>

注意到程序的后面那个例子中,字符 ‘<’ 和 ‘>’ 被替换成了 &lt;&gt;

下面仅供参考,如果你需要手动去转换这些字符,可以使用 xml.sax.saxutils 中的 escape()unescape() 函数。例如:

  1. >>> from xml.sax.saxutils import escape, unescape
  2. >>> escape('<spam>')
  3. '&lt;spam&gt;'
  4. >>> unescape(_)
  5. '<spam>'
  6. >>>

除了能创建正确的输出外,还有另外一个原因推荐你创建 Element 实例而不是字符串,那就是使用字符串组合构造一个更大的文档并不是那么容易。而 Element 实例可以不用考虑解析XML文本的情况下通过多种方式被处理。也就是说,你可以在一个高级数据结构上完成你所有的操作,并在最后以字符串的形式将其输出。

原文:

http://python3-cookbook.readthedocs.io/zh_CN/latest/c06/p05_turning_dictionary_into_xml.html