excel转为xml导入testlink的python实现

hwl198212

class operate():　　def __init__(self):
　　self.temp=easy_excel('test.xlsx')
　　self.dic_testlink={}
　　self.row_flag=2
　　self.testsuite = self.temp.getCell('Sheet1',1,1)
　　self.dic_testlink[self.testsuite]={"node_order":"13","details":"","testcase":[]}
　　self.content = ""
　　self.content_list = []
　　def xlsx_to_dic(self):
　　while True:
　　#print 'loop1'
　　#list_testcase = dic_testlink[testsuite].["testcase"]
　　#1.用例名2.关键字3.摘要4.前提5.步骤6.预期结果
　　testcase = {"name":"","node_order":"100","externalid":"","version":"1","summary":"","preconditions":"","execution_type":"1","importance":"3","steps":[],"keywords":"P1"}
　　testcase["name"] = self.temp.getCell('Sheet1',self.row_flag,1)
　　testcase["summary"] = self.temp.getCell('Sheet1',self.row_flag,3)
　　testcase["preconditions"] = self.temp.getCell('Sheet1',self.row_flag,4)
　　#print self.temp.getCell('Sheet1',self.row_flag,3)
　　step_number=1
　　testcase["keywords"] = self.temp.getCell('Sheet1',self.row_flag,2)
　　print testcase["keywords"]
　　while True:
　　#print 'loop2'
　　step = {"step_number":"","actions":"","expectedresults":"","execution_type":""}
　　step["step_number"] = step_number
　　step["actions"] = self.temp.getCell('Sheet1',self.row_flag,5)
　　step["expectedresults"] = self.temp.getCell('Sheet1',self.row_flag,6)
　　testcase["steps"].append(step)
　　step_number += 1
　　self.row_flag += 1
　　if self.temp.getCell('Sheet1',self.row_flag,1) is not None  or self.temp.getCell('Sheet1',self.row_flag,5) is None:
　　break
　　#print testcase
　　self.dic_testlink[self.testsuite]["testcase"].append(testcase)
　　#print self.row_flag
　　if self.temp.getCell('Sheet1',self.row_flag,5) is None and self.temp.getCell('Sheet1',self.row_flag+1,5) is None:
　　break
　　self.temp.close()
　　#print self.dic_testlink
　　def content_to_xml(self,key,value=None):
　　if key == 'step_number' or key == 'execution_type' or key == 'node_order' or key == 'externalid' or key == 'version' or key == 'importance':
　　return "<"+str(key)+"><![CDATA["+str(value)+"]]></"+str(key)+">"
　　elif key == 'actions' or key == 'expectedresults' or key == 'summary' or key == 'preconditions':
　　return "<"+str(key)+"><![CDATA[<p> "+str(value)+"</p> ]]></"+str(key)+">"
　　elif key == 'keywords':
　　return '<keywords><keyword name="'+value+'"><notes><![CDATA[ 每个版本都会执行的用例 ]]></notes></keyword></keywords>'
　　elif key == 'name':
　　return '<testcase name="'+str(value)+'">'
　　else:
　　return '##########'
　　def dic_to_xml(self):
　　testcase_list = self.dic_testlink[self.testsuite]["testcase"]
　　for testcase in testcase_list:
　　for step in testcase["steps"]:
　　self.content += "<step>"
　　self.content += self.content_to_xml("step_number",step["step_number"])
　　self.content += self.content_to_xml("actions",step["actions"])
　　self.content += self.content_to_xml("expectedresults",step["expectedresults"])
　　self.content += self.content_to_xml("execution_type",step["execution_type"])
　　self.content += "</step>"
　　self.content = "<steps>" + self.content + "</steps>"
　　self.content = self.content_to_xml("importance",testcase["importance"]) + self.content
　　self.content = self.content_to_xml("execution_type",testcase["execution_type"]) + self.content
　　self.content = self.content_to_xml("preconditions",testcase["preconditions"]) + self.content
　　self.content = self.content_to_xml("summary",testcase["summary"]) + self.content
　　self.content = self.content_to_xml("version",testcase["version"]) + self.content
　　self.content = self.content_to_xml("externalid",testcase["externalid"]) + self.content
　　self.content = self.content_to_xml("node_order",testcase["node_order"]) + self.content
　　self.content = self.content + self.content_to_xml("keywords",testcase["keywords"])
　　self.content = self.content_to_xml("name",testcase["name"]) + self.content
　　self.content = self.content + "</testcase>"
　　self.content_list.append(self.content)
　　self.content = ""
　　self.content = "".join(self.content_list)
　　self.content = '<testsuite name="'+self.testsuite+'">'+self.content+"</testsuite>"
　　self.content = '<?xml version="1.0" encoding="UTF-8"?>' + self.content
　　self.write_to_file()
　　def write_to_file(self):
　　cp = open("test.xml","w")
　　cp.write(self.content)
　　cp.close()
　　
test=operate()
　　
test.xlsx_to_dic()
　　
test.dic_to_xml()