004 python | 生成xml

以生成神马站长提交的sitemap.xml为例

import xml.etree.ElementTree as ET
import xml.dom.minidom as minidom
import time

cur_time = time.strftime("%Y-%m-%d", time.localtime())
urls = list()
keys = ["loc", "lastmod", "priority"]

split_size = 9000
for part in range(len(urls)/split_size + 1):
    urlset = ET.Element("urlset")

    for u in urls[split_size*part: split_size*part+split_size]:
        turl = ET.SubElement(urlset, "url")

        for i in range(3):
            key = ET.SubElement(turl, keys[i])
            key.text = u[i]

    content_str = ET.tostring(urlset)
    content = minidom.parseString(content_str)

    with open("/root/sitemap"+str(part)+".xml","w") as f:
        content.writexml(f, addindent=" ", newl="\n", encoding="utf-8")