Python: xml转json-fullerhua的博客

693579551

#! /usr/bin/env python　　
# -*- coding:utf-8 -*-
　　
# xml 转 json
　　
# xml2json.py
　　
# Version 1.0
　　

　　
from xml.parsers.expat import ParserCreate
　　
import json
　　

　　
class Xml2Json:
　　
    LIST_TAGS = ['COMMANDS']
　　

　　
    def __init__(self, data = None):
　　
        self._parser = ParserCreate()
　　
        self._parser.StartElementHandler = self.start
　　
        self._parser.EndElementHandler = self.end
　　
        self._parser.CharacterDataHandler = self.data
　　
        self.result = None
　　
        if data:
　　
            self.feed(data)
　　
            self.close()
　　

　　
    def feed(self, data):
　　
        self._stack = []
　　
        self._data = ''
　　
        self._parser.Parse(data, 0)
　　

　　
    def close(self):
　　
        self._parser.Parse("", 1)
　　
        del self._parser
　　

　　
    def start(self, tag, attrs):
　　
        assert attrs == {}
　　
        assert self._data.strip() == ''
　　
        self._stack.append([tag])
　　
        self._data = ''
　　

　　
    def end(self, tag):
　　
        last_tag = self._stack.pop()
　　
        assert last_tag[0] == tag
　　
        if len(last_tag) == 1: #leaf
　　
            data = self._data
　　
        else:
　　
            if tag not in Xml2Json.LIST_TAGS:
　　
                # build a dict, repeating pairs get pushed into lists
　　
                data = {}
　　
                for k, v in last_tag[1:]:
　　
                    if k not in data:
　　
                        data[k] = v
　　
                    else:
　　
                        el = data[k]
　　
                        if type(el) is not list:
　　
                            data[k] = [el, v]
　　
                        else:
　　
                            el.append(v)
　　
            else: #force into a list
　　
                data = [{k:v} for k, v in last_tag[1:]]
　　
        if self._stack:
　　
            self._stack[-1].append((tag, data))
　　
        else:
　　
            self.result = {tag:data}
　　
        self._data = ''
　　

　　
    def data(self, data):
　　
        self._data = data
　　

　　
if __name__ == '__main__':
　　
    xml = open("京东手机列表_1.xml", 'r', encoding='UTF-8').read()
　　
    result = Xml2Json(xml).result;
　　
    outputfile = open("京东手机列表_1.json", 'w', encoding='UTF-8')
　　
    outputfile.write(str(result))
　　
    outputfile.close()