✨ allow posting single wiki articles

4_post_to_wiki.py

@@ -1,5 +1,6 @@
 import os
 import mwclient
+import click
 import sys
 from tqdm import tqdm
 
@@ -15,10 +16,20 @@ CATEGORY_TEMPLATE = "\n".join(
 site = mwclient.Site(WIKI_HOST, path=WIKI_PATH)
 site.login(WIKI_USERNAME, WIKI_PASSWORD)
 
-# Process and upload each .mediawiki file
-for filename in tqdm(os.listdir(OUTDIR_MEDIAWIKI)):
-    if filename.endswith(".mediawiki"):
-        filepath = os.path.join(OUTDIR_MEDIAWIKI, filename)
+
+@click.command()
+@click.argument('files', nargs=-1)
+def post(files):
+    if files[0] == "all":
+        files = []
+        for filename in tqdm(os.listdir(OUTDIR_MEDIAWIKI)):
+            if filename.endswith(".mediawiki"):
+                filepath = os.path.join(OUTDIR_MEDIAWIKI, filename)
+                files.append(filepath)
+
+    for filepath in files:
+        filename = os.path.basename(filepath)
+
         article_name = WIKI_ARTICLE_PREFIX + filename.replace(".mediawiki", "")
 
         with open(filepath, "r", encoding="utf-8") as f:
@@ -29,4 +40,8 @@ for filename in tqdm(os.listdir(OUTDIR_MEDIAWIKI)):
 
         # Post to MediaWiki
         page = site.pages[article_name]
-        page.save(final_content, summary="Automated upload of recipe page")
+        page.save(final_content, summary="U+1F916 ROBOT FACE")
+
+
+if __name__ == "__main__":
+    post()