add create dir if not exists

pysitemap/format_processors/text.py

@@ -1,18 +1,21 @@
 import asyncio
 from aiofile import AIOFile, Reader, Writer
 import logging
+import os
 
 
-class TextWriter():
-    def __init__(self, filename: str):
-        self.filename = filename
-
+class TextWriter:
+    def __init__(self, filepath: str):
+        self.filepath = filepath
+        self.dir, self.filename = os.path.split(self.filepath)
 
     async def write(self, urls):
-        async with AIOFile(self.filename, 'w') as aiodf:
+        if self.dir and not os.path.exists(self.dir):
+            os.makedirs(self.dir)
+
+        async with AIOFile(self.filepath, "w") as aiodf:
             writer = Writer(aiodf)
 
             for url in urls:
                 await writer("{}\n".format(url))
             await aiodf.fsync()
-

pysitemap/format_processors/xml.py

@@ -1,26 +1,30 @@
 import asyncio
 from aiofile import AIOFile, Reader, Writer
 import logging
+import os
 
 
-class XMLWriter():
-    def __init__(self, filename: str):
-        self.filename = filename
-
+class XMLWriter:
+    def __init__(self, filepath: str):
+        self.filepath = filepath
+        self.dir, self.filename = os.path.split(self.filepath)
 
     async def write(self, urls):
-        async with AIOFile(self.filename, 'w') as aiodf:
+        if self.dir and not os.path.exists(self.dir):
+            os.makedirs(self.dir)
+
+        async with AIOFile(self.filepath, "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')
+                ' 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 writer("<url><loc>{}</loc></url>\n".format(url))
             await aiodf.fsync()
 
-            await writer('</urlset>')
+            await writer("</urlset>")
             await aiodf.fsync()
-