From f61fa0a82e8b91c86f19b913afd1d9fc30051774 Mon Sep 17 00:00:00 2001 From: Solderpunk Date: Mon, 16 Mar 2020 19:45:04 +0100 Subject: Add command line options for base URL and feed title. --- gemfeed.py | 32 +++++++++++++++++++++++++------- 1 file 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() -- cgit v1.2.3-54-g00ecf