From a7352f7c9dd9a864e3f8384146dd879481d83e42 Mon Sep 17 00:00:00 2001 From: PossiblyAxolotl Date: Tue, 13 Aug 2024 12:07:47 -0600 Subject: [PATCH] create output dir if not exist --- parser.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/parser.py b/parser.py index 21d6bc6..7a386a6 100644 --- a/parser.py +++ b/parser.py @@ -12,6 +12,10 @@ output_dir = 'output/' tree = ET.parse(source=dump_file) root = tree.getroot() +# Create output folder if needed +if not os.path.exists(output_dir): + os.mkdir(output_dir) + # Iterate through the pages and save them for page in root.findall('./{http://www.mediawiki.org/xml/export-0.11/}page'): title = page.find('./{http://www.mediawiki.org/xml/export-0.11/}title').text @@ -19,6 +23,7 @@ for page in root.findall('./{http://www.mediawiki.org/xml/export-0.11/}page'): print(title) + # Check for and process files differently if ":" in title: splitName = title.split(":")