pleroma-ebooks/gen.py

#!/usr/bin/env python3
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.

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

parser = argparse.ArgumentParser(description='Generate and post a toot.')
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('config.json'))

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()
	if cfg['strip_paired_punctuation']:
		toot = re.sub(r"[\[\]\(\)\{\}\"“”«»„]", "", toot)
	if not args.simulate:
		try:
			if toot['media'] != None:
				mediaID = client.media_post(toot['media'], description = toot['toot'])
				client.status_post(toot['toot'].replace("\n", " "),
					media_ids = [mediaID], visibility = "unlisted", spoiler_text = cfg['cw'])
			else:
				client.status_post(toot['toot'], visibility = 'unlisted', spoiler_text = cfg['cw'])
		except Exception as err:
			toot = {
			"toot": "An error occurred while submitting the generated post. Contact lynnesbian@fedi.lynnesbian.space for assistance."
			}
			client.status_post(toot['toot'], visibility = 'unlisted', spoiler_text = "Error!")
	try:
		print(toot['toot'])
	except UnicodeEncodeError:
		print(toot['toot'].encode("ascii", "ignore")) # encode as ASCII, dropping any non-ASCII characters