76 lines
2.5 KiB
Python
76 lines
2.5 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import json
|
|
from threading import Thread
|
|
from mastodon import Mastodon, StreamListener
|
|
|
|
|
|
def print_status(status):
|
|
print("time: " + str(status["created_at"]))
|
|
print("from: @" + status["account"]["acct"])
|
|
|
|
class TootListener(StreamListener):
|
|
"""
|
|
Listener class, handling incoming statuses
|
|
"""
|
|
def __init__(self, source, drain):
|
|
self.source = source # Source instance Mastodon object
|
|
self.drain = drain # Drain instance Mastodon object
|
|
|
|
def on_update(self, status):
|
|
# Just printing some stuff to make it look beautiful in terminal
|
|
# More for debugging than anything else
|
|
print("")
|
|
print("--")
|
|
print("RECIEVE " + self.source.api_base_url + "")
|
|
print_status(status)
|
|
print("SEARCH " + self.drain.api_base_url + "")
|
|
|
|
# Searching for the URL of the incoming status in our drain instance
|
|
try:
|
|
search = self.drain.search(status.url)
|
|
for s in search["statuses"]:
|
|
print_status(s)
|
|
except:
|
|
print("failed")
|
|
|
|
print("--")
|
|
|
|
class HashtagSpreader(Thread):
|
|
"""
|
|
Thread class, streaming one hashtag from one the source instance and searching for the post URL in the drain instance
|
|
"""
|
|
def __init__(self, source, drain, hashtag):
|
|
Thread.__init__(self) # Configure the thread
|
|
|
|
self.source = source # Source instance Mastodon object
|
|
self.drain = drain # Drain instance Mastodon object
|
|
self.hashtag = hashtag # The hashtag (without #) as a string, we want to stream
|
|
|
|
def run(self):
|
|
tootListener = TootListener(self.source, self.drain)
|
|
self.source.stream_hashtag(self.hashtag, tootListener) # Stream our hashtag
|
|
|
|
|
|
if __name__ == "__main__":
|
|
instances = {}
|
|
threads = []
|
|
|
|
with open("config.json", 'r') as f:
|
|
config = json.load(f)
|
|
|
|
if config:
|
|
# Create Mastodon objects for every instance
|
|
for instance in config["instances"]:
|
|
instances[instance] = Mastodon(access_token = config["instances"][instance], api_base_url = "https://" + instance)
|
|
|
|
# Create threads for every hashtag of a relation
|
|
for spread in config["spreads"]:
|
|
if spread["source"] in instances and spread["drain"] in instances:
|
|
for hashtag in spread["hashtags"]:
|
|
threads.append(HashtagSpreader(instances[spread["source"]], instances[spread["drain"]], hashtag))
|
|
|
|
# Start all threads
|
|
for thread in threads:
|
|
thread.start()
|