Show me the code之Python练习册 Q17~19 xml操作

__file__ = 'A17.py'
__author__ = 'Jerry Liu'
__date__ = '2016-04-15'

"""
问题： 第 0014 题中的 student.xls 文件中的内容写到 student.xml 文件中，如下所示：

<?xml version="1.0" encoding="UTF-8"?>
<root>
<students>
<!--
学生信息表
"id" : [名字, 数学, 语文, 英文]
-->
{
"1" : ["张三", 150, 120, 100],
"2" : ["李四", 90, 99, 95],
"3" : ["王五", 60, 66, 68]
}
</students>
</root>
"""

from openpyxl.reader.excel import load_workbook
from xml.dom.minidom import Document

def getdata(path):
wb = load_workbook(path)
# 读取第一个sheet
snames = wb.get_sheet_names()
sheet = wb.get_sheet_by_name(snames[0])
# 创建xml
doc = Document()
# 创建根元素
root = doc.createElement('root')
doc.appendChild(root)
# 创建学生列表元素
stus = doc.createElement('students')
root.appendChild(stus)
for line in range(sheet.get_highest_row()):
line += 1
# 创建学生元素
stu = doc.createElement('student')
stus.appendChild(stu)
# 创建学生姓名元素
name = sheet.cell(row=line, column=1).value
stu_name = doc.createElement('Name')
stu_name.appendChild(doc.createTextNode(name))
stu.appendChild(stu_name)
# 创建英文成绩
yy_value = sheet.cell(row=line, column=2).value
yy_name = doc.createElement('English')
yy_name.appendChild(doc.createTextNode(str(yy_value)))
stu.appendChild(yy_name)
# 创建语文成绩
yw_value = sheet.cell(row=line, column=3).value
yw_name = doc.createElement('Chinese')
yw_name.appendChild(doc.createTextNode(str(yw_value)))
stu.appendChild(yw_name)
# 数学语文成绩
sx_value = sheet.cell(row=line, column=4).value
sx_name = doc.createElement('Math')
sx_name.appendChild(doc.createTextNode(str(sx_value)))
stu.appendChild(sx_name)

# 保存文件
f = open("d://student.xml", "w")
doc.writexml(f)
f.close()

if __name__ == '__main__':
getdata('d://student.xlsx')

__file__ = 'A18.py'
__author__ = 'Jerry Liu'
__date__ = '2016-04-15'

"""
问题：将 第 0015 题中的 city.xls 文件中的内容写到 city.xml 文件中，如下所示：
<?xmlversion="1.0" encoding="UTF-8"?>
<root>
<citys>
<!--
城市信息
-->
{
"1" : "上海",
"2" : "北京",
"3" : "成都"
}
</citys>
</root>
"""

from openpyxl.reader.excel import load_workbook
from xml.dom.minidom import Document

def getdata(path):
wb = load_workbook(path)
# 读取第一个sheet
snames = wb.get_sheet_names()
sheet = wb.get_sheet_by_name(snames[0])

cxml = Document()
eRoot =cxml.createElement('root')
cxml.appendChild(eRoot)
eCities = cxml.createElement('cities')
eRoot.appendChild(eCities)
eComments = cxml.createComment('城市信息')
eCities.appendChild(eComments)

# 定义字典
dict = {}
for line in range(sheet.get_highest_row()):
line += 1
id = sheet.cell(row=line, column=1).value
city = sheet.cell(row=line, column=2).value
dict[id] = city

eCities.appendChild(cxml.createTextNode(str(dict)))

f = open('d://city.xml', 'w')
cxml.writexml(f)
f.close()

if __name__ == '__main__':
getdata('d://city.xlsx')

__file__ = 'A19.py'
__author__ = 'Jerry Liu'
__date__ = '2016-04-18'

"""
问题：将 第 0016 题中的 numbers.xls 文件中的内容写到 numbers.xml 文件中，如下所示：
<?xml version="1.0" encoding="UTF-8"?>
<root>
<numbers>
<!--
数字信息
-->

[
[1, 82, 65535],
[20, 90, 13],
[26, 809, 1024]
]

</numbers>
</root>
"""
from openpyxl.reader.excel import load_workbook
from xml.dom.minidom import Document

def getdata(path):
wb = load_workbook(path)
# 读取第一个sheet
snames = wb.get_sheet_names()
sheet = wb.get_sheet_by_name(snames[0])

mdata = []
#读取数据
for line in range(sheet.get_highest_row()):
line += 1
list = []
list.append(sheet.cell(row=line, column=1).value)
list.append(sheet.cell(row=line, column=2).value)
list.append(sheet.cell(row=line, column=3).value)
mdata.append(list)

cxml = Document()
eRoot =cxml.createElement('root')
cxml.appendChild(eRoot)
eNums = cxml.createElement('numbers')
eRoot.appendChild(eNums)
eComments = cxml.createComment('数字信息')
eNums.appendChild(eComments)
eData = cxml.createTextNode(str(mdata))
eNums.appendChild(eData)

f = open('d://numbers.xml', 'w')
cxml.writexml(f)
f.close()

if __name__ == '__main__':
getdata('d://numbers.xlsx')