import xmltodict
import json
import os


def load_json(xml_path,outurl):
    #获取xml文件
    xml_file = open(xml_path, 'r')
    #读取xml文件内容
    xml_str = xml_file.read()
    #将读取的xml内容转为json
    dic = xmltodict.parse(xml_str)
    print(type(dic))
    # jsonstr = json.dumps(dic)
    # outFile = outurl + ".json"
    with open(outurl, 'w') as f:
        # json.dump(file, f)
        json.dump(dic, f, indent=4, ensure_ascii=False)
    # print(type(jsonstr), jsonstr)
    # return jsonstr

# with open("../config/format_json.json", 'w') as write_f:
# 	json.dump(load_dict, write_f, indent=4, ensure_ascii=False)
# load_json(".xml")



url = "E:/WorkSpace/HUAWEI/MindSpore_yolov5/picdata/yolov5pic/yolov5/data/Annotations/"
outurl = "E:/WorkSpace/HUAWEI/MindSpore_yolov5/picdata/yolov5pic/yolov5/data/Annotations_json/"
for filepath in os.listdir(url):
    print(url+filepath)
    filename = os.path.splitext(filepath)[0]
    print(outurl+filename+ ".json")
    load_json(url+filepath,outurl+filename+".json")