ticketfrei/retootbot.py

144 lines
4.7 KiB
Python
Raw Normal View History

2017-06-17 18:49:30 +00:00
#!/usr/bin/env python3
import pytoml as toml
import mastodon
import os
import pickle
import re
import time
2017-06-25 17:15:38 +00:00
import trigger
2017-12-30 15:33:34 +00:00
import logger
import sendmail
import traceback
import sys
2017-12-30 15:33:34 +00:00
2017-06-17 18:49:30 +00:00
class RetootBot(object):
2017-12-30 15:33:34 +00:00
def __init__(self, config, filter, logger):
2017-06-17 18:49:30 +00:00
self.config = config
2017-06-25 15:50:03 +00:00
self.filter = filter
self.client_id = self.register()
self.m = self.login()
2017-06-17 18:49:30 +00:00
# load state
try:
with open('seen_toots.pickle', 'rb') as f:
self.seen_toots = pickle.load(f)
except IOError:
self.seen_toots = set()
# intialize shutdown contact
try:
self.no_shutdown_contact = False
self.contact = self.config['mail']['contact']
except KeyError:
self.no_shutdown_contact = True
2017-12-30 15:33:34 +00:00
self.logger = logger
2017-07-20 20:30:43 +00:00
2017-06-17 18:49:30 +00:00
def register(self):
client_id = os.path.join(
2017-12-30 15:33:34 +00:00
'appkeys',
self.config['mapp']['name'] +
'@' + self.config['muser']['server']
)
2017-06-17 18:49:30 +00:00
if not os.path.isfile(client_id):
2017-06-17 18:49:30 +00:00
mastodon.Mastodon.create_app(
2017-12-30 15:33:34 +00:00
self.config['mapp']['name'],
api_base_url=self.config['muser']['server'],
to_file=client_id
2017-12-30 15:33:34 +00:00
)
2018-01-04 10:05:36 +00:00
return client_id
2017-06-17 18:49:30 +00:00
def login(self):
m = mastodon.Mastodon(
2017-12-30 15:33:34 +00:00
client_id=self.client_id,
api_base_url=self.config['muser']['server']
)
m.log_in(
2017-12-30 15:33:34 +00:00
self.config['muser']['email'],
self.config['muser']['password']
)
return m
2017-06-17 18:49:30 +00:00
2017-06-17 22:35:34 +00:00
def retoot(self, toots=()):
2017-06-17 18:49:30 +00:00
# toot external provided messages
for toot in toots:
self.m.toot(toot)
# boost mentions
retoots = []
for notification in self.m.notifications():
if (notification['type'] == 'mention'
and notification['status']['id'] not in self.seen_toots):
self.seen_toots.add(notification['status']['id'])
text_content = re.sub(r'<[^>]*>', '',
notification['status']['content'])
if not self.filter.is_ok(text_content):
continue
2017-12-30 15:33:34 +00:00
self.logger.log('Boosting toot from %s: %s' % (
# notification['status']['id'],
notification['status']['account']['acct'],
notification['status']['content']))
self.m.status_reblog(notification['status']['id'])
retoots.append('%s: %s' % (
notification['status']['account']['acct'],
re.sub(r'@\S*', '', text_content)))
2017-10-02 18:12:58 +00:00
# If the Mastodon instance returns interesting Errors, add them here:
2017-06-17 18:49:30 +00:00
# save state
2017-12-30 10:31:16 +00:00
with os.fdopen(os.open('seen_toots.pickle.part', os.O_WRONLY | os.O_EXCL | os.O_CREAT), 'wb') as f:
2017-06-17 18:49:30 +00:00
pickle.dump(self.seen_toots, f)
os.rename('seen_toots.pickle.part', 'seen_toots.pickle')
# return mentions for mirroring
return retoots
def shutdown(self, tb):
""" If something breaks, it shuts down the bot and messages the owner.
"""
logmessage = "Shit went wrong, closing down.\n" + tb + "\n\n"
if self.no_shutdown_contact:
self.logger.log(logmessage)
return
logmessage = logmessage + " Sending message to " + self.contact
self.logger.log(logmessage)
# feature yet only in RetweetBot:
# self.save_last_mention()
try:
mailer = sendmail.Mailer(self.config)
mailer.send(tb, self.contact, "Ticketfrei Crash Report")
except:
# traceback.print_exc()
self.logger.log(traceback.extract_tb(sys.exc_info()[2]))
print()
2017-06-17 18:49:30 +00:00
if __name__ == '__main__':
# read config in TOML format (https://github.com/toml-lang/toml#toml)
2017-12-30 09:32:20 +00:00
with open('config.toml') as configfile:
2017-06-25 17:15:38 +00:00
config = toml.load(configfile)
trigger = trigger.Trigger(config)
2017-12-30 15:33:34 +00:00
logger = logger.Logger()
bot = RetootBot(config, trigger, logger)
try:
while True:
bot.retoot()
time.sleep(1)
except KeyboardInterrupt:
2018-01-04 11:20:59 +00:00
print("Good bye. Remember to restart the bot!")
except:
2018-01-04 11:20:59 +00:00
exc = sys.exc_info() # returns tuple [Exception type, Exception object, Traceback object]
tb = traceback.extract_tb(exc[2]) # returns StackSummary object
tb = "\n".join(tb.format()) # string of the actual traceback
message = ("Traceback (most recent call last):\n",
tb,
exc[0].__name__) # the type of the Exception
message = "".join(message) # concatenate to full traceback message
bot.logger.log(message)
bot.shutdown(message)