ticketfrei/retweetbot.py

249 lines
8.1 KiB
Python
Raw Normal View History

2017-06-17 17:55:52 +00:00
#!/usr/bin/env python
import twitter
2017-07-20 20:30:43 +00:00
import os
import datetime
2017-06-17 17:55:52 +00:00
import requests
import pytoml as toml
import trigger
2017-06-17 17:55:52 +00:00
from time import sleep
2017-07-11 20:05:46 +00:00
import traceback
2017-06-17 17:55:52 +00:00
2017-06-17 20:11:44 +00:00
class RetweetBot(object):
2017-06-17 17:55:52 +00:00
"""
This bot retweets all tweets which
1) mention him,
2) contain at least one of the triggerwords provided.
2017-06-17 17:55:52 +00:00
2017-06-25 17:31:40 +00:00
api: The api object, generated with your oAuth keys, responsible for
communication with twitter rest API
triggers: a list of words, one of them has to be in a tweet for it to be
retweeted
2017-06-17 17:55:52 +00:00
last_mention: the ID of the last tweet which mentioned you
"""
2017-07-20 20:30:43 +00:00
def __init__(self, trigger, config, historypath="last_mention", logpath=None):
2017-06-17 17:55:52 +00:00
"""
Initializes the bot and loads all the necessary data.
2017-06-25 17:31:40 +00:00
:param historypath: Path to the file with ID of the last retweeted
Tweet
2017-06-17 17:55:52 +00:00
"""
self.config = config
keys = self.get_api_keys()
self.api = twitter.Api(consumer_key=keys[0],
consumer_secret=keys[1],
access_token_key=keys[2],
access_token_secret=keys[3])
2017-06-17 17:55:52 +00:00
self.historypath = historypath
2017-06-25 17:31:40 +00:00
try:
self.user_id = self.config['tapp']['shutdown_contact_userid']
self.screen_name = \
self.config['tapp']['shutdown_contact_screen_name']
except KeyError:
self.no_shutdown_contact = True
2017-06-17 20:32:20 +00:00
self.last_mention = self.get_history(self.historypath)
self.trigger = trigger
2017-07-20 20:30:43 +00:00
if logpath:
self.logpath = logpath
else:
self.logpath = os.path.join("logs", "{%Y-%m-%d_%H:%M:%S}".format(datetime.datetime.now()))
print "Path of logfile: " + self.logpath
2017-06-17 17:55:52 +00:00
def get_api_keys(self):
2017-06-17 17:55:52 +00:00
"""
How to get these keys is described in doc/twitter_api.md
After you received keys, store them in your ticketfrei.cfg like this:
[tapp]
consumer_key = "..."
consumer_secret = "..."
[tuser]
access_token_key = "..."
access_token_secret = "..."
:return: keys: list of these 4 strings.
2017-06-17 17:55:52 +00:00
"""
keys = []
keys.append(self.config['tapp']['consumer_key'])
keys.append(self.config['tapp']['consumer_secret'])
keys.append(self.config['tuser']['access_token_key'])
keys.append(self.config['tuser']['access_token_secret'])
2017-06-17 17:55:52 +00:00
return keys
2017-07-20 20:30:43 +00:00
def log(self, message, tb=False):
"""
Writing an error message to a logfile in logs/ and prints it.
:param message(string): Log message to be displayed
:param tb: String of the Traceback
"""
2017-07-20 20:44:44 +00:00
time = str(datetime.datetime.now())
2017-07-20 20:30:43 +00:00
if tb:
message = message + " The traceback is located at " + os.path.join("logs" + time)
with open(os.path.join("logs", time), 'w+') as f:
f.write(tb)
line = "[" + time + "] "+ message + "\n"
2017-07-20 20:30:43 +00:00
with open(self.logpath, 'a') as f:
f.write(line)
print line,
2017-07-20 20:30:43 +00:00
2017-06-17 17:55:52 +00:00
def get_history(self, path):
2017-06-25 17:31:40 +00:00
""" This counter is needed to keep track of your mentions, so you
don't double RT them
2017-06-25 17:31:40 +00:00
:param path: string: contains path to the file where the ID of the
last_mention is stored.
:return: last_mention: ID of the last tweet which mentioned the bot
"""
2017-06-17 20:32:20 +00:00
try:
with open(path, "r+") as f:
last_mention = f.read()
except IOError:
with open(path, "w+") as f:
last_mention = "0"
f.write(last_mention)
return int(last_mention)
2017-06-17 17:55:52 +00:00
2017-07-19 10:17:15 +00:00
def save_last_mention(self):
""" Saves the last retweeted tweet in last_mention. """
with open(self.historypath, "w") as f:
f.write(str(self.last_mention))
def format_mastodon(self, status):
2017-06-17 17:55:52 +00:00
"""
Bridge your Retweets to mastodon.
:todo vmann: add all the mastodon API magic.
:param status: Object of a tweet.
2017-06-25 17:31:40 +00:00
:return: toot: text tooted on mastodon, e.g. "_b3yond: There are
uniformed controllers in the U2 at Opernhaus."
2017-06-17 17:55:52 +00:00
"""
toot = status.user.name + ": " + status.text
return toot
def crawl_mentions(self):
"""
crawls all Tweets which mention the bot from the twitter rest API.
:return: list of Status objects
"""
while 1:
try:
mentions = self.api.GetMentions(since_id=self.last_mention)
return mentions
2017-06-17 22:35:34 +00:00
except twitter.TwitterError:
2017-07-20 20:30:43 +00:00
self.log("Twitter API Error: Rate Limit Exceeded.")
sleep(120)
except requests.exceptions.ConnectionError:
2017-07-20 20:30:43 +00:00
self.log("Twitter API Error: Bad Connection.")
sleep(10)
def retweet(self, status):
"""
Retweets a given tweet.
:param status: A tweet object.
:return: toot: string of the tweet, to toot on mastodon.
"""
while 1:
try:
self.api.PostRetweet(status.id)
2017-07-20 20:30:43 +00:00
self.log("Retweeted: " + self.format_mastodon(status))
2017-07-19 10:17:15 +00:00
if status.id > self.last_mention:
self.last_mention = status.id
return self.format_mastodon(status)
2017-06-25 17:31:40 +00:00
# maybe one day we get rid of this error. If not, try to uncomment
# these lines.
except twitter.error.TwitterError:
2017-07-20 20:30:43 +00:00
self.log("Twitter API Error: You probably already retweeted this tweet: " + status.text)
2017-07-19 10:17:15 +00:00
if status.id > self.last_mention:
self.last_mention = status.id
return None
except requests.exceptions.ConnectionError:
2017-07-20 20:30:43 +00:00
self.log("Twitter API Error: Bad Connection.")
sleep(10)
def tweet(self, post):
"""
Tweet a post.
:param post: String with the text to tweet.
"""
2017-07-11 21:50:55 +00:00
if len(post) > 140:
post = post[:140 - 4] + u' ...'
while 1:
try:
self.api.PostUpdate(status=post)
return
except requests.exceptions.ConnectionError:
2017-07-20 20:30:43 +00:00
self.log("Twitter API Error: Bad Connection.")
sleep(10)
def flow(self, to_tweet=()):
""" The flow of crawling mentions and retweeting them.
:param to_tweet: list of strings to tweet
:return list of retweeted tweets, to toot on mastodon
"""
# Tweet the toots the Retootbot gives to us
for post in to_tweet:
self.tweet(post)
2017-06-17 17:55:52 +00:00
# Store all mentions in a list of Status Objects
mentions = self.crawl_mentions()
mastodon = []
for status in mentions:
# Is the Text of the Tweet in the triggerlist?
2017-06-25 17:15:38 +00:00
if self.trigger.is_ok(status.text):
2017-06-17 20:11:44 +00:00
# Retweet status
toot = self.retweet(status)
if toot:
mastodon.append(toot)
# save the id so it doesn't get crawled again
if status.id > self.last_mention:
self.last_mention = status.id
2017-07-19 10:17:15 +00:00
self.save_last_mention()
2017-06-17 20:11:44 +00:00
# Return Retweets for tooting on mastodon
return mastodon
def shutdown(self):
2017-06-25 17:31:40 +00:00
""" If something breaks, it shuts down the bot and messages the owner.
"""
2017-07-20 20:30:43 +00:00
logmessage = "Shit went wrong, closing down."
if self.screen_name:
logmessage = logmessage + " Sending message to " + self.screen_name
self.log(logmessage)
2017-06-25 17:31:40 +00:00
if self.no_shutdown_contact:
return
2017-07-19 10:17:15 +00:00
self.save_last_mention()
2017-07-20 20:30:43 +00:00
try:
self.api.PostDirectMessage("Help! I broke down. restart me pls :$",
2017-06-25 17:31:40 +00:00
self.user_id, self.screen_name)
2017-07-20 20:30:43 +00:00
except:
traceback.print_exc()
print
2017-06-17 17:55:52 +00:00
2017-06-17 20:32:20 +00:00
if __name__ == "__main__":
2017-06-17 17:55:52 +00:00
# create an Api object
with open('ticketfrei.cfg') as configfile:
config = toml.load(configfile)
trigger = trigger.Trigger(config)
bot = RetweetBot(trigger, config)
2017-06-25 19:42:06 +00:00
try:
while True:
bot.flow()
sleep(10)
2017-07-11 20:05:46 +00:00
except:
2017-07-19 10:17:15 +00:00
traceback.print_exc()
2017-07-20 20:30:43 +00:00
print
2017-06-25 19:42:06 +00:00
bot.shutdown()