214 lines
6.3 KiB
Python
Executable File
214 lines
6.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# toot downloader version two!!
|
|
# This Source Code Form is subject to the terms of the Mozilla Public
|
|
# License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
from mastodon import Mastodon
|
|
from os import path
|
|
from bs4 import BeautifulSoup
|
|
import os, sqlite3, signal, sys, json, re
|
|
import requests
|
|
|
|
scopes = ["read:statuses", "read:accounts", "read:follows", "write:statuses", "read:notifications"]
|
|
cfg = json.load(open('config.json', 'r'))
|
|
|
|
#config.json *MUST* contain the instance URL, and the CW text. if they're not provided, we'll fall back to defaults.
|
|
if 'site' not in cfg:
|
|
cfg['website'] = "https://botsin.space"
|
|
if 'cw' not in cfg:
|
|
cfg['cw'] = ""
|
|
|
|
#if the user is using a (very!) old version that still uses the .secret files, migrate to the new method
|
|
if os.path.exists("clientcred.secret"):
|
|
print("Upgrading to new storage method")
|
|
cc = open("clientcred.secret").read().split("\n")
|
|
cfg['client'] = {
|
|
"id": cc[0],
|
|
"secret": cc[1]
|
|
}
|
|
cfg['secret'] = open("usercred.secret").read().rstrip("\n")
|
|
os.remove("clientcred.secret")
|
|
os.remove("usercred.secret")
|
|
|
|
|
|
if "client" not in cfg:
|
|
print("No application info -- registering application with {}".format(cfg['site']))
|
|
client_id, client_secret = Mastodon.create_app("mstdn-ebooks",
|
|
api_base_url=cfg['site'],
|
|
scopes=scopes,
|
|
website="https://github.com/Lynnesbian/mstdn-ebooks")
|
|
|
|
cfg['client'] = {
|
|
"id": client_id,
|
|
"secret": client_secret
|
|
}
|
|
|
|
if "secret" not in cfg:
|
|
print("No user credentials -- logging in to {}".format(cfg['site']))
|
|
client = Mastodon(client_id = cfg['client']['id'],
|
|
client_secret = cfg['client']['secret'],
|
|
api_base_url=cfg['site'])
|
|
|
|
print("Open this URL and authenticate to give mstdn-ebooks access to your bot's account: {}".format(client.auth_request_url(scopes=scopes)))
|
|
cfg['secret'] = client.log_in(code=input("Secret: "), scopes=scopes)
|
|
|
|
json.dump(cfg, open("config.json", "w+"))
|
|
|
|
def extract_toot(toot):
|
|
toot = toot.replace("'", "'")
|
|
toot = toot.replace(""", '"')
|
|
soup = BeautifulSoup(toot, "html.parser")
|
|
|
|
# this is the code that removes all mentions
|
|
for mention in soup.select("span.h-card"):
|
|
mention.a.unwrap()
|
|
mention.span.unwrap()
|
|
|
|
# replace <br> with linebreak
|
|
for lb in soup.select("br"):
|
|
lb.insert_after("\n")
|
|
lb.decompose()
|
|
|
|
# replace <p> with linebreak
|
|
for p in soup.select("p"):
|
|
p.insert_after("\n")
|
|
p.unwrap()
|
|
|
|
# fix hashtags
|
|
for ht in soup.select("a.hashtag"):
|
|
ht.unwrap()
|
|
|
|
# fix links
|
|
for link in soup.select("a"):
|
|
link.insert_after(link["href"])
|
|
link.decompose()
|
|
|
|
toot = soup.get_text()
|
|
toot = toot.rstrip("\n") #remove trailing newline
|
|
toot = toot.replace("@", "@\u200B") #put a zws between @ and username to avoid mentioning
|
|
return(toot)
|
|
|
|
client = Mastodon(
|
|
client_id=cfg['client']['id'],
|
|
client_secret = cfg['client']['secret'],
|
|
access_token=cfg['secret'],
|
|
api_base_url=cfg['site'])
|
|
|
|
me = client.account_verify_credentials()
|
|
following = client.account_following(me.id)
|
|
|
|
db = sqlite3.connect("toots.db")
|
|
db.text_factory=str
|
|
c = db.cursor()
|
|
c.execute("CREATE TABLE IF NOT EXISTS `toots` (id INT NOT NULL UNIQUE PRIMARY KEY, userid INT NOT NULL, uri VARCHAR NOT NULL, content VARCHAR NOT NULL) WITHOUT ROWID")
|
|
db.commit()
|
|
|
|
def handleCtrlC(signal, frame):
|
|
print("\nPREMATURE EVACUATION - Saving chunks")
|
|
db.commit()
|
|
sys.exit(1)
|
|
|
|
signal.signal(signal.SIGINT, handleCtrlC)
|
|
|
|
for f in following:
|
|
last_toot = c.execute("SELECT id FROM `toots` WHERE userid LIKE ? ORDER BY id DESC LIMIT 1", (f.id,)).fetchone()
|
|
if last_toot != None:
|
|
last_toot = last_toot[0]
|
|
else:
|
|
last_toot = 0
|
|
print("Harvesting toots for user @{}, starting from {}".format(f.acct, last_toot))
|
|
|
|
#find the user's activitypub outbox
|
|
print("WebFingering...")
|
|
instance = re.search(r"^.*@(.+)", f.acct)
|
|
if instance == None:
|
|
instance = re.search(r"https?:\/\/(.*)", cfg['site']).group(1)
|
|
else:
|
|
instance = instance.group(1)
|
|
|
|
if instance == "bofa.lol":
|
|
print("rest in piece bofa, skipping")
|
|
continue
|
|
|
|
# print("{} is on {}".format(f.acct, instance))
|
|
try:
|
|
r = requests.get("https://{}/.well-known/host-meta".format(instance), timeout=10)
|
|
uri = re.search(r'template="([^"]+)"', r.text).group(1)
|
|
uri = uri.format(uri = "{}@{}".format(f.username, instance))
|
|
r = requests.get(uri, headers={"Accept": "application/json"}, timeout=10)
|
|
j = r.json()
|
|
if len(j['aliases']) == 1: #TODO: this is a hack on top of a hack, fix it
|
|
uri = j['aliases'][0]
|
|
else:
|
|
uri = j['aliases'][1]
|
|
uri = "{}/outbox?page=true".format(uri)
|
|
r = requests.get(uri, timeout=10)
|
|
j = r.json()
|
|
except Exception:
|
|
print("oopsy woopsy!! we made a fucky wucky!!!\n(we're probably rate limited, please hang up and try again)")
|
|
sys.exit(1)
|
|
|
|
pleroma = False
|
|
if 'first' in j and type(j['first']) != str:
|
|
print("Pleroma instance detected")
|
|
pleroma = True
|
|
j = j['first']
|
|
else:
|
|
print("Mastodon instance detected")
|
|
uri = "{}&min_id={}".format(uri, last_toot)
|
|
r = requests.get(uri)
|
|
j = r.json()
|
|
|
|
print("Downloading and saving toots", end='', flush=True)
|
|
done = False
|
|
try:
|
|
while not done and len(j['orderedItems']) > 0:
|
|
for oi in j['orderedItems']:
|
|
if oi['type'] != "Create":
|
|
continue #not a toost. fuck outta here
|
|
|
|
# its a toost baby
|
|
content = oi['object']['content']
|
|
if oi['object']['summary'] != None:
|
|
#don't download CW'd toots
|
|
continue
|
|
toot = extract_toot(content)
|
|
# print(toot)
|
|
try:
|
|
if pleroma:
|
|
if c.execute("SELECT COUNT(*) FROM toots WHERE id LIKE ?", (oi['object']['id'],)).fetchone()[0] > 0:
|
|
#we've caught up to the notices we've already downloaded, so we can stop now
|
|
done = True
|
|
break
|
|
pid = re.search(r"[^\/]+$", oi['object']['id']).group(0)
|
|
c.execute("REPLACE INTO toots (id, userid, uri, content) VALUES (?, ?, ?, ?)",
|
|
(pid,
|
|
f.id,
|
|
oi['object']['id'],
|
|
toot
|
|
)
|
|
)
|
|
pass
|
|
except:
|
|
pass #ignore any toots that don't successfully go into the DB
|
|
# sys.exit(0)
|
|
if not pleroma:
|
|
r = requests.get(j['prev'], timeout=15)
|
|
else:
|
|
r = requests.get(j['next'], timeout=15)
|
|
j = r.json()
|
|
print('.', end='', flush=True)
|
|
print(" Done!")
|
|
db.commit()
|
|
except:
|
|
print("Encountered an error! Saving toots to database and moving to next followed account.")
|
|
db.commit()
|
|
# db.close()
|
|
|
|
print("Done!")
|
|
|
|
db.commit()
|
|
db.execute("VACUUM") #compact db
|
|
db.commit()
|
|
db.close() |