python 一个XML解析

5d6d网

'''　　
Version:0.5
　　
author;wujianqiang
　　
date:2016/11/13
　　
'''
　　
import xml.etree.ElementTree as ET
　　
from collections import Counter
　　
from pprint import pprint
　　
import json
　　
from fnmatch import fnmatch
　　
import os
　　
class XMLParse(object):
　　
    def __init__(self):
　　
        self.data = {}
　　
    def _open_file(self,filename):
　　
        return ET.parse(filename).getroot()
　　
    def _type_select(self,root, sybol='-'):
　　
        my_type = {}
　　
        flag =False
　　
        attrib_data = root.attrib
　　
        if len(attrib_data) > 0:
　　
            flag = True
　　
            data = {sybol + key: value for key, value in attrib_data.items()}
　　
            my_type.update(data)
　　
        if len(root):
　　
            myset = [child.tag for child in root]
　　
            mydata = Counter(myset)
　　
            for key, value in mydata.items():
　　
                if value > 1:
　　
                    my_type[key] = []
　　
                elif value == 1:
　　
                    my_type[key] = {}
　　
            return my_type
　　
        else:
　　
            if flag:
　　
                if root.text:
　　
                    my_type['#text'] = root.text
　　
                return my_type
　　
            else:
　　
                return root.text
　　
    def _par(self,root, data, sybol='-'):
　　
        utdata = self._type_select(root, sybol)
　　
        if isinstance(data[root.tag], dict):
　　
            data[root.tag] = utdata
　　
        elif isinstance(data[root.tag], list):
　　
            data[root.tag].append(utdata)
　　
        if len(root) > 0:
　　
            for item in root:
　　
                self._par(item, utdata, sybol)
　　
    def getdata_fromFile(self,filename, outfile=None,sybol='-'):
　　
        root = self._open_file(filename)
　　
        self.data[root.tag] = {}
　　
        if root.attrib:
　　
            data = {sybol + key: value for key, value in root.attrib.items()}
　　
            self.data[root.tag].update(data)
　　
        self._par(root, self.data,sybol)
　　
        if outfile:
　　
            file = open(outfile,'w')
　　
            json.dump(self.data,file)
　　
            file.close()
　　
            return True
　　
        else:
　　
            return self.data
　　
    def getdata_fromString(self, stringdata, outfile=None, sybol='-'):
　　
        root = ET.fromstring(stringdata)
　　
        self.data[root.tag] = {}
　　
        if root.attrib:
　　
            data = {sybol + key: value for key, value in root.attrib.items()}
　　
            self.data[root.tag].update(data)
　　
        self._par(root, self.data, sybol)
　　
        if outfile:
　　
            file = open(outfile, 'w')
　　
            json.dump(self.data, file)
　　
            file.close()
　　
            return True
　　
        else:
　　
            return self.data
　　
    def _create_sub_Element(self,parnent, data, tag=None, attrib='-', text='#text'):
　　
        if isinstance(data, dict):
　　
            for key, value in data.items():
　　
                if fnmatch(key, '%s*' % (attrib,)):
　　
                    parnent.attrib.update({key[1:]: value})
　　
                elif fnmatch(key, text):
　　
                    parnent.text = value
　　
                else:
　　
                    if type(value) == str:
　　
                        a = ET.SubElement(parnent, key)
　　
                        a.text = value
　　
                    elif isinstance(value, dict):
　　
                        a = ET.SubElement(parnent, key)
　　
                        self._create_sub_Element(a, value)
　　
                    else:
　　
                        self._create_sub_Element(parnent, value, tag=key)
　　
        elif isinstance(data, list):
　　
            for item in data:
　　
                l = ET.SubElement(parnent, tag)
　　
                self._create_sub_Element(l, item, tag)
　　
        else:
　　
            pass
　　
    def _create_root_Element(self,data):
　　
        root_key = data.keys()[0]
　　
        root = ET.Element(root_key)
　　
        etre = ET.ElementTree(root)
　　
        return [root, data[root_key], etre]
　　
    def _create_Element(self, data, attr='-', text='#text'):
　　
        root, mydata, tree = self._create_root_Element(data)
　　
        self._create_sub_Element(root, mydata, attrib=attr, text=text)
　　
        return tree
　　
    def create_xml(self,data,outfile=None):
　　
        if data:
　　
            root = self._create_Element(data)
　　
            filename = outfile if outfile else '.temp'
　　
            root.write(filename)
　　
            if not outfile:
　　
                f = open(filename)
　　
                data = f.readlines()[0]
　　
                f.close()
　　
os.remove(filename)
　　
                return data
　　
        else:
　　
            return None
　　
if __name__ == '__main__':
　　
    xdata = XMLParse()
　　
    dat = {'data': {'-fictional': 'http://characters.example.com',
　　
                    'country': [{'-name': 'Liechtenstein',
　　
                                 'gdppc': '141100',
　　
                                 'neighbor': [{'-direction': 'E',
　　
                                               '-name': 'Austria'},
　　
                                              {'-direction': 'W',
　　
                                               '-name': 'Switzerland'}],
　　
                                 'rank': '1',
　　
                                 'year': '2008'},
　　
                                {'-name': 'dfd',
　　
                                 'gdppc': '14100',
　　
                                 'neighbor': {'-direction': 'E', '-name': 'wuul'},
　　
                                 'rank': '3',
　　
                                 'year': {'#text': '208', '-d': 'sd'}}]}}
　　
    # pprint(xdata.getdata_fromFile('myxml2.xml',sybol='-'))
　　
    pprint(xdata.create_xml(dat))