sitemap-generator/pysitemap/format_processors/xml.py

import asyncio
from aiofile import AIOFile, Reader, Writer
import logging


class XMLWriter():
    def __init__(self, filename: str):
        self.filename = filename


    async def write(self, urls):
        async with AIOFile(self.filename, 'w') as aiodf:
            writer = Writer(aiodf)
            await writer('<?xml version="1.0" encoding="utf-8"?>\n')
            await writer(
                '<urlset xmlns="http://www.sitemaps.org/schemas/sitemap/0.9"'
                ' xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"'
                ' xsi:schemaLocation="http://www.sitemaps.org/schemas/sitemap/0.9 http://www.sitemaps.org/schemas/sitemap/0.9/sitemap.xsd">\n')
            await aiodf.fsync()
            for url in urls:
                await writer('<url><loc>{}</loc></url>\n'.format(url))
            await aiodf.fsync()

            await writer('</urlset>')
            await aiodf.fsync()