# -*- 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):
  '''''
     實(shí)現(xiàn)Python對(duì)象與xml之間的相互轉(zhuǎn)換
  '''
  root = None#根節(jié)點(diǎn)
  def __init__(self):
    pass
  @staticmethod
  def createRoot(rootTag):
    '''''
          創(chuàng)建根節(jié)點(diǎn)
    '''
    root = ET.Element(rootTag)
    return root
  @staticmethod
  def getXmlString(element,defaultEncoding='utf-8'):
    '''''
          根據(jù)節(jié)點(diǎn)返回格式化的xml字符串
    '''
    try:
      rough_string = ET.tostring(element, defaultEncoding)
      reparsed = minidom.parseString(rough_string)
      return reparsed.toprettyxml(indent=" " , encoding=defaultEncoding)
    except:
      print 'getXmlString:傳入的節(jié)點(diǎn)不能正確轉(zhuǎn)換為xml，請(qǐng)檢查傳入的節(jié)點(diǎn)是否正確'
      return ''
  @staticmethod
  def classToElements(classobj,rootTag=None):
    '''''
          根據(jù)傳入的對(duì)象的實(shí)例，根據(jù)對(duì)象的屬性生成節(jié)點(diǎn)，返回由節(jié)點(diǎn)組成的列表
    classobj：對(duì)象的實(shí)例
    rootTag：根節(jié)點(diǎn)名稱
    '''
    attrs = None#保存對(duì)象的屬性集
    elelist = []#節(jié)點(diǎn)列表
    try:
      attrs = classobj.__dict__.keys()#獲取該對(duì)象的所有屬性(即成員變量)
    except:
      print 'classToElements:傳入的對(duì)象非法，不能正確獲取對(duì)象的屬性'
    if attrs != None and len(attrs) > 0:#屬性存在
      for attr in attrs:
        attrvalue = getattr(classobj, attr)#屬性值
        #屬性節(jié)點(diǎn)
        attrE = ET.Element(attr)
        attrE.text = attrvalue
        #加入節(jié)點(diǎn)列表
        elelist.append(attrE)
    return elelist
  @staticmethod
  def classToXML(classobj,rootTag=None):
    '''''
    Python自定義模型類轉(zhuǎn)換成xml，轉(zhuǎn)換成功返回的是xml根節(jié)點(diǎn)，否則返回None
    classobj：對(duì)象的實(shí)例
    rootTag：根節(jié)點(diǎn)名稱
    '''
    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:轉(zhuǎn)換出錯(cuò)，請(qǐng)檢查的傳入的對(duì)象是否正確'
      return None
  @staticmethod
  def collectionToXML(listobj,rootTag='list'):
    '''''
              集合（列表、元組、字典）轉(zhuǎn)換為xml，轉(zhuǎn)換成功返回的是xml根節(jié)點(diǎn)，否則返回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:#迭代列表中的對(duì)象
            itemE = Converter.classToXML(obj)
            root.append(itemE)
      elif isinstance(listobj, dict):#字典
        if len(listobj) >= 0:
          for key in listobj:#迭代字典中的對(duì)象
            obj = listobj[key]
            itemE = Converter.classToXML(obj)
            itemE.set('key', key)
            root.append(itemE)
      else:
        print 'listToXML：轉(zhuǎn)換錯(cuò)誤，傳入的對(duì)象：'+classname+'不是集合類型'
      return root
    except:
      print 'collectionToXML：轉(zhuǎn)換錯(cuò)誤，集合轉(zhuǎn)換成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)