XML文件处理之对相同.xml的后缀分类

import xml.dom.minidom
import os
import sys
import xml.etree.ElementTree as ET
from lxml.etree import Element, SubElement, tostring
from xml.dom.minidom import parseString


src_path = r'C:\Users\jiabin.ouyang\Desktop\plunger xml'
save_path1 = r"C:\Users\jiabin.ouyang\Desktop\xml_1_sort"
save_path2 = r"C:\Users\jiabin.ouyang\Desktop\xml_2_sort"
save_path3 = r"C:\Users\jiabin.ouyang\Desktop\xml_3_sort"
save_path4 = r"C:\Users\jiabin.ouyang\Desktop\xml_4_sort"
save_path5 = r"C:\Users\jiabin.ouyang\Desktop\xml_5_sort"
save_path6 = r"C:\Users\jiabin.ouyang\Desktop\xml_6_sort"
files = os.listdir(src_path)

for xmlFile in files:
    dom = xml.dom.minidom.parse(os.path.join(src_path, xmlFile))  # 打开xml文件，送到dom解析
    root = dom.documentElement  # 得到文档元素对象

    names = root.getElementsByTagName('filename')   #修改filename块
    a, b= os.path.splitext(xmlFile)  # 分离出文件名a(a:1_1,b:.xml)
    a11 = a.split('_')
   # b11 = a11[0] + '_' + a11[1]
    print("show:::",a11[1])
    #if a11[1]==1
    if a11[1] == '1':
        print("come on!")
        with open(os.path.join(save_path1, xmlFile), 'w') as fh:
            dom.writexml(fh)
            print("come on!")
    elif a11[1] == '2':
        with open(os.path.join(save_path2, xmlFile), 'w') as fh:
            dom.writexml(fh)
    elif a11[1] == '3':
        with open(os.path.join(save_path3, xmlFile), 'w') as fh:
            dom.writexml(fh)
    elif a11[1] == '4':
        with open(os.path.join(save_path4, xmlFile), 'w') as fh:
            dom.writexml(fh)
    elif a11[1] == '5':
        with open(os.path.join(save_path5, xmlFile), 'w') as fh:
            dom.writexml(fh)
    elif a11[1] == '6':
        with open(os.path.join(save_path6, xmlFile), 'w') as fh:
            dom.writexml(fh)
    else:
        print('more!')

    #with open(os.path.join(sv_path, xmlFile), 'w') as fh:
        #dom.writexml(fh)