from bs4 import BeautifulSoup import MySQLdb import markovify from mastodon import Mastodon, MastodonUnauthorizedError import html, re, json cfg = json.load(open('config.json')) class nlt_fixed(markovify.NewlineText): # modified version of NewlineText that never rejects sentences def test_sentence_input(self, sentence): return True # all sentences are valid <3 def extract_post(post): post = html.unescape(post) # convert HTML escape codes to text soup = BeautifulSoup(post, "html.parser") for lb in soup.select("br"): # replace
with linebreak lb.insert_after("\n") lb.decompose() for p in soup.select("p"): # ditto for

p.insert_after("\n") p.unwrap() for ht in soup.select("a.hashtag"): # convert hashtags from links to text ht.unwrap() for link in soup.select("a"): #ocnvert 1: id = args[1] acct = args[3] handle = args[0] client = Mastodon( client_id = bot['client_id'], client_secret = bot['client_secret'], access_token = bot['secret'], api_base_url = "https://{}".format(handle.split("@")[2]) ) bot, post = generate_output(handle) # print(post) visibility = bot['post_privacy'] if len(args) == 1 else args[2] visibilities = ['public', 'unlisted', 'private'] if visibilities.index(visibility) < visibilities.index(bot['post_privacy']): # if post_privacy is set to a more restricted level than the visibility of the post we're replying to, use the user's setting visibility = bot['post_privacy'] if acct is not None: post = "{} {}".format(acct, post) # ensure post isn't longer than bot['length'] # TODO: ehhhhhhhhh post = post[:bot['length']] # send toot!! print(post) return try: client.status_post(post, id, visibility = visibility, spoiler_text = bot['content_warning']) except MastodonUnauthorizedError: # user has revoked the token given to the bot # this needs to be dealt with properly later on, but for now, we'll just disable the bot c.execute("UPDATE bots SET enabled = FALSE WHERE handle = %s", (handle,)) if id == None: # this wasn't a reply, it was a regular post, so update the last post date c.execute("UPDATE bots SET last_post = CURRENT_TIMESTAMP() WHERE handle = %s", (handle,)) db.commit()