python改XML

from lxml import etree
import os


# 查询文件夹中所有的xml文件，存到xml_files中
def findXmlFile(filePath):
    xml_files =[]
    for root, folder, files in os.walk(filePath):
        for file in files:
            if file.endswith('.xml'):
                xml_files.append(os.path.join(root, file))
    return xml_files



#模糊匹配目录xml文件名，符合条件的文件存到target_file列表中
def fuzzyfinder(user_input, xml_files):
    target_file =[]
    for xml_file in xml_files:  # 遍历文件夹中的文件，i：'d:\xml\LiveCheck_Provider_HttpRS1.xml'
        if xml_file.find(user_input) > 0:
            target_file.append(xml_file)
    return target_file


#查找要替换的目标元素
def findByXpath(target_file,target_filed):
    #循环查找遍历xml文件
    ret = []
    elementName = []
    for path in target_file:
        xml = etree.parse(path, etree.HTMLParser()) #利用lxml解析库的 etree.parse方法
        target_filed_lower = target_filed.lower()  # 转换为小写
        command = '//@' + target_filed_lower  # 目标字段转成：//@rateplancode
        result = xml.xpath(command)  # 用xpath找到目标，结果是list,如有RatePlanCode="DSRPLO01",RatePlanCode="DSRPLO02"，用//@rateplancode找到的值是：['DSRPLO01', 'DSRPLO02']
        fileName = "文件名:" + os.path.basename(path)  # 获取文件名,如：文件名:LiveCheck_P1rovider_HttpRS.xml
        print fileName
        for i in range(len(result)):  # 遍历xpath查到的结果
            str = target_filed + '="' + result[i] + '"'  # 把字段和结果拼接起来,RatePlanCode="DSRPLO01",RatePlanCode="DSRPLO02"
            elementName.append(str)  # 拼接结果放到列表
        print(elementName)
        ret.append(elementName)
        elementName = []

def my_Replace(content, target_filed, new_filed, index=0):
    # 替换成修改的数据
    print target_filed, new_filed, index
    str = content.replace(target_filed, new_filed, int(index))
    # 把前面被替换的数据修改回来
    a = int(index) - 1
    ret = str.replace(new_filed, target_filed, a)
    return ret

def writeXML(target_file,target_filed,new_filed,index):
    # 替换内容并写入文件
    for path in target_file:
        with open(path, "r") as file:
            fullcontent = file.read()#获取文件所有内容
        if len(index) == 0:
            content = fullcontent.replace(target_filed,new_filed)
        else:
            content = my_Replace(fullcontent,target_filed,new_filed,index)
        f = open(path, 'w')
        f.write(content)
        f.close()


if __name__ == '__main__':
    xml_files =[]
    elementNames=[]
    #遍历文件夹中xml的数量
    xml_files = findXmlFile("D:/xml")
    if len(xml_files):
        print("文件夹中所有XML文件:")
        print(xml_files)
        user_input = raw_input("请输入需要修改的文件名(大小写敏感):")
        target_file = fuzzyfinder(user_input,xml_files)
        if len(target_file):
            print("匹配到的XML文件:")
            print target_file
            target_filed = raw_input("输入需要查询的元素: ")
            while(1):
                findByXpath(target_file, target_filed)
                old = raw_input("输入需要修改原内容:")
                new = raw_input("输入需要修改后的内容:")
                i = raw_input("输入需要修改的下标:")
                if old == '':
                    break
                else:
                    writeXML(target_file,old,new,i)
        else:
            print ("没有匹配到XML文件")
    else:
        print("文件夹中不存在XML文件:")