patrick/feediverse
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #6 from htgoebel/issue-5

Browse Source 
Add minimal argparse support to get support for `--help`.
This commit is contained in:
Ed Summers
2020-09-29 10:19:48 -04:00
committed by GitHub
parent 350764b352 3f4d051b84
commit bc593134c4
1 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
feediverse.py
View File

@ -2,6 +2,7 @@
import os import os
import sys import sys
import argparse
import yaml import yaml
import dateutil import dateutil
import feedparser import feedparser
@ -9,8 +10,17 @@ import feedparser
from mastodon import Mastodon from mastodon import Mastodon
from datetime import datetime, timezone from datetime import datetime, timezone
DEFAULT_CONFIG_FILE = os.path.join("~", ".feediverse")
def main(): def main():
config_file = get_config_file() parser = argparse.ArgumentParser()
parser.add_argument("config_file", nargs="?", metavar="CONFIG-FILE",
help=("config file to use, default: %s" %
DEFAULT_CONFIG_FILE),
default=os.path.expanduser(DEFAULT_CONFIG_FILE))
args = parser.parse_args()
config_file = args.config_file
if not os.path.isfile(config_file): if not os.path.isfile(config_file):
setup(config_file) setup(config_file)
@ -29,13 +39,6 @@ def main():
save_config(config, config_file) save_config(config, config_file)
def get_config_file():
if __name__ == "__main__" and len(sys.argv) > 1:
config_file = sys.argv[1]
else:
config_file = os.path.join(os.path.expanduser("~"), ".feediverse")
return config_file
def save_config(config, config_file): def save_config(config, config_file):
copy = dict(config) copy = dict(config)
copy['updated'] = datetime.now(tz=timezone.utc).isoformat() copy['updated'] = datetime.now(tz=timezone.utc).isoformat()