XML.py 文件源码

def filter(self, text):
    wanted_dm = ['created_at', 'id', 'text', 'source', 'sender_id', 'recipient_id', 'sender_screen_name', 'recipient_screen_name']
    root = ET.fromstring(text)
    builder = ET.TreeBuilder()
    builder.start('direct-messages', {'type': 'array'})
    for dm in root.findall('direct_message'):
      builder.start('direct_message', {})
      for tag in wanted_dm:
        copy_element(builder, dm, tag)
      builder.end('direct_message')
    builder.end('direct-messages')
    return ET.tostring(builder.close())