Lediver.se/fediverse-blocklist-deploy
2
0
Fork 0
Code Issues 3 Pull requests 1 Projects Releases Wiki Activity

feat: Allow token via environment variables

Browse source
This commit is contained in:
Julian-Samuel Gebühr 2023-01-12 16:22:03 +01:00
parent c7872201ea
commit 0b49740e83
1 changed files with 10 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
mastodon_blocklist_deploy/cli.py
View file

@ -3,7 +3,7 @@ import argparse
import json
import logging
import requests
import os
import toml
from mastodon_blocklist_deploy.models import Instance
@ -76,12 +76,19 @@ def cli():
else:
logging.basicConfig(level=logging.WARN)
if args.token:
token = args.token
else:
token = os.getenv('MBD_TOKEN')
"""if there is a remote blocklist provided load this instead of fetching it from a server (for debugging reasons)"""
if args.remote_blocklist:
with open(args.remote_blocklist) as f:
remote_blocklist = blocklist_json_to_instances(json.load(f))
else:
remote_blocklist = load_blocklist_from_instance(server=args.server, token=args.token)
remote_blocklist = load_blocklist_from_instance(server=args.server, token=token)
"""Load local blocklist only when needed"""
if args.action in ["diff", "deploy"]:
@ -95,7 +102,7 @@ def cli():
Instance.show_diffs(local_blocklist, remote_blocklist)
elif args.action == "deploy":
diffs = Instance.list_diffs(local_blocklist, remote_blocklist)
Instance.apply_blocks_from_diff(diffs, args.server, args.token, args.no_delete)
Instance.apply_blocks_from_diff(diffs, args.server, token, args.no_delete)
elif args.action == "export":
if not args.output:
print("Export currently requires to pass --output as well")