实现Python对象与xml之间的相互转换

来源:互联网 发布:ncbi数据库怎么下载 编辑:程序博客网 时间:2024/05/01 00:02
    # -*- coding:UTF-8 -*-      '''      Created on 2010-4-20            @author: 忧里修斯      '''      import xml.etree.ElementTree as ET      import xml.dom.minidom as minidom      from addrbook.domain import Person            class Converter(object):          '''               实现Python对象与xml之间的相互转换          '''                    root = None#根节点                    def __init__(self):              pass                    @staticmethod          def createRoot(rootTag):              '''                          创建根节点              '''              root = ET.Element(rootTag)              return root                    @staticmethod          def getXmlString(element,defaultEncoding='utf-8'):               '''                          根据节点返回格式化的xml字符串              '''              try:                                    rough_string = ET.tostring(element, defaultEncoding)                  reparsed = minidom.parseString(rough_string)                  return reparsed.toprettyxml(indent="  " , encoding=defaultEncoding)              except:                  print 'getXmlString:传入的节点不能正确转换为xml,请检查传入的节点是否正确'                  return ''                    @staticmethod          def classToElements(classobj,rootTag=None):              '''                          根据传入的对象的实例,根据对象的属性生成节点,返回由节点组成的列表              classobj:对象的实例              rootTag:根节点名称              '''              attrs = None#保存对象的属性集              elelist = []#节点列表              try:                  attrs = classobj.__dict__.keys()#获取该对象的所有属性(即成员变量)              except:                  print 'classToElements:传入的对象非法,不能正确获取对象的属性'                            if attrs != None and len(attrs) > 0:#属性存在                  for attr in attrs:                      attrvalue = getattr(classobj, attr)#属性值                      #属性节点                      attrE = ET.Element(attr)                      attrE.text = attrvalue                      #加入节点列表                      elelist.append(attrE)              return elelist                                      @staticmethod          def classToXML(classobj,rootTag=None):              '''              Python自定义模型类转换成xml,转换成功返回的是xml根节点,否则返回None              classobj:对象的实例              rootTag:根节点名称              '''              try:                  classname = classobj.__class__.__name__ #类名                  if rootTag != None:                      root = Converter.createRoot(rootTag)                  else:                      root = Converter.createRoot(classname)                  elelist = Converter.classToElements(classobj, rootTag)                  for ele in elelist:                      root.append(ele)                  return root              except:                  print 'classToXML:转换出错,请检查的传入的对象是否正确'                  return None                            @staticmethod          def collectionToXML(listobj,rootTag='list'):              '''                                  集合(列表、元组、字典)转换为xml,转换成功返回的是xml根节点,否则返回None              '''              try:                  classname = listobj.__class__.__name__ #类名                  root = Converter.createRoot(rootTag)                  if isinstance(listobj, list) or isinstance(listobj, tuple):#列表或元组                                            if len(listobj) >= 0:                          for obj in listobj:#迭代列表中的对象                              itemE = Converter.classToXML(obj)                              root.append(itemE)                  elif isinstance(listobj, dict):#字典                      if len(listobj) >= 0:                          for key in listobj:#迭代字典中的对象                              obj = listobj[key]                              itemE = Converter.classToXML(obj)                              itemE.set('key', key)                              root.append(itemE)                  else:                      print 'listToXML:转换错误,传入的对象:'+classname+'不是集合类型'                  return root              except:                  print 'collectionToXML:转换错误,集合转换成xml失败'                  return None                          if __name__ == '__main__':          p1 = Person('dredfsam','男','133665')          p2 = Person('dream','女','r')          p3 = Person('得分','男','sdf')          personList = {}          personList['p1']= p1          personList['p2']= p2          personList['p3']= p3                #    personList.append(p1)      #    personList.append(p2)      #    personList.append(p3)          root = Converter.collectionToXML(personList)          print Converter.getXmlString(root)      #    plist = (p1,p2,p3)#{'name':'sdf'}      #    print type(plist)      #    print type(plist),isinstance(plist, list)