lluni/reddit-mastodon-crossposter
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add starting cleanup from arbitrary post IDs to combat rate limits

Browse source
This commit is contained in:
lluni 2023-08-01 22:49:29 +02:00
parent f9b720e2e1
commit a35a6303b6
Signed by: lluni
GPG key ID: ACEEB468BC325D35
1 changed files with 49 additions and 33 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
cleanup.py
View file

@ -6,6 +6,7 @@ from dotenv import load_dotenv
from mastodon import Mastodon from mastodon import Mastodon
from os import getenv from os import getenv
def cleanup(limit: int = 40, id: int | None = None):
# ----- initialization ----- # # ----- initialization ----- #
# load .env # load .env
@ -17,28 +18,40 @@ mastodon = Mastodon(
api_base_url = getenv("MASTODON_URL") api_base_url = getenv("MASTODON_URL")
) )
# ----- go 200 posts into the past ----- # # ----- go 200 posts into the past, if manual id is not set ----- #
if id is None:
# get latest posts # get latest posts
last_posts = mastodon.account_statuses(getenv("MASTODON_USER_ID"), exclude_replies=True, limit=40) last_posts = mastodon.account_statuses(getenv("MASTODON_USER_ID"), exclude_replies=True, limit=limit)
id_oldest_of_last_posts = -1 id_oldest_of_last_posts = -1
for i in range(4): for _ in range(4):
id_oldest_of_last_posts = last_posts[-1]["id"] id_oldest_of_last_posts = last_posts[-1]["id"]
last_posts = mastodon.account_statuses(getenv("MASTODON_USER_ID"), exclude_replies=True, max_id=id_oldest_of_last_posts, limit=40) last_posts = mastodon.account_statuses(getenv("MASTODON_USER_ID"), exclude_replies=True, max_id=id_oldest_of_last_posts, limit=limit)
else:
id_oldest_of_last_posts = id
last_posts = mastodon.account_statuses(getenv("MASTODON_USER_ID"), exclude_replies=True, max_id=id, limit=limit)
# return if there are no posts to check
if len(last_posts) == 0:
return
# ----- delete old posts ----- # # ----- delete old posts ----- #
for i in range(5): # default: check 5 batches of 40 posts # default: check 2 batches of 40 posts (and 6 batches if the id was set manually)
num_batches = 2 if id is None else 6
for i in range(num_batches):
# skip getting new batch in the first iteration if the id was set manually
if i != 0 or id is None:
# find id of the currently oldest post # find id of the currently oldest post
last_post = last_posts[-1] last_post = last_posts[-1]
if last_post["favourites_count"] != 0 or last_post["reblogs_count"] != 0 or last_post["replies_count"] != 0: if last_post["favourites_count"] != 0 or last_post["reblogs_count"] != 0 or last_post["replies_count"] != 0:
id_oldest_of_last_posts = last_post["id"] id_oldest_of_last_posts = last_post["id"]
# get next batch of posts # get next batch of posts
last_posts = mastodon.account_statuses(getenv("MASTODON_USER_ID"), exclude_replies=True, max_id=id_oldest_of_last_posts, limit=40) last_posts = mastodon.account_statuses(getenv("MASTODON_USER_ID"), exclude_replies=True, max_id=id_oldest_of_last_posts, limit=limit)
# break if there are no more posts # return if there are no more posts to check
if len(last_posts) == 0: if len(last_posts) == 0:
break return
# delete all old posts of the current batch # delete all old posts of the current batch
for submission in last_posts: for submission in last_posts:
@ -46,3 +59,6 @@ for i in range(5): # default: check 5 batches of 40 posts
if submission["favourites_count"] == 0 and submission["reblogs_count"] == 0 and submission["replies_count"] == 0: if submission["favourites_count"] == 0 and submission["reblogs_count"] == 0 and submission["replies_count"] == 0:
print("Trying to delete post:", submission["url"]) print("Trying to delete post:", submission["url"])
mastodon.status_delete(submission["id"]) mastodon.status_delete(submission["id"])
if __name__ == "__main__":
cleanup(limit=40, id=None)