Add command line options for base URL and feed title.

1 files changed, 25 insertions, 7 deletions

diff --git a/gemfeed.py b/gemfeed.py
index 5943dd3..9db3b32 100644
--- a/gemfeed.py
+++ b/gemfeed.py
@@ -1,3 +1,4 @@
+import argparse
 import datetime
 import glob
 import os.path
@@ -31,28 +32,45 @@ def populate_entry_from_file(filename, base_url, entry):
             title = filename
     entry.title(title)
 
-def main(base_url="gemini://gemini.circumlunar.space/"):
+def main():
+
+    # Parse arguments
+    parser = argparse.ArgumentParser(description='Generate an Atom feed for Gemini content.')
+    parser.add_argument('-b', '--base', dest='base_url', type=str,
+            required=True, help='base URL for feed and entries')
+    parser.add_argument('-o', '--output', dest='output', type=str,
+            default="atom.xml", help='output filename')
+    parser.add_argument('-t', '--title', dest='title', type=str,
+            default="A Gemini feed", help='feed title')
+    args = parser.parse_args()
+
+    # Normalise base URL, quick and dirty
+    if not args.base_url.startswith("gemini://"):
+        args.base_url = "gemini://" + args.base_url
+    if not args.base_url.endswith("/"):
+        args.base_url = args.base_url + "/"
 
     # Setup feed
     feed = FeedGenerator()
-    feed.id(base_url)
-    feed.title('A Gemini feed.')
-    feed.link(href=base_url, rel='alternate')
-    feed.link(href=base_url + "atom.xml", rel='self')
+    feed.id(args.base_url)
+    feed.title(args.title)
+    feed.link(href=args.base_url, rel='alternate')
+    feed.link(href=args.base_url + "atom.xml", rel='self')
 
     # Add one entry per .gmi file
     files = find_files()
     latest_update = datetime.datetime.fromtimestamp(0, tz=datetime.timezone.utc)
     for filename in files:
         entry = feed.add_entry()
-        populate_entry_from_file(filename, base_url, entry)
+        populate_entry_from_file(filename, args.base_url, entry)
         print("Adding {} with title '{}'...".format(filename, entry.title()))
         if entry.updated() > latest_update:
             latest_update = entry.updated()
     feed.updated(latest_update)
 
     # Write file
-    feed.atom_file('atom.xml', pretty=True)
+    feed.atom_file(args.output, pretty=True)
+    print("Wrote Atom feed to {}.".format(args.output))
 
 if __name__ == "__main__":
     main()