limebooks/gen.py

#!/usr/bin/env python3
# SPDX-License-Identifier: EUPL-1.2

from mastodon import Mastodon
import argparse, json, re
import functions

parser = argparse.ArgumentParser(description='Generate and post a toot.')
parser.add_argument(
	'-c', '--cfg', dest='cfg', default='config.json', nargs='?',
	help="Specify a custom location for config.json.")
parser.add_argument(
	'-s', '--simulate', dest='simulate', action='store_true',
	help="Print the toot without actually posting it. Use this to make sure your bot's actually working.")

args = parser.parse_args()

cfg = json.load(open(args.cfg))

client = None

if not args.simulate:
	client = Mastodon(
		client_id=cfg['client']['id'],
		client_secret=cfg['client']['secret'],
		access_token=cfg['secret'],
		api_base_url=cfg['site'])

if __name__ == '__main__':
	toot = functions.make_toot(cfg)
	if cfg['strip_paired_punctuation']:
		toot = re.sub(r"[\[\]\(\)\{\}\"“”«»„]", "", toot)
	if not args.simulate:
		try:
			client.status_post(toot, visibility='unlisted', spoiler_text=cfg['cw'])
		except Exception:
			toot = "An error occurred while submitting the generated post. Contact lynnesbian@fedi.lynnesbian.space for assistance."
			client.status_post(toot, visibility='unlisted', spoiler_text="Error!")
	try:
		print(toot)
	except UnicodeEncodeError:
		print(toot.encode("ascii", "ignore"))  # encode as ASCII, dropping any non-ASCII characters