Python数组中的数据输出

49 阅读2分钟

一位初次学习Python的用户遇到了一个问题,他想从API获取数据并将其输出为XML格式。他使用了一个数组(projectData)来存储数据,并编写了一些代码来输出XML,但怀疑这种方法效率不高且容易出错,希望得到更优的解决方案。

2、解决方案

一位用户给出了两个解决方案:

  • 使用genshietree等库来生成XML,而不是手动构建。
  • 提供了一个使用lxml.etree库的示例,该示例将数据转换为XML并输出。

以下是使用lxml.etree库的解决方案的代码示例:

import datetime

projects = [{'code': 'demo',
 'created_at': datetime.datetime(2008, 6, 11, 7, 35, 19),
 'created_by': None,
 'id': 4,
 'image': 'https://website.com/files/0000/0000/blah.jpg',
 'name': 'Demo Project',
 'description': 'This is for demonstration purposes',
 'due': '2009-05-30',
 'start': '2009-05-06',
 'status': 'Active',
 'stype': 'Demo',
 'tag_list': [],
 'type': 'Project',
 'updated_at': datetime.datetime(2009, 5, 27, 1, 41, 55),
 'updated_by': {'id': 24, 'name': 'Test', 'type': 'HumanUser'},
 'users': [{'id': 19, 'name': 'User 1', 'type': 'HumanUser'},
           {'id': 18, 'name': 'User 2', 'type': 'HumanUser'},
           {'id': 17, 'name': 'User 3', 'type': 'HumanUser'},
           {'id': 16, 'name': 'User 4', 'type': 'HumanUser'},
           {'id': 15, 'name': 'User 5', 'type': 'HumanUser'},
           {'id': 14, 'name': 'User 6', 'type': 'HumanUser'},
           {'id': 13, 'name': 'User 7', 'type': 'HumanUser'},
           {'id': 12, 'name': 'User 8', 'type': 'HumanUser'},
           {'id': 20, 'name': 'Client 1', 'type': 'HumanUser'}]},
 ]

from lxml import etree

def E(tag, parent=None, content=None, children=None, **kw):
  e = etree.Element(tag)
  if not content is None:
    e.text = str(content)
  for k,v in kw.items():
    e.set(k, str(v))
  if not parent is None:
    parent.append(e)
  if not children is None:
    for c in children:
      e.append(c)
  return e

def processProject(data):
  attrs = ('name','type','id')
  p = E('Project')
  for item in attrs:
    p.set(item,str(data[item]))
  for k,v in [ x for x in data.items() if x[0] not in attrs ]:
    if v is None:
      E(k,parent=p)
    elif isinstance(v,basestring):
      E(k,content=v,parent=p)
    elif isinstance(v,(float,long,int)):
      E(k,content=str(v),parent=p)
    elif isinstance(v,datetime.datetime):
      E(k,content=v.strftime('%Y-%m-%d %H%M'),parent=p)
    elif k == 'users':
      users = E(k,parent=p)
      for u in v:
        E('user',parent=users,**dict([ (x,str(y)) for (x,y) in u.items()]))
    elif k in ('updated_by','created_by'):
      E(k,parent=p,**dict([ (x,str(y)) for (x,y) in v.items()]))
    elif k == 'tag_list':
      taglist = E(k,parent=p)
      for t in v:
        E('tag',parent=taglist,content=t)
  return p

>>> projxml = processProject(projects[0])
>>> etree.dump(projxml)

输出结果如下:

<Project name="Demo Project" type="Project" id="4">
  <status>Active</status>
  <code>demo</code>
  <created_at>2008-06-11 0735</created_at>
  <due>2009-05-30</due>
  <created_by/>
  <updated_at>2009-05-27 0141</updated_at>
  <start>2009-05-06</start>
  <image>https://website.com/files/0000/0000/blah.jpg</image>
  <updated_by type="HumanUser" id="24" name="Test"/>
  <users>
    <user type="HumanUser" id="19" name="User 1"/>
    <user type="HumanUser" id="18" name="User 2"/>
    <user type="HumanUser" id="17" name="User 3"/>
    <user type="HumanUser" id="16" name="User 4"/>
    <user type="HumanUser" id="15" name="User 5"/>
    <user type="HumanUser" id="14" name="User 6"/>
    <user type="HumanUser" id="13" name="User 7"/>
    <user type="HumanUser" id="12" name="User 8"/>
    <user type="HumanUser" id="20" name="Client 1"/>
  </users>
  <tag_list/>
  <stype>Demo</stype>
  <description>This is for demonstration purposes</description>
</Project>

这种方法使用lxml.etree库将数据转换为XML,并输出格式化的XML文档。与原先的解决方案相比,它更加简洁、高效,并且更易于扩展和维护。