combine bots

This commit is contained in:
Thomas L 2017-06-17 20:49:30 +02:00
parent 485b877ea4
commit 3838cf442b
2 changed files with 97 additions and 74 deletions

83
retootbot.py Normal file
View file

@ -0,0 +1,83 @@
#!/usr/bin/env python3
import pytoml as toml
import mastodon
import os
import pickle
import re
import time
class RetootBot(object):
def __init__(self, config):
self.config = config
self.register()
self.login()
# load state
try:
with open('seen_toots.pickle', 'rb') as f:
self.seen_toots = pickle.load(f)
except IOError:
self.seen_toots = set()
def register(self):
self.client_id = os.path.join(
'appkeys',
self.config['mapp']['name'] +
'@' + self.config['muser']['server']
)
if not os.path.isfile(self.client_id):
mastodon.Mastodon.create_app(
self.config['mapp']['name'],
api_base_url=self.config['muser']['server'],
to_file=self.client_id
)
def login(self):
self.m = mastodon.Mastodon(
client_id=self.client_id,
api_base_url=self.config['muser']['server']
)
self.m.log_in(
self.config['muser']['email'],
self.config['muser']['password']
)
def retoot(self, toots=[]):
# 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):
print('Boosting toot %d from %s: %s' % (
notification['status']['id'],
notification['status']['account']['acct'],
notification['status']['content']))
self.m.status_reblog(notification['status']['id'])
retoots.append(re.sub('<[^>]*>', '',
notification['status']['content']))
self.seen_toots.add(notification['status']['id'])
# save state
with open('seen_toots.pickle.part', 'xb') as f:
pickle.dump(self.seen_toots, f)
os.rename('seen_toots.pickle.part', 'seen_toots.pickle')
# return mentions for mirroring
return retoots
if __name__ == '__main__':
# read config in TOML format (https://github.com/toml-lang/toml#toml)
with open('ticketfrei.cfg') as configfile:
bot = RetootBot(toml.load(configfile))
while True:
bot.retoot()
time.sleep(1)

View file

@ -1,83 +1,23 @@
#!/usr/bin/env python3
import pytoml as toml import pytoml as toml
import mastodon
import os
import pickle
import re
import time import time
from retootbot import RetootBot
class RetootBot(object): from retweetbot import RetweetBot
def __init__(self, config):
self.config = config
self.register()
self.login()
# load state
try:
with open('seen_toots.pickle', 'rb') as f:
self.seen_toots = pickle.load(f)
except IOError:
self.seen_toots = set()
def register(self):
self.client_id = os.path.join(
'appkeys',
self.config['mapp']['name'] +
'@' + self.config['muser']['server']
)
if not os.path.isfile(self.client_id):
mastodon.Mastodon.create_app(
self.config['mapp']['name'],
api_base_url=self.config['muser']['server'],
to_file=self.client_id
)
def login(self):
self.m = mastodon.Mastodon(
client_id=self.client_id,
api_base_url=self.config['muser']['server']
)
self.m.log_in(
self.config['muser']['email'],
self.config['muser']['password']
)
def retoot(self, toots=[]):
# 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):
print('Boosting toot %d from %s: %s' % (
notification['status']['id'],
notification['status']['account']['acct'],
notification['status']['content']))
self.m.status_reblog(notification['status']['id'])
retoots.append(re.sub('<[^>]*>', '',
notification['status']['content']))
self.seen_toots.add(notification['status']['id'])
# save state
with open('seen_toots.pickle.part', 'xb') as f:
pickle.dump(self.seen_toots, f)
os.rename('seen_toots.pickle.part', 'seen_toots.pickle')
# return mentions for mirroring
return retoots
if __name__ == '__main__': if __name__ == '__main__':
# read config in TOML format (https://github.com/toml-lang/toml#toml) # read config in TOML format (https://github.com/toml-lang/toml#toml)
with open('ticketfrei.cfg') as configfile: with open('ticketfrei.cfg') as configfile:
bot = RetootBot(toml.load(configfile)) config = toml.load(configfile)
while True: mbot = RetootBot(config)
bot.retoot() tbot = RetweetBot()
time.sleep(1)
try:
statuses = []
while True:
statuses = mbot.retoot(statuses)
statuses = tbot.flow(statuses) # XXX not implemented in RetweetBot
time.sleep(1)
except:
tbot.shutdown()