import xml.etree.ElementTree as ET
from datetime import datetime

def generate_sitemap(urls):
    # 创建根节点
    urlset = ET.Element("urlset", xmlns="http://www.sitemaps.org/schemas/sitemap/0.9")
    
    for url in urls:
        # 创建URL节点
        url_elem = ET.SubElement(urlset, "url")
        
        # 添加子节点
        ET.SubElement(url_elem, "loc").text = url['loc']
        ET.SubElement(url_elem, "lastmod").text = url.get('lastmod', datetime.now().strftime('%Y-%m-%d'))
        ET.SubElement(url_elem, "changefreq").text = url.get('changefreq', 'weekly')
        ET.SubElement(url_elem, "priority").text = str(url.get('priority', 0.5))
    
    # 生成XML文件
    tree = ET.ElementTree(urlset)
    tree.write("sitemap.xml", encoding="utf-8", xml_declaration=True)

# 示例URL数据
urls = [
    {"loc": "https://kaishunhaotea.com/", "changefreq": "daily", "priority": 1.0},
    {"loc": "https://kaishunhaotea.com/about", "changefreq": "monthly", "priority": 0.8},
    {"loc": "https://kaishunhaotea.com/products", "changefreq": "weekly", "priority": 0.9},
]

# 生成sitemap
generate_sitemap(urls)