python 实现 loadrunner xml脚本格式化

 

import re

from lxml import etree

def indent(elem, level=0):
    i = "\n" + level*"\t"
    if len(elem):
        if not elem.text or not elem.text.strip():
            elem.text = i + "\t"
        for e in elem:
            indent(e, level+1)
        if not e.tail or not e.tail.strip():
            e.tail = i
    if level and (not elem.tail or not elem.tail.strip()):
        elem.tail = i
    return elem


file_name = 'lr_xml.txt'

elems_root = etree.parse(file_name).getroot()
final_elem=indent(elems_root)
init_content=etree.tostring(final_elem, encoding="utf-8", method="xml")



# 每行左对齐
regex1 = re.compile("\t")
convert_first= re.sub(regex1,"",init_content)

# 每行开始加上"
regex2=re.compile("^<",flags=re.M)
convert_second = re.sub(regex2,"\"<",convert_first)

#每行末尾加上"
regex3 = re.compile(">$", flags=re.M)
final_content=re.sub(regex3,">\"",convert_second)

print final_content