Python对文本文件的简单操作（一）

工作背景

　　性能测试工程师，主要测试工具--loadrunner，主要是接口测试。

实现功能

　　loadrunner对报文格式的转换存在问题，部分报文无法转换，故使用Python编写脚本自动将soap协议报文转换为loadrunner默认的格式。

转换步骤

soap协议原报文如下：

 <soapenv:Envelope xmlns:soapenv="http://schemas.xmlsoap.org/soap/envelope/" xmlns:ser="URL">
    <soapenv:Header/>
    <soapenv:Body>
       <ser:queryRiskPolicyCaseInfo>
          <!--Zero or more repetitions:-->
          <arg0>
             <!--Optional:-->
             <clientInfo>
                <!--Optional:-->
                <batchNo>?</batchNo>
                <!--Optional:-->
                <size>?</size>
             </clientInfo>
             <!--Optional:-->
             <riskPolicyCaseSubDto>
                <!--Optional:-->
                <riskPolicyCaseRequestInfo>
                   <!--Optional:-->
                   <companyCode>?</companyCode>
                   <!--Optional:-->
                   <currentPage>?</currentPage>
                   <!--Optional:-->
                   <districtCode>?</districtCode>
                   <!--Optional:-->
                   <districtLevel>?</districtLevel>
                   <!--Optional:-->
                   <endDate>?</endDate>
                   <!--Optional:-->
                   <pageSize>?</pageSize>
                   <!--Optional一页多少条:-->
                   <riskCode>?</riskCode>
                   <!--Optional风险等级:-->
                   <riskRuleCode>?</riskRuleCode>
                   <!--Optional:-->
                   <ruleType>?</ruleType>
                   <!--Optional:-->
                   <startDate>?</startDate>
                </riskPolicyCaseRequestInfo>
             </riskPolicyCaseSubDto>
          </arg0>
       </ser:queryRiskPolicyCaseInfo>
    </soapenv:Body>
 </soapenv:Envelope>

代码如下：

 import re
 """
 将soap格式的报文修改为loadrunner格式
 """
 file = r"C:\\Users\\zg\\Desktop\\报文转换\\报文.txt"
 file_out = r"C:\\Users\\zg\\Desktop\\报文转换\\转换后的报文.txt"
 #报文头部
 header = 'soap_request("StepName=google", \n"ExpectedResponse=AnySoap", \n'+'"URL=www.baidu.com'+'"'\
          +', \n"SOAPEnvelope= "\n"<?xml version=\\"1.0\\" encoding=\\"utf-8\\"?>"'
 #报文尾部
 last = '"Snapshot=t1.inf",\n"ResponseParam=result",\nLAST );'
 with open(file_out,'w') as f_clear:
     #写文件前清空文件
     f_clear.truncate()
     f_clear.write(header+'\n')
 #确定line_num
 line_num = 1
 #确定文件的行数
 numbers = len(open(file,'r').readlines())
 #编辑原报文
 with open(file) as f_in:
     for line in f_in.readlines():
         #判断是否为空行
         if len(line.strip()) > 0:
             # 找出每行第一个非空字符的位置(正则表达式)
             num = re.search(r'\S', line).span()[0]
            #判断是否为最后一行
             if line_num < numbers:
                 # 对每行进行拼接
                 line = line[:num] + '"' + line[num:].replace('"', r'\"').rstrip() + '"'
                 line_num += 1
             else:
                 line = line[:num] + '"' + line[num:].replace('"', r'\"').rstrip() + '"' + ','
        #在最后一行行尾添加逗号‘,’
         else:
             line = '\r\t\n'
             line_num += 1
         #将修改后的内容写入新的文件
         with open(file_out,'a') as f_out:
             f_out.write(line + "\n")
     #追加报文尾部
     with open(file_out,'a') as f_out:
         f_out.write(last)            

运行代码，转换后的报文如下：

 soap_request("StepName=google",
 "ExpectedResponse=AnySoap",
 "URL=www.baidu.com",
 "SOAPEnvelope= "
 "<?xml version=\"1.0\" encoding=\"utf-8\"?>"
 "<soapenv:Envelope xmlns:soapenv=\"http://schemas.xmlsoap.org/soap/envelope/\" xmlns:ser=\"URL\">"
    "<soapenv:Header/>"
    "<soapenv:Body>"
       "<ser:queryRiskPolicyCaseInfo>"
          "<!--Zero or more repetitions:-->"
          "<arg0>"
             "<!--Optional:-->"
             "<clientInfo>"
                "<!--Optional:-->"
                "<batchNo>?</batchNo>"
                "<!--Optional:-->"
                "<size>?</size>"
             "</clientInfo>"
             "<!--Optional:-->"
             "<riskPolicyCaseSubDto>"
                "<!--Optional:-->"
                "<riskPolicyCaseRequestInfo>"
                   "<!--Optional:-->"
                   "<companyCode>?</companyCode>"
                   "<!--Optional:-->"
                   "<currentPage>?</currentPage>"
                   "<!--Optional:-->"
                   "<districtCode>?</districtCode>"
                   "<!--Optional:-->"
                   "<districtLevel>?</districtLevel>"
                   "<!--Optional:-->"
                   "<endDate>?</endDate>"
                   "<!--Optional:-->"
                   "<pageSize>?</pageSize>"
                   "<!--Optional一页多少条:-->"
                   "<riskCode>?</riskCode>"
                   "<!--Optional风险等级:-->"
                   "<riskRuleCode>?</riskRuleCode>"
                   "<!--Optional:-->"
                   "<ruleType>?</ruleType>"
                   "<!--Optional:-->"
                   "<startDate>?</startDate>"
                "</riskPolicyCaseRequestInfo>"
             "</riskPolicyCaseSubDto>"
          "</arg0>"
       "</ser:queryRiskPolicyCaseInfo>"
    "</soapenv:Body>"
 "</soapenv:Envelope>",
 "Snapshot=t1.inf",
 "ResponseParam=result",
 LAST );

　　后期会使用Tkinter模块实现其图形界面化，代码中用到正则表达式模块re，以后会有专门的文章对其进行描述，本人为Python入门菜鸟，大神请绕过，不喜勿喷！