! encoding=utf-8
import xlrd
import xml.dom.minidom
import os
def open_excel(file):
try:
data = xlrd.open_workbook(file)
return data
except Exception as e:
print (str(e))
def translate_excel_to_xml(excel_absolute_path, name, generate_xml_dir, colnnameindex=0, by_index=1):
#解析excel文件
data = open_excel(excel_absolute_path)
#获取需要的工作表
table = data.sheets()[by_index]
#行数
nrows = table.nrows
#列数
ncols = table.ncols
#创建dom文档对象
doc = xml.dom.minidom.Document()
#创建根元素
info = doc.createElement('info')
#将根元素添加到文档中区
doc.appendChild(info)
for nrow in range(4, nrows):
#创建元素
item = doc.createElement('item')
for ncol in range(0, ncols):
#colnames = table.col_values(ncol)
#print colnames
#print table.cell(nrow, ncol).value
key = u"%s" % table.cell(0, ncol).value
value = table.cell(nrow, ncol).value
if isinstance(value, float):
value = '%0d' % value
#将数据都作为xml中元素的属性,属性名就是第一行的值,属性值就是某一行某一列的值
item.setAttribute(key, value)
#将此元素作为根元素的子节点
info.appendChild(item)
#要生成的xml文件名
generate_xml_name = name.strip().split('.')[0] + '2.xml'
#要生成的xml文件到某个目录的绝对路径
geneate_xml_dir = os.path.join(generate_xml_dir, generate_xml_name)
f = open(geneate_xml_dir, 'w')
f.write(doc.toprettyxml())
f.close()
def find_assign_xlsx(xlsx_path, generate_xml_dir):
for name in os.listdir(xlsx_path):
if name.endswith(‘.xlsx’):
#生成excel文件的绝对路径
excel_absolute_path = os.path.join(xlsx_path, name)
#解析excel并转成xml
translate_excel_to_xml(excel_absolute_path, name, generate_xml_dir)
if name == “main“:
excel_src_path = ‘/’
generate_xml_dir = ‘/’
find_assign_xlsx(excel_src_path, generate_xml_dir)