diff --git a/retootbot.py b/retootbot.py
index 63e16da..d1c6fce 100644
--- a/retootbot.py
+++ b/retootbot.py
@@ -6,12 +6,12 @@ import os
import pickle
import re
import time
-
+import datetime
import trigger
class RetootBot(object):
- def __init__(self, config, filter):
+ def __init__(self, config, filter, logpath=None):
self.config = config
self.filter = filter
self.register()
@@ -24,6 +24,28 @@ class RetootBot(object):
except IOError:
self.seen_toots = set()
+ if logpath:
+ self.logpath = logpath
+ else:
+ self.logpath = os.path.join("logs", "{%Y-%m-%d_%H:%M:%S}".format(datetime.datetime.now()))
+
+ 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
+ """
+ time = "{%Y-%m-%d_%H:%M:%S}".format(datetime.datetime.now())
+ 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
+ with open(self.logpath, 'a') as f:
+ f.write(line)
+ print line
+
def register(self):
self.client_id = os.path.join(
'appkeys',
@@ -63,7 +85,7 @@ class RetootBot(object):
notification['status']['content'])
if not self.filter.is_ok(text_content):
continue
- print('Boosting toot %d from %s: %s' % (
+ self.log('Boosting toot %d from %s: %s' % (
notification['status']['id'],
notification['status']['account']['acct'],
notification['status']['content']))
diff --git a/retweetbot.py b/retweetbot.py
index 3ec7812..bd1ef31 100644
--- a/retweetbot.py
+++ b/retweetbot.py
@@ -1,6 +1,7 @@
#!/usr/bin/env python
import twitter
+import os
import datetime
import requests
import pytoml as toml
@@ -22,8 +23,7 @@ class RetweetBot(object):
last_mention: the ID of the last tweet which mentioned you
"""
- def __init__(self, trigger, config,
- historypath="last_mention"):
+ def __init__(self, trigger, config, historypath="last_mention", logpath=None):
"""
Initializes the bot and loads all the necessary data.
@@ -45,6 +45,11 @@ class RetweetBot(object):
self.no_shutdown_contact = True
self.last_mention = self.get_history(self.historypath)
self.trigger = trigger
+ 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
def get_api_keys(self):
"""
@@ -68,6 +73,23 @@ class RetweetBot(object):
keys.append(self.config['tuser']['access_token_secret'])
return keys
+ 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
+ """
+ time = "{%Y-%m-%d_%H:%M:%S}".format(datetime.datetime.now())
+ 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
+ with open(self.logpath, 'a') as f:
+ f.write(line)
+ print line
+
def get_history(self, path):
""" This counter is needed to keep track of your mentions, so you
don't double RT them
@@ -113,11 +135,10 @@ class RetweetBot(object):
mentions = self.api.GetMentions(since_id=self.last_mention)
return mentions
except twitter.TwitterError:
- traceback.print_exc()
- print "[ERROR] Rate Limit Exceeded at {:%d.%m.%Y %H:%M:%S}".format(datetime.datetime.now()) # debug
+ self.log("Twitter API Error: Rate Limit Exceeded.")
sleep(60)
except requests.exceptions.ConnectionError:
- print("[ERROR] Bad Connection.")
+ self.log("Twitter API Error: Bad Connection.")
sleep(10)
def retweet(self, status):
@@ -130,21 +151,19 @@ class RetweetBot(object):
while 1:
try:
self.api.PostRetweet(status.id)
- print("Tweeted: " + self.format_mastodon(status)) # debug
+ self.log("Retweeted: " + self.format_mastodon(status))
if status.id > self.last_mention:
self.last_mention = status.id
return self.format_mastodon(status)
# maybe one day we get rid of this error. If not, try to uncomment
# these lines.
except twitter.error.TwitterError:
- traceback.print_exc()
- print("[ERROR] probably you already retweeted this tweet: " + status.text)
+ self.log("Twitter API Error: You probably already retweeted this tweet: " + status.text)
if status.id > self.last_mention:
self.last_mention = status.id
return None
except requests.exceptions.ConnectionError:
- traceback.print_exc()
- print("[ERROR] Bad Connection.")
+ self.log("Twitter API Error: Bad Connection.")
sleep(10)
def tweet(self, post):
@@ -160,8 +179,7 @@ class RetweetBot(object):
self.api.PostUpdate(status=post)
return
except requests.exceptions.ConnectionError:
- traceback.print_exc()
- print("[ERROR] Bad Connection.")
+ self.log("Twitter API Error: Bad Connection.")
sleep(10)
def flow(self, to_tweet=()):
@@ -197,12 +215,19 @@ class RetweetBot(object):
def shutdown(self):
""" If something breaks, it shuts down the bot and messages the owner.
"""
- print("[ERROR] Shit went wrong, closing down.")
+ logmessage = "Shit went wrong, closing down."
+ if self.screen_name:
+ logmessage = logmessage + " Sending message to " + self.screen_name
+ self.log(logmessage)
if self.no_shutdown_contact:
return
self.save_last_mention()
- self.api.PostDirectMessage("Help! I broke down. restart me pls :$",
+ try:
+ self.api.PostDirectMessage("Help! I broke down. restart me pls :$",
self.user_id, self.screen_name)
+ except:
+ traceback.print_exc()
+ print
if __name__ == "__main__":
@@ -219,4 +244,5 @@ if __name__ == "__main__":
sleep(10)
except:
traceback.print_exc()
+ print
bot.shutdown()
diff --git a/ticketfrei.py b/ticketfrei.py
index 460f5ca..9fea877 100644
--- a/ticketfrei.py
+++ b/ticketfrei.py
@@ -1,6 +1,10 @@
+#!/usr/bin/env python
+
import pytoml as toml
import time
import traceback
+import os
+import datetime
from retootbot import RetootBot
from retweetbot import RetweetBot
@@ -14,8 +18,10 @@ if __name__ == '__main__':
trigger = Trigger(config)
- mbot = RetootBot(config, trigger)
- tbot = RetweetBot(trigger, config)
+ logpath = os.path.join("logs", "{%Y-%m-%d_%H:%M:%S}".format(datetime.datetime.now()))
+
+ mbot = RetootBot(config, trigger, logpath=logpath)
+ tbot = RetweetBot(trigger, config, logpath=logpath)
try:
statuses = []