第 0018 題： 将 第 0015 題中的 city.xls 檔案中的内容寫到 city.xml 檔案中，如下 所示：

<root>
<cities>
<!-- 
	城市資訊
-->
{
	"1" : "上海",
	"2" : "北京",
	"3" : "成都"
}
</cities>
</root>
           

import xlrd
import json
from lxml import etree


def read_exl(file_name):
    exl = xlrd.open_workbook(file_name)
    exl_sheet = exl.sheet_by_name('city')
    data = {}
    for i in range(exl_sheet.nrows):
        data[exl_sheet.row_values(i)[0]] = exl_sheet.row_values(i)[1]
    return json.dumps(data, encoding='utf-8')

def save_to_xml(data, new_file_name):
    root = etree.Element('root')
    students = etree.SubElement(root, 'citys')
    students.append(etree.Comment(u"""城市資訊"""))
    students.text = data

    student_xml = etree.ElementTree(root)
    student_xml.write(new_file_name, pretty_print=True, xml_declaration=True, encoding='utf-8')


if __name__ == '__main__':
    content = read_exl('city.xls')
    save_to_xml(content, 'city.xml')