amber-ebooks/gen.py

43 lines
1.7 KiB
Python
Raw Normal View History

2018-10-09 01:11:51 +00:00
#!/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
2019-01-11 12:47:42 +00:00
import functions
2018-10-09 01:11:51 +00:00
parser = argparse.ArgumentParser(description='Generate and post a toot.')
parser.add_argument('-s', '--simulate', dest='simulate', action='store_true',
2019-01-11 12:47:42 +00:00
help="Print the toot without actually posting it. Use this to make sure your bot's actually working.")
2018-10-09 01:11:51 +00:00
args = parser.parse_args()
2018-10-25 14:35:25 +00:00
cfg = json.load(open('config.json'))
2018-10-09 01:11:51 +00:00
client = Mastodon(
2018-10-25 14:35:25 +00:00
client_id=cfg['client']['id'],
client_secret=cfg['client']['secret'],
access_token=cfg['secret'],
api_base_url=cfg['site'])
2018-10-09 01:11:51 +00:00
if __name__ == '__main__':
2019-01-11 12:47:42 +00:00
toot = functions.make_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 = {
2019-01-11 12:47:42 +00:00
"toot": "An unknown error that should never happen occurred. Maybe it's because of the spoiler text, which is {}. If not, I have no idea what went wrong. This is an error message -- contact lynnesbian@fedi.lynnesbian.space for assistance.".format(cfg['cw'])
}
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