from pathlib import Path
from xml.etree import ElementTree as ET

urls = [
    "https://drsafeskin.com/",
    "https://drsafeskin.com/services/",
    "https://drsafeskin.com/about-us/",
    "https://drsafeskin.com/appointments/",
    "https://drsafeskin.com/contact-us/",
]

xml_content = """<?xml version="1.0" encoding="UTF-8"?>
<urlset xmlns="http://www.sitemaps.org/schemas/sitemap/0.9">
  <url>
    <loc>https://drsafeskin.com/</loc>
  </url>
  <url>
    <loc>https://drsafeskin.com/services/</loc>
  </url>
  <url>
    <loc>https://drsafeskin.com/about-us/</loc>
  </url>
  <url>
    <loc>https://drsafeskin.com/appointments/</loc>
  </url>
  <url>
    <loc>https://drsafeskin.com/contact-us/</loc>
  </url>
</urlset>
"""

output = Path("/mnt/data/sitemap.xml")
output.write_text(xml_content, encoding="utf-8")

# Validate XML and URL count
tree = ET.parse(output)
root = tree.getroot()
namespace = {"sm": "http://www.sitemaps.org/schemas/sitemap/0.9"}
found_urls = [node.text for node in root.findall("sm:url/sm:loc", namespace)]

assert found_urls == urls
assert output.exists()
print(f"Created {output.name} with {len(found_urls)} URLs ({output.stat().st_size} bytes).")
