This repository has been archived on 2022-03-22. You can view files and clone it, but cannot push or open issues or pull requests.
amber-ebooks-archived/gen.py

43 lines
1.5 KiB
Python
Raw Permalink 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
2019-07-02 10:44:48 +00:00
import argparse, json, re
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.')
2019-08-07 03:48:45 +00:00
parser.add_argument('-c', '--cfg', dest='cfg', default='config.json', nargs='?',
help="Specify a custom location for config.json.")
2018-10-09 01:11:51 +00:00
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()
cfg = json.load(open(args.cfg))
2018-10-09 01:11:51 +00:00
2019-05-11 14:54:27 +00:00
client = None
if not args.simulate:
client = Mastodon(
client_id=cfg['client']['id'],
2019-05-19 13:06:31 +00:00
client_secret=cfg['client']['secret'],
access_token=cfg['secret'],
2019-05-11 14:54:27 +00:00
api_base_url=cfg['site'])
2018-10-09 01:11:51 +00:00
if __name__ == '__main__':
toot = functions.make_toot(cfg)
2019-07-01 07:23:18 +00:00
if cfg['strip_paired_punctuation']:
toot = re.sub(r"[\[\]\(\)\{\}\"“”«»„]", "", toot)
if not args.simulate:
try:
2019-07-02 10:44:13 +00:00
client.status_post(toot, visibility = 'unlisted', spoiler_text = cfg['cw'])
except Exception as err:
2019-07-02 10:43:34 +00:00
toot = "An error occurred while submitting the generated post. Contact lynnesbian@fedi.lynnesbian.space for assistance."
2019-07-02 10:44:13 +00:00
client.status_post(toot, visibility = 'unlisted', spoiler_text = "Error!")
try:
2019-07-02 10:43:34 +00:00
print(toot)
except UnicodeEncodeError:
2019-07-02 10:43:34 +00:00
print(toot.encode("ascii", "ignore")) # encode as ASCII, dropping any non-ASCII characters